前言
都說go標準庫實用，Api設計簡潔。這次就用go 標準庫中的net/http包實現(xiàn)一個簡潔的http web服務器，包括三種版本。
v1最簡單版
直接使用http.HandleFunc(partern,function(http.ResponseWriter,*http.Request){})
HandleFunc接受兩個參數(shù)，第一個為路由地址，第二個為處理方法。
//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版本實際上是調(diào)用了handle方法,傳入的HandlerFunc實現(xiàn)了Handler的ServeHTTP方法，實際上是ServeHTTP在做http請求處理。
HandleFunc調(diào)用.png
HandleFunc實現(xiàn)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這個結(jié)構(gòu)體中,這個對象能配置監(jiān)聽地址端口，配置讀寫超時時間，配置handler,配置請求頭最大字節(jié)數(shù)...，所有稍微改造一下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“路由會出現(xiàn)沒有響應的現(xiàn)象
time.Sleep(4 * time.Second)
w.Write([]byte("bye bye ,this is v3 httpServer"))
}
拓展一下(如何平滑關(guān)閉http服務)
在go1.8中新增了一個新特性，利用Shutdown(ctx context.Context) 優(yōu)雅地關(guān)閉http服務。
文檔中描述:
Shutdown 將無中斷的關(guān)閉正在活躍的連接，然后平滑的停止服務。處理流程如下：
- 首先關(guān)閉所有的監(jiān)聽;
- 然后關(guān)閉所有的空閑連接;
- 然后無限期等待連接處理完畢轉(zhuǎn)為空閑，并關(guān)閉;
- 如果提供了 帶有超時的Context，將在服務關(guān)閉前返回 Context的超時錯誤;
利用這個特性改造一下v3版本的程序，實現(xiàn)一個關(guān)閉http的提示
// 主動關(guān)閉服務器
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"))
}
// 關(guān)閉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 在控制臺會得到以下提示結(jié)果，平滑關(guān)閉http服務成功:
成功平滑關(guān)閉.png
到此這篇關(guān)于詳解Golang開啟http服務的三種方式的文章就介紹到這了,更多相關(guān)Golang開啟http服務內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:- 解決golang處理http response碰到的問題和需要注意的點
- 詳解golang中發(fā)送http請求的幾種常見情況
- 解決golang讀取http的body時遇到的坑
- golang HTTP 服務器 處理 日志/Stream流的操作
