golang http請求超時-Golang-PHP中文網

golang http請求超時

PHPz

發布： 2023-05-10 18:03:37

原創

1420 人瀏覽過

在 Golang 編寫 HTTP 應用程式時，由於網路環境的不穩定性，我們可能會遇到 HTTP 請求逾時的問題。

本文將介紹如何在 Golang 中設定 HTTP 請求逾時時間，以防止阻塞和無限等待。

什麼是 HTTP 請求逾時

HTTP 請求超時是指客戶端向伺服器發送請求後等待回應的時間超過了預定時間，導致請求被取消。這種情況一般發生在網路延遲、負載高、伺服器資源不足等情況。

對於 Golang 中的 HTTP 應用程序，當出現無回應或回應時間過長的請求時，為了防止應用程式阻塞和無限等待，我們需要設定 HTTP 請求逾時時間。

為什麼需要設定超時時間

在 Golang 應用程式中，每個 HTTP 請求都是一個 goroutine，它將在網路中發送請求並等待回應。如果沒有設定請求逾時時間，就可能會導致程式在等待回應時阻塞並進入無限循環，增加伺服器負擔和降低效能。

此外，如果應用程式中存在惡意使用者或攻擊者，他們可能會發送虛假請求以使應用程式長時間等待回應，從而耗盡伺服器資源。

因此，設定 HTTP 請求逾時時間可以確保接收到回應之前不會浪費太多時間和資源，並防止應用程式被攻擊耗盡資源。

在 Golang 中設定 HTTP 請求逾時時間

在 Golang 中，我們可以使用 net/http 函式庫來傳送 HTTP 請求並設定逾時時間。在該庫中定義的 Client 類型具有一個 Timeout 屬性，該屬性可以設定逾時時間。

以下是範例程式碼：

package main

import (
    "fmt"
    "net/http"
    "time"
)

func main() {
    client := http.Client{
        Timeout: time.Second * 5, // 设置超时时间为5秒
    }

    req, err := http.NewRequest(http.MethodGet, "https://www.example.com", nil)
    if err != nil {
        panic(err)
    }

    res, err := client.Do(req)
    if err != nil {
        panic(err)
    }

    fmt.Println(res)
}

登入後複製

在此範例程式碼中，我們建立了一個 HTTP 用戶端，設定逾時時間為5秒。然後，我們發送一個 GET 請求到 https://www.example.com，並等待回應。如果請求逾時或回應時間過長，則會在5秒內結束。

如何處理逾時錯誤

在上面的範例程式碼中，我們使用 panic 函數處理了 HTTP 請求逾時錯誤。但是，在實際應用程式中，應該使用更可靠的方法來處理逾時錯誤。

以下是一個更可靠的方法：在請求中新增一個Context，並使用select 語句處理逾時和正常回應：

package main

import (
    "context"
    "fmt"
    "net/http"
    "time"
)

func main() {
    client := http.Client{
        Timeout: time.Second * 5, // 设置超时时间为5秒
    }

    req, err := http.NewRequest(http.MethodGet, "https://www.example.com", nil)
    if err != nil {
        panic(err)
    }

    ctx, cancel := context.WithTimeout(req.Context(), time.Second*5)
    defer cancel() // 在函数退出时取消Context

    req = req.WithContext(ctx) // 使用带有超时的Context创建新请求

    resp, err := client.Do(req)
    if err != nil {
        select {
        case <-ctx.Done(): // 超时错误
            fmt.Println("Request timed out")
        default: // 其他错误
            panic(err)
        }
    }

    defer resp.Body.Close()
    fmt.Println(resp.StatusCode)
}

登入後複製

在此範例程式碼中，我們建立了一個Context，並使用了WithTimeout 函數來設定超時時間。然後，我們使用帶有逾時的 Context 建立了一個新的請求，並發送此請求。

如果請求逾時或出現其他錯誤，根據不同的錯誤類型採取不同的處理方法。這使得我們可以更好地控制應用程式並優雅地處理錯誤情況。