Go 标准库之 net.http 之 transport 自定义

database/sql 包提供了保证SQL或类SQL数据库的泛用接口，常用的数据库驱动有 github.com/go-sql-driver/mysql

https://pkg.go.dev/database/sql

过滤请求方法

我们可以使用 http.Transport 的RoundTrip方法来定制HTTP请求。通过定义一个实现了http.RoundTripper接口的结构体，并在其中重写RoundTrip方法，我们可以对请求进行过滤和修改。

type FilterTransport struct {
    Transport http.RoundTripper
}
 
func (t *FilterTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 过滤请求方法
    if req.Method == http.MethodGet {
        return nil, errors.New("Method Not Allowed")
    }
 
    return t.Transport.RoundTrip(req)
}
 
func main() {
    // 创建一个Transport
    transport := &FilterTransport{
        Transport: http.DefaultTransport,
    }
 
    // 创建一个Client
    client := &http.Client{
        Transport: transport,
    }
 
    // 创建一个GET请求
    req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
 
    // 发起请求
    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println("Response:", resp.Status)
    }
}

过滤请求头

除了过滤请求方法，我们还可以对请求头进行过滤和修改。

type HeaderFilterTransport struct {
    Transport http.RoundTripper
}
 
func (t *HeaderFilterTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 过滤请求头
    req.Header.Del("User-Agent")
    req.Header.Set("User-Agent", "Custom User-Agent")
 
    return t.Transport.RoundTrip(req)
}
 
func main() {
    // 创建一个Transport
    transport := &HeaderFilterTransport{
        Transport: http.DefaultTransport,
    }
 
    // 创建一个Client
    client := &http.Client{
        Transport: transport,
    }
 
    // 创建一个GET请求
    req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
 
    // 发起请求
    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println("Response:", resp.Status)
    }
}

请求拦截

除了过滤请求，我们还可以拦截请求，修改请求体或者对请求进行进一步处理。

type BodyInterceptorTransport struct {
    Transport http.RoundTripper
}
 
func (t *BodyInterceptorTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 拦截请求并修改请求体
    if req.Method == http.MethodPost {
        // 读取原始请求体
        body, err := ioutil.ReadAll(req.Body)
        if err != nil {
            return nil, err
        }
        req.Body.Close()
 
        // 修改请求体
        newBody := bytes.NewReader([]byte("Modified Body"))
 
        // 替换请求体
        req.Body = ioutil.NopCloser(newBody)
 
        // 设置Content-Type
        req.Header.Set("Content-Type", "text/plain")
    }
 
    return t.Transport.RoundTrip(req)
}
 
func main() {
    // 创建一个Transport
    transport := &BodyInterceptorTransport{
        Transport: http.DefaultTransport,
    }
 
    // 创建一个Client
    client := &http.Client{
        Transport: transport,
    }
 
    // 创建一个POST请求
    req, _ := http.NewRequest(http.MethodPost, "http://example.com", strings.NewReader("Original Body"))
 
    // 发起请求
    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println("Response:", resp.Status)
    }
}