您可以在代码的处理本地请求的goroutine之前添加一个判断来满足您的要求。下面是修改后的代码示例:
// 创建HTTP处理函数来同时处理本地请求和代理请求
handlerFunc := func(w http.ResponseWriter, r *http.Request) {
var wg sync.WaitGroup
wg.Add(2)
// 判断是否启用HA且host可达
if haEnabled && !isHostReachable(haHost) {
// 直接处理本地请求,不进行代理转发
go func() {
defer wg.Done()
// 处理本地请求的逻辑
// ...
}()
} else {
// 处理本地请求的goroutine
go func() {
defer wg.Done()
// 处理本地请求的逻辑
// ...
}()
// 将本地请求转发给代理服务器的goroutine
go func() {
defer wg.Done()
proxyURL := "http://proxy-server:port" // 代理服务器地址
proxyReq, err := http.NewRequest(r.Method, proxyURL, r.Body)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
proxyReq.Header = r.Header.Clone()
resp, err := proxyClient.Do(proxyReq)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer resp.Body.Close()
// 将代理响应返回给本地客户端
for key, values := range resp.Header {
for _, value := range values {
w.Header().Add(key, value)
}
}
w.WriteHeader(resp.StatusCode)
io.Copy(w, resp.Body)
}()
}
wg.Wait()
}
// 判断主机是否可达
func isHostReachable(host string) bool {
_, err := net.Dial("tcp", host)
return err == nil
}
请注意,上述代码中的 haEnabled
是一个布尔值,表示HA是否启用,haHost
是代理服务器的主机地址。根据您的实际情况修改这些变量,并在适当位置调用 isHostReachable
函数来判断主机是否可达。
内容由零声教学AI助手提供,问题来源于学员提问