Go实现HTTP请求转发的示例代码

package main

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"time"
)

// ForwardRequest 是一个通用的HTTP请求转发函数
func ForwardRequest(req *http.Request, targetURL string, headers map[string]string) (*http.Response, error) {
	// 创建一个新的HTTP客户端
	client := &http.Client{
		Timeout: 5 * time.Second,
	}

	// 复制原始请求的所有头信息，并添加额外的头（如果有）
	req, err := http.NewRequest(req.Method, targetURL, req.Body)
	if err != nil {
		return nil, err
	}

	// 复制原始请求的头信息
	req.Header = make(http.Header)
	for kk, vv := range req.Header {
		for _, v := range vv {
			req.Header.Add(kk, v)
		}
	}

	// 添加额外的头信息（如果有）
	for k, v := range headers {
		req.Header.Set(k, v)
	}

	// 发送新的请求到目标URL
	resp, err := client.Do(req)
	if err != nil {
		return nil, err
	}

	return resp, nil
}

func main() {
	// 创建一个模拟的HTTP请求（这里使用POST方法，但也可以是GET或其他方法）
	req, err := http.NewRequest("POST", "http://localhost:80/hello", bytes.NewBufferString("data"))
	if err != nil {
		log.Fatalf("Failed to create request: %v", err)
	}

	// 设置请求头（如果需要）
	req.Header.Set("Content-Type", "application/json")

	// 目标URL，即请求将被转发到的地址
	targetURL := "http://localhost:8080/hello"

	// 额外的头信息（可选）
	extraHeaders := map[string]string{
		"X-Forwarded-For": "client-ip",
	}

	// 调用转发函数
	resp, err := ForwardRequest(req, targetURL, extraHeaders)
	if err != nil {
		log.Fatalf("Failed to forward request: %v", err)
	}
	defer resp.Body.Close()

	// 读取并打印响应体
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatalf("Failed to read response body: %v", err)
	}
	fmt.Printf("Response: %s\n", body)

	// 打印响应状态码和其他头信息（可选）
	fmt.Printf("Status Code: %d\n", resp.StatusCode)
	for k, vv := range resp.Header {
		for _, v := range vv {
			fmt.Printf("%s: %s\n", k, v)
		}
	}
}