前言
都說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
到此這篇關于詳解Golang開啟http服務的三種方式的文章就介紹到這了,更多相關Golang開啟http服務內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家!
您可能感興趣的文章:- 解決golang處理http response碰到的問題和需要注意的點
- 詳解golang中發送http請求的幾種常見情況
- 解決golang讀取http的body時遇到的坑
- golang HTTP 服務器 處理 日志/Stream流的操作