在 go 语言里启动一个 http 服务非常简单，只需要一行代码http.ListenAndServe()就可以搞定，这个方法会一直阻塞着直到进程关闭，如果这个时候来了些特殊的需求比如：
- 监听服务启动
- 手动关闭服务
- 监听服务关闭
在 go 中应该怎么实现呢？下面来一一举例。
监听服务启动
方法一（推荐）
将Listen步骤拆分出来，先监听端口，再绑定到server上，代码示例：
l, _ := net.Listen("tcp", ":8080")
// 服务启动成功，进行初始化
doInit()
// 绑定到server上
http.Serve(l, nil)
方法二
通过一个协程去轮询监听服务启动状态，代码示例：
go func() {
for {
if _, err := net.Dial("tcp", "127.0.0.1:8080"); err == nil {
// 服务启动成功，进行初始化
doInit()
//退出协程
break
}
// 每隔一秒检查一次服务是否启动成功
time.Sleep(time.Second)
}
}()
http.ListenAndServe(":8080", nil)
手动关闭服务
优雅关闭（推荐）
在http包中并没有暴露服务的关闭方法，通过http.ListenAndServe()方法启动的 http 服务默认帮我们创建了一个*http.Server对象，源码如下：
func ListenAndServe(addr string, handler Handler) error {
server := &Server{Addr: addr, Handler: handler}
return server.ListenAndServe()
}
实际上在*http.Server中是有提供Shutdown方法的，所以我们只需要手动构造一个*http.Server对象，就可以进行优雅关闭了，代码示例：
srv := &http.Server{Addr: ":8080"}
go func(){
// 10秒之后关闭服务
time.Sleep(time.Second * 10)
srv.Shutdown(context.TODO())
}()
// 启动服务
srv.ListenAndServe()
强制关闭
强制关闭和上面步骤是一样的，只是调用的方法换成了srv.Close()，这会导致所有的请求立即中断，所以需要特别注意。
监听服务关闭
当我们手动将服务关闭之后，srv.ListenAndServe()方法就会立即返回，这里需要注意的是该方法会返回一个error，当然这个error是一个特殊的 error http.ErrServerClosed，帮助我们区分是否为正常的服务关闭，所以需要对它特殊处理下，代码示例：
if err := server.ListenAndServe(); err != nil {
// 服务关闭，进行处理
doShutdown()
if err != http.ErrServerClosed{
// 异常宕机，打印错误信息
log.Fatal(err)
}
}
参考资料
本文首发于我的博客：https://monkeywie.cn，欢迎收藏！不定期分享
JAVA、Golang、前端、docker、k8s等干货知识。
网友评论