Go语言设置HTTP代理：从环境变量中获取代理信息

在使用Go语言发送HTTP请求时，有时需要设置代理服务器才能访问目标网站。本文将介绍如何从环境变量中获取代理信息，并在Go代码中设置HTTP代理。

代码示例

以下是一个完整的示例代码，演示了如何从环境变量中获取代理信息并设置代理：gopackage main

import ( 'fmt' 'net/http' 'net/url' 'os')

func main() { // 从环境变量中获取代理信息 proxyURL := getProxyFromEnvironment()

// 创建一个自定义的Transport，并设置代理	transport := &http.Transport{		Proxy: http.ProxyURL(proxyURL),	}

// 创建一个带有自定义Transport的http.Client	client := &http.Client{		Transport: transport,	}

// 发送HTTP请求	resp, err := client.Get('http://example.com')	if err != nil {		fmt.Println('请求失败:', err)		return	}	defer resp.Body.Close()

// 处理响应	fmt.Println('响应状态码:', resp.StatusCode)}

func getProxyFromEnvironment() *url.URL { // 从环境变量中获取代理信息 proxyURL := os.Getenv('HTTP_PROXY') if proxyURL == '' { proxyURL = os.Getenv('http_proxy') }

// 解析代理URL	parsedURL, err := url.Parse(proxyURL)	if err != nil {		fmt.Println('无效的代理URL:', err)		return nil	}

return parsedURL}

代码解释

getProxyFromEnvironment 函数: - 该函数首先尝试从名为 HTTP_PROXY 的环境变量中获取代理信息。 - 如果 HTTP_PROXY 为空，则尝试从 http_proxy 环境变量中获取。 - 获取到代理信息后，使用 url.Parse 函数解析代理URL，并返回解析后的 *url.URL 对象。
main 函数: - 调用 getProxyFromEnvironment 函数获取代理信息。 - 创建一个 http.Transport 对象，并将获取到的代理URL设置到 Proxy 字段中。 - 创建一个 http.Client 对象，并将自定义的 Transport 设置到 Transport 字段中。 - 使用自定义的 http.Client 发送HTTP请求。

总结

通过上述步骤，您可以轻松地在Go语言中设置HTTP代理，并从环境变量中获取代理信息。这种方法可以使您的代码更加灵活，并且可以方便地在不同的环境中使用。