Golang

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
// 将服务启动放入协程中，当接收到终止信号后，主进程销毁，协称也会随着销毁
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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
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()