网络代理【2】Golang实现简单的正向代理

待我称王封你为后i 2023-03-01 12:25 32阅读 0赞

目录:
    	一、实现步骤
    	二、代码
    	三、浏览器配置

**一、实现步骤:**

*  代理接收客户端请求，复制原请求对象，并根据数据配置新请求各种参数
 *  把新请求发送到真实服务端，并接收待服务器返回
 *  代理服务器对响应做一些处理，返回给客户端

**二、代码:**

// Pxy ServerHTTP方法绑定的对象
    type Pxy struct{ }
    
    // ServerHTTP 实现http.Handler接口中的方法
    func (p *Pxy) ServeHTTP(rw http.ResponseWriter, req *http.Request) { 
    	fmt.Printf(`Received request {"Method":"%s","Host":"%s","RemoteAddr":"%s"}`, req.Method, req.Host, req.RemoteAddr)
    	fmt.Println()
    	//1. 复制请求, 修改头部信息
    	pxyReq := new(http.Request)
    	*pxyReq = *req
    	if clientIP, _, err := net.SplitHostPort(pxyReq.RemoteAddr); err == nil { 
    		if prior, ok := pxyReq.Header["X-Forwarded-For"]; ok { 
    			clientIP = strings.Join(prior, ", ") + ", " + clientIP
    		}
    		pxyReq.Header.Set("X-Forwarded-For", clientIP)
    	}
    	//2. 请求目的机器
    	transport := http.DefaultTransport
    	res, err := transport.RoundTrip(pxyReq)
    	if err != nil { 
    		rw.WriteHeader(http.StatusBadGateway)
    		return
    	}
    	//3. 回写客户端
    	for key, values := range res.Header { 
    		for _, v := range values { 
    			rw.Header().Add(key, v)
    		}
    	}
    	rw.WriteHeader(res.StatusCode)
    	io.Copy(rw, res.Body)
    	res.Body.Close()
    }
    
    func main() { 
    	http.Handle("/", &Pxy{ })
    	http.ListenAndServe(":8080", nil)
    }

**三、浏览器配置:**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4OTAwNTY1_size_16_color_FFFFFF_t_70]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4OTAwNTY1_size_16_color_FFFFFF_t_70]: /images/20230208/5fd99f7c0bed4d6d90a539998443c4f3.png