go系列【仅供参考】：Go语言中http.Transport的请求过滤与拦截技巧与应用

本文链接：https://blog.youkuaiyun.com/weixin_54626591/article/details/146735780

Go语言中http.Transport的请求过滤与拦截技巧与应用

Go语言中http.Transport的请求过滤与拦截技巧与应用

Go语言中http.Transport的请求过滤与拦截技巧与应用

在Go语言中，http.Transport作为底层的HTTP传输层实现，提供了强大的功能用于发起HTTP请求。通过使用http.Transport的请求过滤和拦截技巧，开发者可以定制化地对HTTP请求进行过滤、修改和拦截，这在实现一些特殊的功能和需求时非常有用。以下是对这些技巧及其应用的详细简述：

请求过滤

过滤请求方法：

可以创建一个实现了http.RoundTripper接口的结构体，并在其中重写RoundTrip方法以过滤特定的HTTP请求方法。
例如，可以过滤掉所有的GET请求，如果收到GET请求，就返回一个错误。

过滤和修改请求头：

在重写的RoundTrip方法中，可以访问和修改请求的头部信息。
例如，可以删除原有的User-Agent请求头，并设置一个自定义的User-Agent。

请求拦截

拦截并修改请求体：

在重写的RoundTrip方法中，可以拦截请求并修改其请求体。
例如，对于POST请求，可以读取原始请求体，然后替换为自定义的修改后的请求体，并设置相应的Content-Type。

应用示例

以下是一个简单的应用示例，展示了如何使用上述技巧来过滤GET请求、修改User-Agent请求头以及拦截并修改POST请求的请求体：

package main
 
import (
	"bytes"
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)
 
// 过滤GET请求的结构体
type FilterTransport struct {
	Transport http.RoundTripper
}
 
func (t *FilterTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	// 过滤GET请求
	if req.Method == http.MethodGet {
		return nil, errors.New("Method Not Allowed")
	}
	return t.Transport.RoundTrip(req)
}
 
// 修改User-Agent请求头的结构体
type HeaderFilterTransport struct {
	Transport http.RoundTripper
}
 
func (t *HeaderFilterTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	// 修改User-Agent请求头
	req.Header.Del("User-Agent")
	req.Header.Set("User-Agent", "Custom User-Agent")
	return t.Transport.RoundTrip(req)
}
 
// 拦截并修改POST请求请求体的结构体
type BodyInterceptorTransport struct {
	Transport http.RoundTripper
}
 
func (t *BodyInterceptorTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	// 拦截POST请求并修改请求体
	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)
		req.Header.Set("Content-Type", "text/plain")
	}
	return t.Transport.RoundTrip(req)
}
 
func main() {
	// 创建过滤GET请求的Transport
	filterTransport := &FilterTransport{
		Transport: http.DefaultTransport,
	}
	// 创建修改User-Agent请求头的Transport
	headerFilterTransport := &HeaderFilterTransport{
		Transport: http.DefaultTransport,
	}
	// 创建拦截并修改POST请求请求体的Transport
	bodyInterceptorTransport := &BodyInterceptorTransport{
		Transport: http.DefaultTransport,
	}
 
	// 使用不同的Transport创建Client并发送请求
	client := &http.Client{
		Transport: filterTransport, // 可以替换为headerFilterTransport或bodyInterceptorTransport进行测试
	}
 
	// 创建一个GET请求（应该被过滤）
	getReq, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)
	getResp, err := client.Do(getReq)
	if err != nil {
		fmt.Println("GET请求错误:", err)
	} else {
		fmt.Println("GET响应状态:", getResp.Status)
	}
 
	// 创建一个修改后的Client（使用修改User-Agent的Transport）并发送GET请求
	client = &http.Client{
		Transport: headerFilterTransport,
	}
	getReq, _ = http.NewRequest(http.MethodGet, "http://example.com", nil)
	getResp, err = client.Do(getReq)
	if err != nil {
		fmt.Println("GET请求错误:", err)
	} else {
		fmt.Println("修改User-Agent后的GET响应状态:", getResp.Status)
	}
 
	// 创建一个POST请求（应该被拦截并修改请求体）
	postReq, _ := http.NewRequest(http.MethodPost, "http://example.com", strings.NewReader("Original Body"))
	postResp, err := client.Do(postReq) // 注意：这里应该使用拦截并修改请求体的Transport进行测试
	// 但由于示例中只有一个client变量，且为了保持示例简洁性，这里未重新创建client
	// 实际使用时，应确保client的Transport为bodyInterceptorTransport
	if err != nil {
		fmt.Println("POST请求错误:", err)
	} else {
		fmt.Println("拦截并修改请求体后的POST响应状态:", postResp.Status)
		// 注意：此处可以进一步读取和验证响应内容以确保请求体已被正确修改
	}
 
	// 注意：上述代码中的POST请求部分为了保持示例简洁性，并未重新创建使用bodyInterceptorTransport的client
	// 在实际使用中，应确保为需要拦截和修改请求体的请求创建正确的client实例
}