如何在grpc中侦听优美的服务器终止

Question

I want to listen for graceful server termination in grpc in my handler. When the server is being stopped gracefully, I want to add some logic in my code to close open ports, files, flush results etc. How can I do that?

How is it different in case of unary and streaming handler?

Answer 1

You can have shutdown hook by listening to the signals something like this

In your main function or where you start your server create channel for signals that you want to listen to

c := make(chan os.Signal, 1)
signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
// call your cleanup method with this channel as a routine
go cleanup(c)

In your clean up method

func cleanup(c chan os.Signal) {
   // Wait for termination
   <- c

  // Do your cleanups here
}

Create the signal channel and call the cleanup function as a go routine before you start the gRPC server. Whenever the application (gRPC server) stopped or interrupted this channel will get the signal in the cleanup function where you can do the necessary cleanups