golang daemon 进程 可处理信号

用go 开源库 GitHub - sevlyar/go-daemon: A library for writing system daemons in golang.实现守护进程

package main

import (
	"flag"
	"github.com/sevlyar/go-daemon"
	"log"
	"os"
	"syscall"
	"time"
)

var (
	signal = flag.String("s", "", `Send signal to the daemon:
  quit — graceful shutdown
  stop — fast shutdown
  reload — reloading the configuration file`)
)

func main() {
	flag.Parse()
    /*注册信号处理函数，这里可以增加我们自己的信号处理函数，自定义对应的handler函数*/
	daemon.AddCommand(daemon.StringFlag(signal, "quit"), syscall.SIGQUIT, termHandler)
	daemon.AddCommand(daemon.StringFlag(signal, "stop"), syscall.SIGTERM, termHandler)
	daemon.AddCommand(daemon.StringFlag(signal, "reload"), syscall.SIGHUP, reloadHandler)

	cntxt := &daemon.Context{
		PidFileName: "sample.pid",//存进程pid的文件
		PidFilePerm: 0644,
		LogFileName: "sample.log",//进程日志文件
		LogFilePerm: 0640,
		WorkDir:     "./",//切换工作目录
		Umask:       027,
		Args:        []string{"[go-daemon sample]"},//进程名
	}

	if len(daemon.ActiveFlags()) > 0 {
		d, err := cntxt.Search()
		if err != nil {
			log.Fatalf("Unable send signal to the daemon: %s", err.Error())
		}
		daemon.SendCommands(d)
		return
	}

	d, err := cntxt.Reborn()
	if err != nil {
		log.Fatalln(err)
	}
	if d != nil {
		return
	}
	defer cntxt.Release()

	log.Println("- - - - - - - - - - - - - - -")
	log.Println("daemon started")

	go worker()//启动主循环函数

	err = daemon.ServeSignals()//信号处理函数
	if err != nil {
		log.Printf("Error: %s", err.Error())
	}

	log.Println("daemon terminated")
}

var (
	stop = make(chan struct{})
	done = make(chan struct{})
)

func worker() {
LOOP:
	for {
        //do something 在这里实现我们的主体逻辑，理论上执行完休眠一秒
		time.Sleep(time.Second) // this is work to be done by worker.
		select {
		case <-stop:
			break LOOP
		default:
		}
	}
	done <- struct{}{}
}

func termHandler(sig os.Signal) error {
	log.Println("terminating...")
	stop <- struct{}{}
	if sig == syscall.SIGQUIT {
		<-done
	}
	return daemon.ErrStop
}

func reloadHandler(sig os.Signal) error {
	log.Println("configuration reloaded")
	return nil
}

go mod init daemon

go mod tidy 

go build daemon.go

然后就会生成 daemon可执行文件。

我在上面的代码中增加一个打印，来看是否是定时执行（睡眠一秒）：

func worker() {
LOOP:
        for {
        //do something 在这里实现我们的主体逻辑，理论上执行完休眠一秒
        log.Println("this is a test")
                time.Sleep(time.Second) // this is work to be done by worker.
                select {
                case <-stop:
                        break LOOP
                default:
                }
        }
        done <- struct{}{}
}

我们看输出：

 结果如我们预期

 进程名也对应上了。

如果我们在这个代码中修改打印为如下代码，将会怎样

func worker() {
LOOP:
        for {
        //do something 在这里实现我们的主体逻辑，理论上执行完休眠一秒
        fmt.Println("this is a test")
                time.Sleep(time.Second) // this is work to be done by worker.
                select {
                case <-stop:
                        break LOOP
                default:
                }
        }
        done <- struct{}{}
}

使用fmt，当然需要在import中增加"fmt"。

结果如下：还是打印到sample.log中。

 

 具体原因我们可以先看一下/proc/110144/fd目录

 这下就可以明白，原来标准输出 2 就是sample.log文件。