gRPC Golang Python优雅中断服务

Golang

	// 将服务启动放入协程中，当接收到终止信号后，主进程销毁，协称也会随着销毁
	go func() {
		zap.S().Infof("%s 服务启动！ http://%s:%d", global.Config.Name, global.Config.Host, global.Config.Port)
		if err := r.Run(fmt.Sprintf(":%d", global.Config.Port)); err != nil {
			zap.S().Panicf("服务启动失败, port:%d,err:%s", global.Config.Port, err.Error())
		}
	}()

	// 注册服务
	registerClient := consul.NewRegistryClient(global.Config.Consul.Host, global.Config.Consul.Port)
	sc := global.Config
	serviceId := fmt.Sprintf("%s", uuid.NewV4())
	err := registerClient.Register(sc.Host, sc.Port, sc.Name, serviceId, sc.Consul.Tags)
	if err != nil {
		zap.S().Fatalw("服务注册失败", err.Error())
	}
	zap.S().Infow("服务注册成功")

	// 优雅的退出程序
	quit := make(chan os.Signal)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit
	zap.S().Infow("服务注销中...")
	// 等待信号，如果接受到信号执行以下内容
	err = registerClient.DeRegister(serviceId)
	if err != nil {
		zap.S().Fatalw("服务注销失败", err.Error())
	}
	zap.S().Infow("服务注销成功")

Python

import sys
import os

import grpc
import logging
import signal
from concurrent import futures
from loguru import logger

# 因为在终端上运行会找不到根目录所以需要指明项目根目录
BASE_PATH = os.path.dirname(os.path.abspath(os.path.dirname(__file__)))
sys.path.insert(0, BASE_PATH)

from user_srv.proto import user_pb2, user_pb2_grpc
from user_srv.handler.user import UserService


# 处理中断函数
def on_exit(signal, frame):
    logger.info("grpc server exit")
   
   	# 服务注销逻辑.......
    
    sys.exit(0)


def serve():
    logger.add("logs/user_srv_{time}.log")
    # 多线程启动
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    # 注册服务
    user_pb2_grpc.add_UserServicer_to_server(UserService(), server)
    # 端口
    server.add_insecure_port("[::]:50051")
    # 监听退出信号
    """
        windows下支持的信号是有限的：
            SIGINT ctrl+C终端
            SIGTERM kill发出的软件终止
    """
    signal.signal(signal.SIGINT, on_exit)
    signal.signal(signal.SIGTERM, on_exit)

    logger.info(f"启动服务：http://127.0.0.1:50051")
    server.start()
    server.wait_for_termination()


if __name__ == "__main__":
    logging.basicConfig()
    serve()