package main

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

// ForwardRequest 是一個通用的HTTP請求轉(zhuǎn)發(fā)函數(shù)
func ForwardRequest(req *http.Request, targetURL string, headers map[string]string) (*http.Response, error) {
	// 創(chuàng)建一個新的HTTP客戶端
	client := &http.Client{
		Timeout: 5 * time.Second,
	}

	// 復(fù)制原始請求的所有頭信息，并添加額外的頭（如果有）
	req, err := http.NewRequest(req.Method, targetURL, req.Body)
	if err != nil {
		return nil, err
	}

	// 復(fù)制原始請求的頭信息
	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)
	}

	// 發(fā)送新的請求到目標(biāo)URL
	resp, err := client.Do(req)
	if err != nil {
		return nil, err
	}

	return resp, nil
}

func main() {
	// 創(chuàng)建一個模擬的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)
	}

	// 設(shè)置請求頭（如果需要）
	req.Header.Set("Content-Type", "application/json")

	// 目標(biāo)URL，即請求將被轉(zhuǎn)發(fā)到的地址
	targetURL := "http://localhost:8080/hello"

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

	// 調(diào)用轉(zhuǎn)發(fā)函數(shù)
	resp, err := ForwardRequest(req, targetURL, extraHeaders)
	if err != nil {
		log.Fatalf("Failed to forward request: %v", err)
	}
	defer resp.Body.Close()

	// 讀取并打印響應(yīng)體
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatalf("Failed to read response body: %v", err)
	}
	fmt.Printf("Response: %s\n", body)

	// 打印響應(yīng)狀態(tài)碼和其他頭信息（可選）
	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)
		}
	}
}