golang-grpc如何实现平滑重启

这篇文章给大家分享的是有关golang-grpc如何实现平滑重启的内容。小编觉得挺实用的，因此分享给大家做个参考，一起跟随小编过来看看吧。

package gooimport (        "fmt"        "github.com/facebookgo/grace/gracenet"        "google.golang.org/grpc"        "io/ioutil"        "log"        "os"        "os/signal"        "syscall")type GRPCGraceful struct {        nett string        addr string        s    *grpc.Server        net  *gracenet.Net}func NewGRPCGraceful(nett, addr string, s *grpc.Server) *GRPCGraceful {        return &GRPCGraceful{                nett: nett,                addr: addr,                s:    s,                net:  &gracenet.Net{},        }}func (g *GRPCGraceful) Serve() error {        lis, err := g.net.Listen(g.nett, g.addr)        if err != nil {                return err        }        errs := make(chan error)        // 启动serve        AsyncFunc(func() {                errs <- g.s.Serve(lis)        })        // 判断并关闭旧进程        AsyncFunc(g.killPPID)        // 存储pid        AsyncFunc(g.storePID)        // 监听信号        quit := g.handleSignal(errs)        // 监听退出信号，错误信息        select {        case err := <-errs:                return err        case <-quit:                return nil        }}// 监听信号func (g *GRPCGraceful) handleSignal(errs chan error) <-chan struct{} {        // 通道，是否退出        quit := make(chan struct{})        AsyncFunc(func() {                ch := make(chan os.Signal)                signal.Notify(ch, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT, syscall.SIGUSR1, syscall.SIGUSR2)                for sig := range ch {                        switch sig {                        // 监听退出                        case syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT:                                signal.Stop(ch)                                g.s.GracefulStop()                                close(quit)                                return                        // 监听重启                        case syscall.SIGUSR1, syscall.SIGUSR2:                                if _, err := g.net.StartProcess(); err != nil {                                        errs <- err                                }                        }                }        })        return quit}// 记录进程号到.pid文件func (g *GRPCGraceful) storePID() {        pid := fmt.Sprintf("%d", os.Getpid())        ioutil.WriteFile(".pid", []byte(pid), 0644)        log.Println(fmt.Sprintf("server is running, address=%s, pid=%s", g.addr, pid))}// 判断进程是否继承进程，平滑重启时，关闭旧进程func (g *GRPCGraceful) killPPID() {        inherit := os.Getenv("LISTEN_FDS") != ""        if !inherit {                return        }        ppid := os.Getppid()        if ppid == 1 {                return        }        syscall.Kill(ppid, syscall.SIGTERM)}

感谢各位的阅读！关于"golang-grpc如何实现平滑重启"这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，让大家可以学到更多知识，如果觉得文章不错，可以把它分享出去让更多的人看到吧！