支持GRPC注册服务

This commit is contained in:
kercylan98
2023-05-10 16:17:52 +08:00
parent 02e4cad427
commit a0a38793aa
4 changed files with 12 additions and 2 deletions
+8 -1
View File
@@ -296,7 +296,7 @@ func (slf *Server) Shutdown(err error) {
slf.initMessageChannel = false
}
if slf.grpcServer != nil {
slf.grpcServer.Stop()
slf.grpcServer.GracefulStop()
}
if slf.httpServer != nil {
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
@@ -315,6 +315,13 @@ func (slf *Server) Shutdown(err error) {
}
}
func (slf *Server) GRPCServer() *grpc.Server {
if slf.grpcServer == nil {
panic(ErrNetworkOnlySupportGRPC)
}
return slf.grpcServer
}
// HttpRouter 当网络类型为 NetworkHttp 时将被允许获取路由器进行路由注册,否则将会发生 panic
func (slf *Server) HttpRouter() gin.IRouter {
if slf.ginServer == nil {