gf/net/ghttp/ghttp_server_admin_unix.go

// Copyright 2017 gf Author(https://github.com/gogf/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.

// +build !windows

package ghttp

import (
	"github.com/gogf/gf/internal/intlog"
	"os"
	"os/signal"
	"syscall"
)

// 进程信号量监听消息队列
var procSignalChan = make(chan os.Signal)

// 信号量处理
func handleProcessSignal() {
	var sig os.Signal
	signal.Notify(
		procSignalChan,
		syscall.SIGINT,
		syscall.SIGQUIT,
		syscall.SIGKILL,
		syscall.SIGTERM,
		syscall.SIGABRT,
		syscall.SIGUSR1,
		syscall.SIGUSR2,
	)
	for {
		sig = <-procSignalChan
		intlog.Printf(`signal received: %s`, sig.String())
		switch sig {
		// 进程终止，停止所有子进程运行
		case syscall.SIGINT, syscall.SIGQUIT, syscall.SIGKILL, syscall.SIGTERM, syscall.SIGABRT:
			shutdownWebServers(sig.String())
			return

		// 用户信号，重启服务
		case syscall.SIGUSR1:
			restartWebServers(sig.String())
			return

		default:
		}
	}
}