前言
都说go标准库实用，Api设计简洁。这次就用go 标准库中的net/http包实现一个简洁的http web服务器，包括三种版本。
v1最简单版
直接使用http.HandleFunc(partern,function(http.ResponseWriter,
*http.Request){})
HandleFunc接受两个参数，第一个为路由地址，第二个为处理方法。
//v1
func main() {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("httpserver v1"))
})
http.HandleFunc("/bye", sayBye)
log.Println("Starting v1 server ...")
log.Fatal(http.ListenAndServe(":1210", nil))
}
func sayBye(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("bye bye ,this is v1 httpServer"))
}
v2自定义Handler
查看标准库源码，v1版本实际上是调用了handle方法,传入的HandlerFunc实现了Handler的ServeHTTP方法，实际上是ServeHTTP在做http请求处理。
HandleFunc调用.png
HandleFunc实现Handler.png
Handler接口定义.png
由此我们可以自定义自己的Handler，v2版本代码如下:
// v2
func main() {
mux := http.NewServeMux()
mux.Handle("/", &myHandler{})
mux.HandleFunc("/bye", sayBye)
log.Println("Starting v2 httpserver")
log.Fatal(http.ListenAndServe(":1210", mux))
}
type myHandler struct{}
func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("this is version 2"))
}
func sayBye(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("bye bye ,this is v2 httpServer"))
}
v3自定义server配置
前面对Handler开了一次刀，下面我们看看http.ListenAndServe()中有些什么秘密。
ListenAndServe.png
原来这里可以自定义http服务器配置，都在Server这个结构体中,这个对象能配置监听地址端口，配置读写超时时间，配置handler,配置请求头最大字节数...，所有稍微改造一下v2的程序得到v3版:
// v3
func main() {
mux := http.NewServeMux()
mux.Handle("/", &myHandler{})
mux.HandleFunc("/bye", sayBye)
server := &http.Server{
Addr: ":1210",
WriteTimeout: time.Second * 3, //设置3秒的写超时
Handler: mux,
}
log.Println("Starting v3 httpserver")
log.Fatal(server.ListenAndServe())
}
type myHandler struct{}
func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("this is version 3"))
}
func sayBye(w http.ResponseWriter, r *http.Request) {
// 睡眠4秒 上面配置了3秒写超时，所以访问 “/bye“路由会出现没有响应的现象
time.Sleep(4 * time.Second)
w.Write([]byte("bye bye ,this is v3 httpServer"))
}
拓展一下(如何平滑关闭http服务)
在go1.8中新增了一个新特性，利用Shutdown(ctx context.Context) 优雅地关闭http服务。
文档中描述:
Shutdown 将无中断的关闭正在活跃的连接，然后平滑的停止服务。处理流程如下：
- 首先关闭所有的监听;
- 然后关闭所有的空闲连接;
- 然后无限期等待连接处理完毕转为空闲，并关闭;
- 如果提供了 带有超时的Context，将在服务关闭前返回 Context的超时错误;
利用这个特性改造一下v3版本的程序，实现一个关闭http的提示
// 主动关闭服务器
var server *http.Server
func main() {
// 一个通知退出的chan
quit := make(chan os.Signal)
signal.Notify(quit, os.Interrupt)
mux := http.NewServeMux()
mux.Handle("/", &myHandler{})
mux.HandleFunc("/bye", sayBye)
server = &http.Server{
Addr: ":1210",
WriteTimeout: time.Second * 4,
Handler: mux,
}
go func() {
// 接收退出信号
<-quit
if err := server.Close(); err != nil {
log.Fatal("Close server:", err)
}
}()
log.Println("Starting v3 httpserver")
err := server.ListenAndServe()
if err != nil {
// 正常退出
if err == http.ErrServerClosed {
log.Fatal("Server closed under request")
} else {
log.Fatal("Server closed unexpected", err)
}
}
log.Fatal("Server exited")
}
type myHandler struct{}
func (*myHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("this is version 3"))
}
// 关闭http
func sayBye(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("bye bye ,shutdown the server")) // 没有输出
err := server.Shutdown(nil)
if err != nil {
log.([]byte("shutdown the server err"))
}
}
尝试访问http://localhost:1210/bye 在控制台会得到以下提示结果，平滑关闭http服务成功:
成功平滑关闭.png