Golang守护进程

package main  
  
import (  
    "flag"  
    "fmt"    
    "os"    
    "os/exec"    
    "time"
 )  
  
func main() {  
    DAEMON := "daemon"  
    FOREVER := "forever"  
    //后台启动  
    daemon := flag.Bool(DAEMON, false, "run in daemon")  
    // 开启守护进程,在程序挂掉时自动重启  
    forever := flag.Bool(FOREVER, false, "run forever")  
    flag.Parse()  
    fmt.Printf("[*] PID: %d PPID: %d ARG: %s\n", os.Getpid(), os.Getppid(), os.Args)  
  
    if *daemon {  
       SubProcess(StripSlice(os.Args, "-"+DAEMON))  
       fmt.Printf("[*] Daemon running in PID: %d PPID: %d\n", os.Getpid(), os.Getppid())  
       os.Exit(0)  
    } else if *forever {  
       for {  
          cmd := SubProcess(StripSlice(os.Args, "-"+FOREVER))  
          fmt.Printf("[*] Forever running in PID: %d PPID: %d\n", os.Getpid(), os.Getppid())  
          cmd.Wait()  
       }  
    } else {  
       fmt.Printf("[*] Service running in PID: %d PPID: %d\n", os.Getpid(), os.Getppid())  
       // 程序入口  
       Run()  
    }  
}  
  
func StripSlice(slice []string, element string) []string {  
    for i := 0; i < len(slice); {  
       if slice[i] == element && i != len(slice)-1 {  
          slice = append(slice[:i], slice[i+1:]...)  
       } else if slice[i] == element && i == len(slice)-1 {  
          slice = slice[:i]  
       } else {  
          i++  
       }  
    }  
    return slice  
}  
  
func SubProcess(args []string) *exec.Cmd {  
    cmd := exec.Command(args[0], args[1:]...)  
    cmd.Stdin = os.Stdin  
    cmd.Stdout = os.Stdout  
    cmd.Stderr = os.Stderr  
    err := cmd.Start()  
    if err != nil {  
       fmt.Fprintf(os.Stderr, "[-] Error: %s\n", err)  
    }  
    return cmd  
}  
  
func Run() {  
    time.Sleep(5 * time.Second)  
    panic("panic")  // 模拟崩溃,测试forever
}

使用方法:

// 后台永久执行
go run .\main.go -forever -daemon