func run(application *app.Application) { pid, err := pidfile.New(path.Join(application.Options.Root, "h2forward.pid")) if err != nil { } defer pid.Kill() exitc := make(chan int) signc := make(chan os.Signal, 1) go func() { for { sig := <-signc switch sig { case syscall.SIGHUP: application.Refresh() continue default: exitc <- 1 break } } }() application.Main() <-exitc application.Exit() }
func daemonize(application *app.Application) { termHandler := func(sig os.Signal) error { application.Exit() return nil } reloadHandler := func(sig os.Signal) error { application.Refresh() return nil } daemon.AddCommand(nil, syscall.SIGQUIT, termHandler) daemon.AddCommand(nil, syscall.SIGTERM, termHandler) daemon.AddCommand(nil, syscall.SIGHUP, reloadHandler) cntxt := &daemon.Context{ PidFileName: "", PidFilePerm: 0644, LogFileName: "", LogFilePerm: 0640, WorkDir: "", Umask: 027, Args: []string{}, } d, err := cntxt.Reborn() if err != nil { fmt.Println("daemonize ", err) os.Exit(1) } if d != nil { os.Exit(1) } defer cntxt.Release() pid, err := pidfile.New(path.Join(application.Options.Root, "h2forward.pid")) if err != nil { fmt.Println("[h2forward] failed:", err) os.Exit(1) } defer pid.Kill() application.Main() fmt.Println("[h2forward] start", success) err = daemon.ServeSignals() if err != nil { fmt.Println("[h2forward] failed:", err) os.Exit(1) } }