Go 語言基礎之 net-http

【導讀】本文介紹了 go 語言基礎庫 net/http。

Go 語言內置的net/http包十分的優秀，提供了 HTTP 客戶端和服務端的實現。

net/http 介紹

Go 語言內置的net/http包提供了 HTTP 客戶端和服務端的實現。

HTTP 協議

超文本傳輸協議（HTTP，HyperText Transfer Protocol) 是互聯網上應用最爲廣泛的一種網絡傳輸協議，所有的 WWW 文件都必須遵守這個標準。設計 HTTP 最初的目的是爲了提供一種發佈和接收 HTML 頁面的方法。

HTTP 客戶端

基本的 HTTP/HTTPS 請求

Get、Head、Post 和 PostForm 函數發出 HTTP/HTTPS 請求。

resp, err := http.Get("http://example.com/")
...
resp, err := http.Post("http://example.com/upload", "image/jpeg", &buf)
...
resp, err := http.PostForm("http://example.com/form",
    url.Values{"key": {"Value"}, "id": {"123"}})

程序在使用完 response 後必須關閉回覆的主體。

resp, err := http.Get("http://example.com/")
if err != nil {
    // handle error
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
// ...

GET 請求示例

使用net/http包編寫一個簡單的發送 HTTP 請求的 Client 端，代碼如下：

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
)

func main() {
    resp, err := http.Get("https://www.liwenzhou.com/")
    if err != nil {
        fmt.Printf("get failed, err:%v\n", err)
        return
    }
    defer resp.Body.Close()
    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Printf("read from resp.Body failed, err:%v\n", err)
        return
    }
    fmt.Print(string(body))
}

將上面的代碼保存之後編譯成可執行文件，執行之後就能在終端打印liwenzhou.com網站首頁的內容了，我們的瀏覽器其實就是一個發送和接收 HTTP 協議數據的客戶端，我們平時通過瀏覽器訪問網頁其實就是從網站的服務器接收 HTTP 數據，然後瀏覽器會按照 HTML、CSS 等規則將網頁渲染展示出來。

帶參數的 GET 請求示例

關於 GET 請求的參數需要使用 Go 語言內置的net/url這個標準庫來處理。

func main() {
    apiUrl := "http://127.0.0.1:9090/get"
    // URL param
    data := url.Values{}
    data.Set("name", "mi")
    data.Set("age", "18")
    u, err := url.ParseRequestURI(apiUrl)
    if err != nil {
        fmt.Printf("parse url requestUrl failed, err:%v\n", err)
    }
    u.RawQuery = data.Encode() // URL encode
    fmt.Println(u.String())
    resp, err := http.Get(u.String())
    if err != nil {
        fmt.Printf("post failed, err:%v\n", err)
        return
    }
    defer resp.Body.Close()
    b, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Printf("get resp failed, err:%v\n", err)
        return
    }
    fmt.Println(string(b))
}

對應的 Server 端 HandlerFunc 如下：

func getHandler(w http.ResponseWriter, r *http.Request) {
    defer r.Body.Close()
    data := r.URL.Query()
    fmt.Println(data.Get("name"))
    fmt.Println(data.Get("age"))
    answer := `{"status": "ok"}`
    w.Write([]byte(answer))
}

Post 請求示例

上面演示了使用net/http包發送GET請求的示例，發送POST請求的示例代碼如下：

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "strings"
)

// net/http post demo

func main() {
    url := "http://127.0.0.1:9090/post"
    // 表單數據
    //contentType := "application/x-www-form-urlencoded"
    //data := "
    // json
    contentType := "application/json"
    data := `{"name":"mi","age":18}`
    resp, err := http.Post(url, contentType, strings.NewReader(data))
    if err != nil {
        fmt.Printf("post failed, err:%v\n", err)
        return
    }
    defer resp.Body.Close()
    b, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Printf("get resp failed, err:%v\n", err)
        return
    }
    fmt.Println(string(b))
}

對應的 Server 端 HandlerFunc 如下：

func postHandler(w http.ResponseWriter, r *http.Request) {
    defer r.Body.Close()
    // 1. 請求類型是application/x-www-form-urlencoded時解析form數據
    r.ParseForm()
    fmt.Println(r.PostForm) // 打印form數據
    fmt.Println(r.PostForm.Get("name"), r.PostForm.Get("age"))
    // 2. 請求類型是application/json時從r.Body讀取數據
    b, err := ioutil.ReadAll(r.Body)
    if err != nil {
        fmt.Printf("read request.Body failed, err:%v\n", err)
        return
    }
    fmt.Println(string(b))
    answer := `{"status": "ok"}`
    w.Write([]byte(answer))
}

自定義 Client

要管理 HTTP 客戶端的頭域、重定向策略和其他設置，創建一個 Client：

client := &http.Client{
    CheckRedirect: redirectPolicyFunc,
}
resp, err := client.Get("http://example.com")
// ...
req, err := http.NewRequest("GET", "http://example.com", nil)
// ...
req.Header.Add("If-None-Match", `W/"wyzzy"`)
resp, err := client.Do(req)
// ...

自定義 Transport

要管理代理、TLS 配置、keep-alive、壓縮和其他設置，創建一個 Transport：

tr := &http.Transport{
    TLSClientConfig:    &tls.Config{RootCAs: pool},
    DisableCompression: true,
}
client := &http.Client{Transport: tr}
resp, err := client.Get("https://example.com")

Client 和 Transport 類型都可以安全的被多個 goroutine 同時使用。出於效率考慮，應該一次建立、儘量重用。

服務端

默認的 Server

ListenAndServe 使用指定的監聽地址和處理器啓動一個 HTTP 服務端。處理器參數通常是 nil，這表示採用包變量 DefaultServeMux 作爲處理器。

Handle 和 HandleFunc 函數可以向 DefaultServeMux 添加處理器。

http.Handle("/foo", fooHandler)
http.HandleFunc("/bar", func(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello, %q", html.EscapeString(r.URL.Path))
})
log.Fatal(http.ListenAndServe(":8080", nil))

默認的 Server 示例

使用 Go 語言中的net/http包來編寫一個簡單的接收 HTTP 請求的 Server 端示例，net/http包是對 net 包的進一步封裝，專門用來處理 HTTP 協議的數據。具體的代碼如下：

// http server
package main

import (
    "fmt"
    "net/http"
)

func sayHello(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "Hello shanghai！")
}

func main() {
    http.HandleFunc("/", sayHello)
    err := http.ListenAndServe(":9090", nil)
    if err != nil {
        fmt.Printf("http server failed, err:%v\n", err)
        return
    }
}

將上面的代碼編譯之後執行，打開你電腦上的瀏覽器在地址欄輸入127.0.0.1:9090回車，此時就能夠看到如下頁面了。

自定義 Server

要管理服務端的行爲，可以創建一個自定義的 Server：

s := &http.Server{
    Addr:           ":8080",
    Handler:        myHandler,
    ReadTimeout:    10 * time.Second,
    WriteTimeout:   10 * time.Second,
    MaxHeaderBytes: 1 << 20,
}
log.Fatal(s.ListenAndServe())

轉自：

jianshu.com/p/8bf41fef20c7

本文由 Readfog 進行 AMP 轉碼，版權歸原作者所有。
來源：https://mp.weixin.qq.com/s/K3BPfFwglCK8eW24o6wcag