温馨提示×

用Go写的HTTP代理服务器

小亿
93
2023-12-19 05:43:52
栏目: 云计算

下面是一个用Go语言编写的简单的HTTP代理服务器示例:

package main

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

func handleRequest(w http.ResponseWriter, r *http.Request) {
	// 创建一个新的HTTP请求
	req, err := http.NewRequest(r.Method, r.URL.String(), r.Body)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	// 复制原始请求的所有Header字段
	for key, values := range r.Header {
		for _, value := range values {
			req.Header.Add(key, value)
		}
	}

	// 创建一个HTTP客户端并发送新请求
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	defer resp.Body.Close()

	// 复制响应的所有Header字段
	for key, values := range resp.Header {
		for _, value := range values {
			w.Header().Add(key, value)
		}
	}

	// 将响应状态码写入ResponseWriter
	w.WriteHeader(resp.StatusCode)

	// 将响应正文复制到ResponseWriter
	_, err = io.Copy(w, resp.Body)
	if err != nil {
		log.Println(err)
	}
}

func main() {
	// 设置代理监听地址
	proxyAddr := ":8080"

	// 创建一个HTTP代理服务器
	proxy := &http.Server{
		Addr:    proxyAddr,
		Handler: http.HandlerFunc(handleRequest),
	}

	// 启动代理服务器
	fmt.Printf("Starting HTTP proxy server on %s\n", proxyAddr)
	err := proxy.ListenAndServe()
	if err != nil {
		log.Fatal(err)
	}
}

在上面的示例中,我们创建了一个HTTP代理服务器,它会接收来自客户端的HTTP请求,并将其转发给目标服务器。代理服务器将原始请求的所有Header字段复制到新请求中,然后使用HTTP客户端发送新请求并获取响应。最后,代理服务器将响应的所有Header字段复制到响应中,并将响应正文复制到客户端的响应中。

要使用此代理服务器,您可以将您的应用程序或浏览器的HTTP代理设置为代理服务器的地址(例如:http://localhost:8080)。然后,所有的HTTP请求将通过代理服务器进行中转。

0