Golang中实现典型的fork调用

English Version https://github.com/moby/moby/tree/master/pkg/reexec Golang中没有提供 fork 调用,只有 syscall.ForkExec os....

English Version

https://github.com/moby/moby/tree/master/pkg/reexec

Golang中没有提供 fork 调用,只有

这三个都类似于 fork exec ,但是没有类似C中的fork调用,在fork之后根据返回的pid 然后进入不同的函数。原因在:

https://stackoverflow.com/questions/28370646/how-do-i-fork-a-go-process/28371586#28371586

简要翻译一下:

fork

先看一下C里的传统使用方式:

#include <sys/types.h>
#include <unistd.h>
#include <stdio.h>
#include <sys/wait.h>

void child() {
    printf("child process\n");
}

int main() {
    printf("main process\n");
    pid_t pid = fork();
    int wstatus;

    if (pid == 0) {
        child();
    } else {
        printf("main exit\n");
        waitpid(pid, &wstatus, 0);
    }
}

运行一下:

$ gcc main.c && ./a.out 
main process
main exit
child process

我们看看Docker提供的实现的使用方式:

package main

import (
    "log"
    "os"

    "github.com/docker/docker/pkg/reexec"
)

func init() {
    log.Printf("init start, os.Args = % v\n", os.Args)
    reexec.Register("childProcess", childProcess)
    if reexec.Init() {
        os.Exit(0)
    }
}

func childProcess() {
    log.Println("childProcess")
}

func main() {
    log.Printf("main start, os.Args = % v\n", os.Args)
    cmd := reexec.Command("childProcess")
    cmd.Stdin = os.Stdin
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    if err := cmd.Start(); err != nil {
        log.Panicf("failed to run command: %s", err)
    }
    if err := cmd.Wait(); err != nil {
        log.Panicf("failed to wait command: %s", err)
    }
    log.Println("main exit")
}

跑一下:

$ go run main.go
2018/03/08 19:52:39 init start, os.Args = [/tmp/go-build209640177/b001/exe/main]
2018/03/08 19:52:39 main start, os.Args = [/tmp/go-build209640177/b001/exe/main]
2018/03/08 19:52:39 init start, os.Args = [childProcess]
2018/03/08 19:52:39 childProcess
2018/03/08 19:52:39 main exit

其原理是 init 一定会在 main 之前执行。而初次执行程序的时候 os.Args[0] 是可执行文件的名字。 但是 reexec.Command 却可以修改子进程的 os.Args[0] 。所以子进程会直接找到 reexec.Initreexec.Register 所注册的函数,然后执行,返回true。然后调用 os.Exit(0)

  • 发表于 2018-11-14 16:52
  • 阅读 ( 1292 )
  • 分类:默认分类

你可能感兴趣的文章

相关问题

0 条评论

请先 登录 后评论
不写代码的码农
Kemin

架构工程师

45 篇文章

作家榜 »

  1. Kemin 45 文章
  2. golanglover 5 文章
  3. D.Chen 4 文章
  4. salamander 1 文章
  5. 深圳-伟 1 文章
  6. 广训 1 文章
  7. PHP小菜 1 文章
  8. Undefined 0 文章