Golang 实现守护进程实例

package main

import (
  "fmt"
  "os"
  "os/signal"
  "runtime"
  "time"

  "log"
  "syscall"
)

func Agent(nochdir,noclose int) int {
  var ret1,ret2 uintptr
  var err syscall.Errno

  darwin := runtime.GOOS == "darwin"

  //already a daemon process
  if syscall.Getppid() == 1 {
    return 0
  }

  //fork off the parent process
  ret1,ret2,err = syscall.RawSyscall(syscall.SYS_FORK, 0, 0, 0)
  if err != 0 {
    return -1
  }

  //failure
  if ret2 < 0 {
    os.Exit(-1)
  }

  //hand exception for darwin
  if darwin && ret2 == 1 {
    ret1 = 0
  }

  //new process success, exit the parent process
  if ret1 > 0 {
    os.Exit(0)
  }

  //change the filem mode mask
  _ = syscall.Umask(0)

  //create the new SID for the child process
  s_ret,s_errno := syscall.Setsid()
  if s_errno != nil {
    log.Printf("Error: syscall.Setsid error:%d",s_errno)
  }

  if s_ret < 0 {
    return -1
  }

  //modify the process execute directory
  if nochdir == 0 {
    os.Chdir("/")
  }

  //redirect the IO stream
  if noclose == 0 {
    f,e := os.OpenFile("/dev/null", os.O_RDWR, 0)
    if e == nil {
      fd := f.Fd()
      syscall.Dup2(int(fd), int(os.Stdin.Fd()))
      syscall.Dup2(int(fd), int(os.Stdout.Fd()))
      syscall.Dup2(int(fd), int(os.Stderr.Fd()))
    }
  }
  fmt.Println(os.Getpid())
  return 0
}

func main() {
  fmt.Println("Hello, World.\n")
  fmt.Println(os.Getpid())
  fmt.Println("this is a first go program!")
  Agent(0,1)
  for {
    fmt.Println("hello")
    fmt.Println(os.Getpid())
    time.Sleep(1 * time.Second)
  }
}

猜你喜欢

转载自www.cnblogs.com/chmyee/p/9970692.html