milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix external grpc server doesn't graceful stop (#28309) 

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
pull/28327/head
wei liu 2023-11-10 00:14:18 +08:00 committed by GitHub
parent 14c8a90517
commit 8a370f115a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
internal/distributed/proxy/service.go
View File

@ -650,18 +650,25 @@ func (s *Server) Stop() error {
gracefulWg.Add(1)
go func() {
defer gracefulWg.Done()
if s.grpcInternalServer != nil {
utils.GracefulStopGRPCServer(s.grpcInternalServer)
}
if s.tcpServer != nil {
log.Info("Graceful stop Proxy tcp server...")
log.Info("Proxy stop tcp server...")
s.tcpServer.Close()
} else if s.grpcExternalServer != nil {
}
if s.grpcExternalServer != nil {
log.Info("Proxy stop external grpc server")
utils.GracefulStopGRPCServer(s.grpcExternalServer)
if s.httpServer != nil {
log.Info("Graceful stop grpc http server...")
s.httpServer.Close()
}
}
if s.httpServer != nil {
log.Info("Proxy stop http server...")
s.httpServer.Close()
}
if s.grpcInternalServer != nil {
log.Info("Proxy stop internal grpc server")
utils.GracefulStopGRPCServer(s.grpcInternalServer)
}
}()
gracefulWg.Wait()