select实现超时保护机制

本文主要是介绍select实现超时保护机制，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1、使用channel优雅地关闭服务

package mainimport ("context""fmt""net/http""os""os/signal""syscall""time"
)func IndexHandler(w http.ResponseWriter, r *http.Request) {if r.Method != http.MethodGet {return}_, _ = fmt.Fprintf(w, "测试", "")
}func initRouters() {http.HandleFunc("/", IndexHandler)
}func main() {initRouters()srv := http.Server{Addr: ":8081",}go func() {err := srv.ListenAndServe()if err != nil {return}}()// 优雅地关闭go服务quit := make(chan os.Signal)signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)<-quit // 阻塞// 定时关闭ctx, cancle := context.WithTimeout(context.Background(), 2*time.Second)defer cancle()if err := srv.Shutdown(ctx); err != nil {fmt.Println("Shutdown err:", err)}fmt.Println("Shutdown")
}

2、使用channel实现超时保护机制

package mainimport ("fmt""time"
)// select 可以优雅的处理超时
// 我限制我这个程序运行不可以超过1秒
func timeouting() {timeout := time.After(1 * time.Second) // 如果其它程序运行时间超过1s，那么出发保护机制 <-timeout 的操作ch := make(chan bool)go func() {time.Sleep(time.Second * 2)ch <- true}()select {case <-ch:fmt.Println("程序在1秒内启动")case <-timeout:fmt.Println("程序启动超时，请重新启动")}
}func main() {timeouting()
}

这篇关于select实现超时保护机制的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---