用Go寫的HTTP代理服務(wù)器

小億
83
2023-12-19 05:43:52
欄目: 云計(jì)算

下面是一個(gè)用Go語(yǔ)言編寫的簡(jiǎn)單的HTTP代理服務(wù)器示例：

package main

import (
	"fmt"
	"io"
	"log"
	"net"
	"net/http"
)

func handleRequest(w http.ResponseWriter, r *http.Request) {
	// 創(chuàng)建一個(gè)新的HTTP請(qǐng)求
	req, err := http.NewRequest(r.Method, r.URL.String(), r.Body)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	// 復(fù)制原始請(qǐng)求的所有Header字段
	for key, values := range r.Header {
		for _, value := range values {
			req.Header.Add(key, value)
		}
	}

	// 創(chuàng)建一個(gè)HTTP客戶端并發(fā)送新請(qǐng)求
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	defer resp.Body.Close()

	// 復(fù)制響應(yīng)的所有Header字段
	for key, values := range resp.Header {
		for _, value := range values {
			w.Header().Add(key, value)
		}
	}

	// 將響應(yīng)狀態(tài)碼寫入ResponseWriter
	w.WriteHeader(resp.StatusCode)

	// 將響應(yīng)正文復(fù)制到ResponseWriter
	_, err = io.Copy(w, resp.Body)
	if err != nil {
		log.Println(err)
	}
}

func main() {
	// 設(shè)置代理監(jiān)聽地址
	proxyAddr := ":8080"

	// 創(chuàng)建一個(gè)HTTP代理服務(wù)器
	proxy := &http.Server{
		Addr:    proxyAddr,
		Handler: http.HandlerFunc(handleRequest),
	}

	// 啟動(dòng)代理服務(wù)器
	fmt.Printf("Starting HTTP proxy server on %s\n", proxyAddr)
	err := proxy.ListenAndServe()
	if err != nil {
		log.Fatal(err)
	}
}

在上面的示例中，我們創(chuàng)建了一個(gè)HTTP代理服務(wù)器，它會(huì)接收來(lái)自客戶端的HTTP請(qǐng)求，并將其轉(zhuǎn)發(fā)給目標(biāo)服務(wù)器。代理服務(wù)器將原始請(qǐng)求的所有Header字段復(fù)制到新請(qǐng)求中，然后使用HTTP客戶端發(fā)送新請(qǐng)求并獲取響應(yīng)。最后，代理服務(wù)器將響應(yīng)的所有Header字段復(fù)制到響應(yīng)中，并將響應(yīng)正文復(fù)制到客戶端的響應(yīng)中。

要使用此代理服務(wù)器，您可以將您的應(yīng)用程序或?yàn)g覽器的HTTP代理設(shè)置為代理服務(wù)器的地址（例如：http://localhost:8080）。然后，所有的HTTP請(qǐng)求將通過(guò)代理服務(wù)器進(jìn)行中轉(zhuǎn)。

0 踩