chore: update restful/grpc servers shutdown stages (#3662)
This commit is contained in:
@@ -76,7 +76,7 @@ func (s *rpcServer) Start(register RegisterFn) error {
|
||||
|
||||
// we need to make sure all others are wrapped up,
|
||||
// so we do graceful stop at shutdown phase instead of wrap up phase
|
||||
waitForCalled := proc.AddWrapUpListener(func() {
|
||||
waitForCalled := proc.AddShutdownListener(func() {
|
||||
if s.health != nil {
|
||||
s.health.Shutdown()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user