大橙子网站建设,新征程启航
为企业提供网站建设、域名注册、服务器等服务
这篇文章给大家分享的是有关golang-grpc如何实现平滑重启的内容。小编觉得挺实用的,因此分享给大家做个参考,一起跟随小编过来看看吧。
10年积累的成都网站设计、做网站经验,可以快速应对客户对网站的新想法和需求。提供各种问题对应的解决方案。让选择我们的客户得到更好、更有力的网络服务。我虽然不认识你,你也不认识我。但先网站制作后付款的网站建设流程,更有成安免费网站建设让你可以放心的选择与我们合作。
package goo import ( "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如何实现平滑重启”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,让大家可以学到更多知识,如果觉得文章不错,可以把它分享出去让更多的人看到吧!