支持GRPC注册服务
This commit is contained in:
+8
-1
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user