mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
33 lines
681 B
Go
33 lines
681 B
Go
package utils
|
|
|
|
import (
|
|
"time"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/milvus-io/milvus/pkg/log"
|
|
"github.com/milvus-io/milvus/pkg/util/paramtable"
|
|
)
|
|
|
|
func GracefulStopGRPCServer(s *grpc.Server) {
|
|
if s == nil {
|
|
return
|
|
}
|
|
ch := make(chan struct{})
|
|
go func() {
|
|
defer close(ch)
|
|
log.Debug("try to graceful stop grpc server...")
|
|
// will block until all rpc finished.
|
|
s.GracefulStop()
|
|
}()
|
|
select {
|
|
case <-ch:
|
|
case <-time.After(paramtable.Get().ProxyGrpcServerCfg.GracefulStopTimeout.GetAsDuration(time.Second)):
|
|
// took too long, manually close grpc server
|
|
log.Debug("stop grpc server...")
|
|
s.Stop()
|
|
// concurrent GracefulStop should be interrupted
|
|
<-ch
|
|
}
|
|
}
|