Golang搭建简易HTTP代理教程

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《Golang实现简易HTTP代理教程》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

Golang制作简易HTTP代理需监听端口接收请求并转发。1. 创建HTTP服务器，监听指定端口，接收客户端请求；2. 修改请求头、URL等信息，若URL无Scheme默认使用http，处理无Host情况并返回错误；3. 使用http.NewRequest创建请求，复制原始请求头，通过http.Client发送至目标服务器；4. 处理响应头、状态码及响应体，将其返回客户端。代码中加入对错误的处理，并将错误码设为StatusBadGateway以符合代理场景。处理HTTPS请求时需配置TLSClientConfig，可跳过证书验证（仅测试用）。WebSocket连接需用httputil.NewSingleHostReverseProxy实现反向代理，并设置Director函数修改请求。高级功能如请求过滤可在handleRequest中检查URL或方法进行拦截；缓存可通过sync.Map或第三方库实现。

Golang制作简易HTTP代理，核心在于利用net/http包监听端口，接收客户端请求，然后将请求转发到目标服务器，并将响应返回给客户端。这涉及到请求的修改、转发以及响应的处理。

解决方案

首先，我们需要创建一个HTTP服务器来监听指定的端口。

package main

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

func handleRequest(w http.ResponseWriter, r *http.Request) {
    log.Printf("Received request: %s %s", r.Method, r.URL.String())

    // 1. 修改请求
    // 这里可以根据需要修改请求头、URL等

    // 2. 创建并发送请求
    targetURL := r.URL
    if targetURL.Scheme == "" {
        targetURL.Scheme = "http" // 默认使用http
        if targetURL.Host == "" {
            // 处理没有Host的情况，例如直接访问路径
            fmt.Println("No host provided in URL")
            http.Error(w, "Invalid URL", http.StatusBadRequest)
            return
        }
    }

    proxyReq, err := http.NewRequest(r.Method, targetURL.String(), r.Body)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 复制请求头
    proxyReq.Header = r.Header

    client := &http.Client{}
    resp, err := client.Do(proxyReq)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadGateway) // 使用StatusBadGateway更合适
        return
    }
    defer resp.Body.Close()

    // 3. 处理响应
    // 复制响应头
    for key, values := range resp.Header {
        for _, value := range values {
            w.Header().Add(key, value)
        }
    }

    // 设置状态码
    w.WriteHeader(resp.StatusCode)

    // 复制响应体
    _, err = io.Copy(w, resp.Body)
    if err != nil {
        log.Println("Error copying response body:", err)
    }
}

func main() {
    http.HandleFunc("/", handleRequest)

    fmt.Println("Starting proxy server on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

这段代码创建了一个简单的HTTP代理，它接收所有请求，并将它们转发到目标服务器。注意，这里有一个小小的改进：如果请求的URL没有指定Scheme（例如http或https），我们默认使用http。 同时，加入了对没有Host的情况的处理。 并且将错误码改为了StatusBadGateway，更符合代理服务器的场景。

如何处理HTTPS请求的转发？

处理HTTPS请求与HTTP请求类似，但需要注意TLS握手和证书验证。 如果目标服务器使用了自签名证书，你可能需要配置客户端跳过证书验证。 这可以通过设置http.Transport的TLSClientConfig来实现。

// ... 在main函数中

    transport := &http.Transport{
        TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, // 生产环境不要这样做
    }
    client := &http.Client{Transport: transport}

// ...

请注意，在生产环境中跳过证书验证是非常危险的，因为它会使你的代理容易受到中间人攻击。 应该配置正确的证书链，而不是简单地跳过验证。

代理服务器如何处理WebSocket连接？

WebSocket连接需要升级协议，不能简单地像HTTP请求那样转发。 你需要使用net/http/httputil包中的ReverseProxy来实现WebSocket代理。

import (
    "net/http"
    "net/http/httputil"
    "net/url"
)

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    target, err := url.Parse("ws://target-websocket-server:8080") // 替换为你的WebSocket服务器地址
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    proxy := httputil.NewSingleHostReverseProxy(target)
    proxy.Director = func(req *http.Request) {
        req.URL.Scheme = target.Scheme
        req.URL.Host = target.Host
        req.Header.Set("X-Forwarded-For", req.RemoteAddr)
    }

    proxy.ServeHTTP(w, r)
}

// 在main函数中添加路由
http.HandleFunc("/ws", handleWebSocket)

这段代码创建了一个反向代理，专门用于处理WebSocket连接。 Director函数负责修改请求，使其指向目标WebSocket服务器。

如何实现更高级的功能，例如请求过滤和缓存？

请求过滤可以通过在handleRequest函数中检查请求的URL、方法、头部等信息来实现。 你可以根据需要阻止某些请求，或者修改它们。

缓存可以使用内存缓存、磁盘缓存或分布式缓存来实现。 你可以使用sync.Map来实现简单的内存缓存，或者使用github.com/patrickmn/go-cache等第三方库来实现更高级的缓存功能。

请求过滤的例子：

func handleRequest(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path == "/forbidden" {
        http.Error(w, "Forbidden", http.StatusForbidden)
        return
    }
    // ... 转发请求的代码
}

这个例子展示了如何阻止访问/forbidden路径的请求。 你可以根据需要添加更复杂的过滤规则。

本篇关于《Golang搭建简易HTTP代理教程》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！