快捷導(dǎo)航

Go語言中HTTP客戶端超時與重試的實現(xiàn)代碼

更新時間：2026年04月08日 10:27:29 作者：王碼碼2035哦

Golang標準庫雖然提供了基礎(chǔ)的HTTP客戶端實現(xiàn),但在高并發(fā)、高可用場景下,我們需要更精細化的策略來應(yīng)對復(fù)雜的網(wǎng)絡(luò)環(huán)境,這篇文章主要介紹了Go語言中HTTP客戶端超時與重試實現(xiàn)的相關(guān)資料,需要的朋友可以參考下

1. HTTP客戶端的基本概念

HTTP客戶端是Go語言中進行網(wǎng)絡(luò)請求的重要工具，標準庫net/http提供了強大的HTTP客戶端功能。在實際應(yīng)用中，合理設(shè)置超時和實現(xiàn)重試機制是保證服務(wù)穩(wěn)定性和可靠性的關(guān)鍵。

本文將詳細介紹Go語言中的HTTP客戶端，重點講解超時設(shè)置和重試機制的實現(xiàn)，幫助開發(fā)者構(gòu)建健壯的HTTP客戶端。

2. 基礎(chǔ)HTTP客戶端

2.1 基本用法

package main
import (
	"fmt"
	"io"
	"net/http"
)
func main() {
	resp, err := http.Get("https://api.example.com/data")
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer resp.Body.Close()
	body, _ := io.ReadAll(resp.Body)
	fmt.Println(string(body))
}

2.2 自定義客戶端

package main
import (
	"fmt"
	"net/http"
	"time"
)
func main() {
	client := &http.Client{
		Timeout: 10 * time.Second,
	}
	resp, err := client.Get("https://api.example.com/data")
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer resp.Body.Close()
	fmt.Println("Status:", resp.Status)
}

3. 超時設(shè)置

3.1 連接超時

package main
import (
	"fmt"
	"net"
	"net/http"
	"time"
)
func main() {
	client := &http.Client{
		Transport: &http.Transport{
			DialContext: (&net.Dialer{
				Timeout:   5 * time.Second,
				KeepAlive: 30 * time.Second,
			}).DialContext,
		},
		Timeout: 10 * time.Second,
	}
	resp, err := client.Get("https://api.example.com/data")
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer resp.Body.Close()
	fmt.Println("Success")
}

3.2 完整的超時配置

package main
import (
	"fmt"
	"net"
	"net/http"
	"time"
)
func createHTTPClient() *http.Client {
	return &http.Client{
		Transport: &http.Transport{
			DialContext: (&net.Dialer{
				Timeout:   5 * time.Second,
				KeepAlive: 30 * time.Second,
			}).DialContext,
			TLSHandshakeTimeout:   5 * time.Second,
			ResponseHeaderTimeout: 10 * time.Second,
			ExpectContinueTimeout: 1 * time.Second,
			IdleConnTimeout:       90 * time.Second,
			MaxIdleConns:          100,
			MaxIdleConnsPerHost:   10,
		},
		Timeout: 30 * time.Second,
	}
}
func main() {
	client := createHTTPClient()
	resp, err := client.Get("https://api.example.com/data")
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer resp.Body.Close()
	fmt.Println("Success")
}

4. 重試機制

4.1 簡單重試

package main
import (
	"fmt"
	"net/http"
	"time"
)
func retryRequest(url string, maxRetries int) (*http.Response, error) {
	var resp *http.Response
	var err error
	for i := 0; i < maxRetries; i++ {
		resp, err = http.Get(url)
		if err == nil && resp.StatusCode == http.StatusOK {
			return resp, nil
		}
		if resp != nil {
			resp.Body.Close()
		}
		time.Sleep(time.Second * time.Duration(i+1))
	}
	return nil, fmt.Errorf("failed after %d retries: %v", maxRetries, err)
}
func main() {
	resp, err := retryRequest("https://api.example.com/data", 3)
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer resp.Body.Close()
	fmt.Println("Success")
}

4.2 指數(shù)退避重試

package main
import (
	"fmt"
	"math"
	"net/http"
	"time"
)
func exponentialBackoffRetry(url string, maxRetries int) (*http.Response, error) {
	client := &http.Client{
		Timeout: 10 * time.Second,
	}
	var resp *http.Response
	var err error
	for i := 0; i < maxRetries; i++ {
		resp, err = client.Get(url)
		if err == nil && resp.StatusCode == http.StatusOK {
			return resp, nil
		}
		if resp != nil {
			resp.Body.Close()
		}
		// 指數(shù)退避
		backoff := time.Duration(math.Pow(2, float64(i))) * time.Second
		time.Sleep(backoff)
	}
	return nil, fmt.Errorf("failed after %d retries: %v", maxRetries, err)
}
func main() {
	resp, err := exponentialBackoffRetry("https://api.example.com/data", 3)
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	defer resp.Body.Close()
	fmt.Println("Success")
}

5. 完整的HTTP客戶端示例

package main
import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net"
	"net/http"
	"time"
)
type HTTPClient struct {
	client     *http.Client
	maxRetries int
}
func NewHTTPClient() *HTTPClient {
	return &HTTPClient{
		client: &http.Client{
			Transport: &http.Transport{
				DialContext: (&net.Dialer{
					Timeout:   5 * time.Second,
					KeepAlive: 30 * time.Second,
				}).DialContext,
				TLSHandshakeTimeout:   5 * time.Second,
				ResponseHeaderTimeout: 10 * time.Second,
				IdleConnTimeout:       90 * time.Second,
				MaxIdleConns:          100,
				MaxIdleConnsPerHost:   10,
			},
			Timeout: 30 * time.Second,
		},
		maxRetries: 3,
	}
}
func (c *HTTPClient) Get(ctx context.Context, url string) ([]byte, error) {
	return c.doRequest(ctx, http.MethodGet, url, nil)
}
func (c *HTTPClient) Post(ctx context.Context, url string, body interface{}) ([]byte, error) {
	jsonBody, _ := json.Marshal(body)
	return c.doRequest(ctx, http.MethodPost, url, bytes.NewReader(jsonBody))
}
func (c *HTTPClient) doRequest(ctx context.Context, method, url string, body io.Reader) ([]byte, error) {
	var lastErr error
	for i := 0; i < c.maxRetries; i++ {
		req, err := http.NewRequestWithContext(ctx, method, url, body)
		if err != nil {
			return nil, err
		}
		req.Header.Set("Content-Type", "application/json")
		resp, err := c.client.Do(req)
		if err != nil {
			lastErr = err
			time.Sleep(time.Second * time.Duration(i+1))
			continue
		}
		defer resp.Body.Close()
		if resp.StatusCode == http.StatusOK {
			return io.ReadAll(resp.Body)
		}
		lastErr = fmt.Errorf("unexpected status code: %d", resp.StatusCode)
		time.Sleep(time.Second * time.Duration(i+1))
	}
	return nil, fmt.Errorf("failed after %d retries: %v", c.maxRetries, lastErr)
}
func main() {
	client := NewHTTPClient()
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()
	data, err := client.Get(ctx, "https://api.example.com/data")
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	fmt.Println(string(data))
}