詳解Golang開啟http服務的三種方式

更新時間：2020年06月17日 14:23:39 作者：L千年老妖

這篇文章主要介紹了詳解Golang開啟http服務的三種方式，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前言

都說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版本實際上是調用了handle方法,傳入的HandlerFunc實現(xiàn)了Handler的ServeHTTP方法，實際上是ServeHTTP在做http請求處理。

HandleFunc調用.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ā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"))
}

拓展一下(如何平滑關閉http服務)

在go1.8中新增了一個新特性，利用Shutdown(ctx context.Context) 優(yōu)雅地關閉http服務。

文檔中描述:

Shutdown 將無中斷的關閉正在活躍的連接，然后平滑的停止服務。處理流程如下：

首先關閉所有的監(jiān)聽;
然后關閉所有的空閑連接;
然后無限期等待連接處理完畢轉為空閑，并關閉;
如果提供了帶有超時的Context，將在服務關閉前返回 Context的超時錯誤;

利用這個特性改造一下v3版本的程序，實現(xiàn)一個關閉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服務成功: