在Go语言中，编写一个HTTP代理服务器相对简单且直观。代理服务器的主要职责是接收客户端的请求，然后将请求转发到目标服务器，再将目标服务器的响应返回给客户端。下面是一个简单的示例，展示如何使用Go语言编写一个基本的HTTP代理服务器：
go 复制代码
|---|---------------------------------------------------------------|
| | package main |
| | |
| | import ( |
| | "io" |
| | "log" |
| | "net" |
| | "net/http" |
| | ) |
| | |
| | func handleRequest(w http.ResponseWriter, r *http.Request) { |
| | // 创建到目标服务器的连接 |
| | destConn, err := net.Dial("tcp", r.Host) |
| | if err != nil { |
| | log.Printf("无法连接到目标服务器: %v", err) |
| | http.Error(w, "无法连接到目标服务器", http.StatusInternalServerError) |
| | return |
| | } |
| | defer destConn.Close() |
| | |
| | // 将客户端的请求发送到目标服务器 |
| | err = r.Write(destConn) |
| | if err != nil { |
| | log.Printf("请求发送失败: %v", err) |
| | http.Error(w, "请求发送失败", http.StatusInternalServerError) |
| | return |
| | } |
| | |
| | // 将目标服务器的响应返回给客户端 |
| | buf := make([]byte, 4096) |
| | for { |
| | n, err := destConn.Read(buf) |
| | if err != nil && err != io.EOF { |
| | log.Printf("读取响应时出错: %v", err) |
| | http.Error(w, "读取响应时出错", http.StatusInternalServerError) |
| | return |
| | } |
| | if n == 0 { |
| | break |
| | } |
| | _, err = w.Write(buf[:n]) |
| | if err != nil { |
| | log.Printf("响应写入失败: %v", err) |
| | http.Error(w, "响应写入失败", http.StatusInternalServerError) |
| | return |
| | } |
| | } |
| | } |
| | |
| | func main() { |
| | // 创建HTTP代理服务器 |
| | proxy := &http.Server{ |
| | Addr: ":8080", // 代理服务器监听的端口号，可以根据需要进行修改 |
| | Handler: http.HandlerFunc(handleRequest), |
| | } |
| | |
| | log.Println("启动HTTP代理服务器，监听端口8080...") |
| | err := proxy.ListenAndServe() |
| | if err != nil { |
| | log.Fatalf("代理服务器启动失败: %v", err) |
| | } |
| | } |
这个示例代码创建了一个简单的HTTP代理服务器，它监听本地的8080端口。当客户端发送请求时，代理服务器会将请求转发到目标服务器，并将目标服务器的响应返回给客户端。请注意，这只是一个基本示例，可能需要根据实际需求进行进一步的定制和优化。