golang 讀取請求body

在使用Golang進行網路開發時，有時候需要讀取HTTP請求的body部分（例如POST請求中所攜帶的提交資料），以便進行處理。本文將介紹Golang如何讀取HTTP請求的body。

一、使用ioutil.ReadAll()讀取

Golang標準庫中的ioutil套件提供了一個ReadAll()函數，可以讀取任何io.Reader類型的物件中的所有數據，並傳回[]byte類型的數據。因此，我們可以透過將http.Request.Body欄位作為參數傳遞給ReadAll()函數來讀取HTTP請求的body部分。範例程式碼如下：

package main

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

func handler(w http.ResponseWriter, r *http.Request) {
    body, err := ioutil.ReadAll(r.Body)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer r.Body.Close()

    // 处理body数据
    fmt.Fprint(w, string(body))
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

在上述範例程式碼中，我們定義了一個HTTP請求處理函數handler()，在該函數中讀取了HTTP請求的body部分，並將其轉換為string類型後輸出到HTTP響應中。

要注意的是，在使用ioutil.ReadAll()函數讀取HTTP請求的body時，應該在函數返回後關閉http.Request.Body欄位以釋放資源。這裡使用了defer語句確保在函數執行完畢後關閉該欄位。

二、使用json.Decoder讀取JSON資料

如果HTTP請求的body部分是JSON格式的數據，我們可以使用Golang中的json.Decoder類型來直接讀取JSON數據並解析為Golang的資料結構。範例程式碼如下：

package main

import (
    "encoding/json"
    "fmt"
    "net/http"
)

type User struct {
    Username string `json:"username"`
    Password string `json:"password"`
}

func handler(w http.ResponseWriter, r *http.Request) {
    var user User
    err := json.NewDecoder(r.Body).Decode(&user)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer r.Body.Close()

    // 处理user对象
    fmt.Fprintf(w, "Username: %s, Password: %s", user.Username, user.Password)
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

在上述範例程式碼中，我們定義了一個名為User的結構體類型，並在HTTP請求處理函數中使用json.NewDecoder()函數建立了一個json.Decoder類型的對象，然後使用其Decode()方法直接將HTTP請求的body部分解析為User類型的數據，並進行處理。

要注意的是，在使用json.Decoder讀取HTTP請求的body時，應該在函數返回前關閉http.Request.Body欄位以釋放資源。並且，需要同時傳遞一個指向目標結構體類型的指標作為Decode()方法的參數（例如&user），以便將解析得到的JSON資料填入該結構體物件中。

三、使用xml.Decoder讀取XML數據

如果HTTP請求的body部分是XML格式的數據，我們可以類似地使用Golang中的xml.Decoder類型來直接讀取XML資料並解析為Golang的資料結構。範例程式碼如下：

package main

import (
    "encoding/xml"
    "fmt"
    "net/http"
)

type User struct {
    XMLName  xml.Name `xml:"user"`
    Username string   `xml:"username"`
    Password string   `xml:"password"`
}

func handler(w http.ResponseWriter, r *http.Request) {
    var user User
    err := xml.NewDecoder(r.Body).Decode(&user)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer r.Body.Close()

    // 处理user对象
    fmt.Fprintf(w, "Username: %s, Password: %s", user.Username, user.Password)
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

在上述範例程式碼中，我們定義了一個名為User的結構體類型，並在HTTP請求處理函數中使用xml.NewDecoder()函數建立了一個xml.Decoder類型的對象，然後使用其Decode()方法直接將HTTP請求的body部分解析為User類型的數據，並進行處理。

要注意的是，在使用xml.Decoder讀取HTTP請求的body時，應該在函數返回前關閉http.Request.Body欄位以釋放資源。並且，需要為目標結構體類型中的每個欄位指定對應XML節點的名稱（例如Username欄位對應的XML節點為username）。

總結

本文介紹了Golang如何讀取HTTP請求的body。透過使用標準函式庫中的ioutil套件、json.Decoder類型和xml.Decoder類型，我們可以輕鬆地將HTTP請求的body部分讀取並解析為Golang的資料結構，以便進行處理。

以上是golang 讀取請求body的詳細內容。更多資訊請關注PHP中文網其他相關文章！