mirror of
https://gitee.com/johng/gf.git
synced 2024-12-03 20:58:47 +08:00
273 lines
7.7 KiB
Go
273 lines
7.7 KiB
Go
// Copyright 2018 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.
|
||
// pprof封装.
|
||
|
||
package ghttp
|
||
|
||
import (
|
||
"bytes"
|
||
"errors"
|
||
"fmt"
|
||
"os"
|
||
"runtime"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/gogf/gf/container/gtype"
|
||
"github.com/gogf/gf/encoding/gjson"
|
||
"github.com/gogf/gf/os/glog"
|
||
"github.com/gogf/gf/os/gproc"
|
||
"github.com/gogf/gf/os/gtime"
|
||
"github.com/gogf/gf/os/gtimer"
|
||
"github.com/gogf/gf/util/gconv"
|
||
)
|
||
|
||
const (
|
||
gADMIN_ACTION_INTERVAL_LIMIT = 2000 // (毫秒)服务开启后允许执行管理操作的间隔限制
|
||
gADMIN_ACTION_NONE = 0
|
||
gADMIN_ACTION_RESTARTING = 1
|
||
gADMIN_ACTION_SHUTINGDOWN = 2
|
||
gADMIN_ACTION_RELOAD_ENVKEY = "GF_SERVER_RELOAD"
|
||
gADMIN_ACTION_RESTART_ENVKEY = "GF_SERVER_RESTART"
|
||
gADMIN_GPROC_COMM_GROUP = "GF_GPROC_HTTP_SERVER"
|
||
)
|
||
|
||
// 用于服务管理的对象
|
||
type utilAdmin struct{}
|
||
|
||
// (进程级别)用于Web Server管理操作的互斥锁，保证管理操作的原子性
|
||
var serverActionLocker sync.Mutex
|
||
|
||
// (进程级别)用于记录上一次操作的时间(毫秒)
|
||
var serverActionLastTime = gtype.NewInt64(gtime.TimestampMilli())
|
||
|
||
// 当前服务进程所处的互斥管理操作状态
|
||
var serverProcessStatus = gtype.NewInt()
|
||
|
||
// 重启Web Server，参数支持自定义重启的可执行文件路径，不传递时默认和原有可执行文件路径一致。
|
||
// 针对*niux系统: 平滑重启
|
||
// 针对windows : 完整重启
|
||
func RestartAllServer(newExeFilePath ...string) error {
|
||
serverActionLocker.Lock()
|
||
defer serverActionLocker.Unlock()
|
||
if err := checkProcessStatus(); err != nil {
|
||
return err
|
||
}
|
||
if err := checkActionFrequence(); err != nil {
|
||
return err
|
||
}
|
||
return restartWebServers("", newExeFilePath...)
|
||
}
|
||
|
||
// 关闭所有的WebServer
|
||
func ShutdownAllServer() error {
|
||
serverActionLocker.Lock()
|
||
defer serverActionLocker.Unlock()
|
||
if err := checkProcessStatus(); err != nil {
|
||
return err
|
||
}
|
||
if err := checkActionFrequence(); err != nil {
|
||
return err
|
||
}
|
||
shutdownWebServers()
|
||
return nil
|
||
}
|
||
|
||
// 检查当前服务进程的状态
|
||
func checkProcessStatus() error {
|
||
status := serverProcessStatus.Val()
|
||
if status > 0 {
|
||
switch status {
|
||
case gADMIN_ACTION_RESTARTING:
|
||
return errors.New("server is restarting")
|
||
case gADMIN_ACTION_SHUTINGDOWN:
|
||
return errors.New("server is shutting down")
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 检测当前操作的频繁度
|
||
func checkActionFrequence() error {
|
||
interval := gtime.TimestampMilli() - serverActionLastTime.Val()
|
||
if interval < gADMIN_ACTION_INTERVAL_LIMIT {
|
||
return errors.New(fmt.Sprintf("too frequent action, please retry in %d ms", gADMIN_ACTION_INTERVAL_LIMIT-interval))
|
||
}
|
||
serverActionLastTime.Set(gtime.TimestampMilli())
|
||
return nil
|
||
}
|
||
|
||
// 平滑重启：创建一个子进程，通过环境变量传参
|
||
func forkReloadProcess(newExeFilePath ...string) error {
|
||
path := os.Args[0]
|
||
if len(newExeFilePath) > 0 {
|
||
path = newExeFilePath[0]
|
||
}
|
||
p := gproc.NewProcess(path, os.Args, os.Environ())
|
||
// 创建新的服务进程，子进程自动从父进程复制文件描述来监听同样的端口
|
||
sfm := getServerFdMap()
|
||
// 将sfm中的fd按照子进程创建时的文件描述符顺序进行整理，以便子进程获取到正确的fd
|
||
for name, m := range sfm {
|
||
for fdk, fdv := range m {
|
||
if len(fdv) > 0 {
|
||
s := ""
|
||
for _, item := range strings.Split(fdv, ",") {
|
||
array := strings.Split(item, "#")
|
||
fd := uintptr(gconv.Uint(array[1]))
|
||
if fd > 0 {
|
||
s += fmt.Sprintf("%s#%d,", array[0], 3+len(p.ExtraFiles))
|
||
p.ExtraFiles = append(p.ExtraFiles, os.NewFile(fd, ""))
|
||
} else {
|
||
s += fmt.Sprintf("%s#%d,", array[0], 0)
|
||
}
|
||
}
|
||
sfm[name][fdk] = strings.TrimRight(s, ",")
|
||
}
|
||
}
|
||
}
|
||
buffer, _ := gjson.Encode(sfm)
|
||
p.Env = append(p.Env, gADMIN_ACTION_RELOAD_ENVKEY+"="+string(buffer))
|
||
if _, err := p.Start(); err != nil {
|
||
glog.Errorf("%d: fork process failed, error:%s, %s", gproc.Pid(), err.Error(), string(buffer))
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 完整重启：创建一个新的子进程
|
||
func forkRestartProcess(newExeFilePath ...string) error {
|
||
path := os.Args[0]
|
||
if len(newExeFilePath) > 0 {
|
||
path = newExeFilePath[0]
|
||
}
|
||
// 去掉平滑重启的环境变量参数
|
||
os.Unsetenv(gADMIN_ACTION_RELOAD_ENVKEY)
|
||
env := os.Environ()
|
||
env = append(env, gADMIN_ACTION_RESTART_ENVKEY+"=1")
|
||
p := gproc.NewProcess(path, os.Args, env)
|
||
if _, err := p.Start(); err != nil {
|
||
glog.Errorf("%d: fork process failed, error:%s", gproc.Pid(), err.Error())
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 获取所有Web Server的文件描述符map
|
||
func getServerFdMap() map[string]listenerFdMap {
|
||
sfm := make(map[string]listenerFdMap)
|
||
serverMapping.RLockFunc(func(m map[string]interface{}) {
|
||
for k, v := range m {
|
||
sfm[k] = v.(*Server).getListenerFdMap()
|
||
}
|
||
})
|
||
return sfm
|
||
}
|
||
|
||
// 二进制转换为FdMap
|
||
func bufferToServerFdMap(buffer []byte) map[string]listenerFdMap {
|
||
sfm := make(map[string]listenerFdMap)
|
||
if len(buffer) > 0 {
|
||
j, _ := gjson.LoadContent(buffer)
|
||
for k, _ := range j.ToMap() {
|
||
m := make(map[string]string)
|
||
for k, v := range j.GetMap(k) {
|
||
m[k] = gconv.String(v)
|
||
}
|
||
sfm[k] = m
|
||
}
|
||
}
|
||
return sfm
|
||
}
|
||
|
||
// Web Server重启
|
||
func restartWebServers(signal string, newExeFilePath ...string) error {
|
||
serverProcessStatus.Set(gADMIN_ACTION_RESTARTING)
|
||
if runtime.GOOS == "windows" {
|
||
if len(signal) > 0 {
|
||
// 在终端信号下，立即执行重启操作
|
||
forceCloseWebServers()
|
||
forkRestartProcess(newExeFilePath...)
|
||
} else {
|
||
// 非终端信号下，异步1秒后再执行重启，目的是让接口能够正确返回结果，否则接口会报错(因为web server关闭了)
|
||
gtimer.SetTimeout(time.Second, func() {
|
||
forceCloseWebServers()
|
||
forkRestartProcess(newExeFilePath...)
|
||
})
|
||
}
|
||
} else {
|
||
if err := forkReloadProcess(newExeFilePath...); err != nil {
|
||
glog.Printf("%d: server restarts failed", gproc.Pid())
|
||
serverProcessStatus.Set(gADMIN_ACTION_NONE)
|
||
return err
|
||
} else {
|
||
if len(signal) > 0 {
|
||
glog.Printf("%d: server restarting by signal: %s", gproc.Pid(), signal)
|
||
} else {
|
||
glog.Printf("%d: server restarting by web admin", gproc.Pid())
|
||
}
|
||
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 关闭所有Web Server
|
||
func shutdownWebServers(signal ...string) {
|
||
serverProcessStatus.Set(gADMIN_ACTION_SHUTINGDOWN)
|
||
if len(signal) > 0 {
|
||
glog.Printf("%d: server shutting down by signal: %s", gproc.Pid(), signal[0])
|
||
// 在终端信号下，立即执行关闭操作
|
||
forceCloseWebServers()
|
||
allDoneChan <- struct{}{}
|
||
} else {
|
||
glog.Printf("%d: server shutting down by api", gproc.Pid())
|
||
// 非终端信号下，异步1秒后再执行关闭，
|
||
// 目的是让接口能够正确返回结果，否则接口会报错(因为web server关闭了)
|
||
gtimer.SetTimeout(time.Second, func() {
|
||
forceCloseWebServers()
|
||
allDoneChan <- struct{}{}
|
||
})
|
||
}
|
||
}
|
||
|
||
// 关优雅闭进程所有端口的Web Server服务
|
||
// 注意，只是关闭Web Server服务，并不是退出进程
|
||
func gracefulShutdownWebServers() {
|
||
serverMapping.RLockFunc(func(m map[string]interface{}) {
|
||
for _, v := range m {
|
||
for _, s := range v.(*Server).servers {
|
||
s.shutdown()
|
||
}
|
||
}
|
||
})
|
||
}
|
||
|
||
// 强制关闭进程所有端口的Web Server服务
|
||
// 注意，只是关闭Web Server服务，并不是退出进程
|
||
func forceCloseWebServers() {
|
||
serverMapping.RLockFunc(func(m map[string]interface{}) {
|
||
for _, v := range m {
|
||
for _, s := range v.(*Server).servers {
|
||
s.close()
|
||
}
|
||
}
|
||
})
|
||
}
|
||
|
||
// 异步监听进程间消息
|
||
func handleProcessMessage() {
|
||
for {
|
||
if msg := gproc.Receive(gADMIN_GPROC_COMM_GROUP); msg != nil {
|
||
if bytes.EqualFold(msg.Data, []byte("exit")) {
|
||
gracefulShutdownWebServers()
|
||
allDoneChan <- struct{}{}
|
||
return
|
||
}
|
||
}
|
||
}
|
||
}
|