TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-12 22:40:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

[反向代理]反向代理部分的错误从500修改为502

Browse Source
This commit is contained in:
GoEdgeLab
2020-11-30 20:57:16 +08:00
parent f61aa657fc
commit ba8a7d6fdb
2 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/nodes/http_request_error.go
View File

@@ -30,3 +30,17 @@ func (this *HTTPRequest) write500(err error) {
this.writer.WriteHeader(statusCode)
_, _ = this.writer.Write([]byte(http.StatusText(statusCode)))
}
func (this *HTTPRequest) write502(err error) {
if err != nil {
this.addError(err)
}
statusCode := http.StatusBadGateway
if this.doPage(statusCode) {
return
}
this.processResponseHeaders(statusCode)
this.writer.WriteHeader(statusCode)
_, _ = this.writer.Write([]byte(http.StatusText(statusCode)))
}

10
internal/nodes/http_request_reverse_proxy.go
View File

@@ -31,7 +31,7 @@ func (this *HTTPRequest) doReverseProxy() {
if origin == nil {
err := errors.New(this.requestPath() + ": no available backends for reverse proxy")
logs.Error("REQUEST_REVERSE_PROXY", err.Error())
this.write500(err)
this.write502(err)
return
}
this.origin = origin // 设置全局变量是为了日志等处理
@@ -51,7 +51,7 @@ func (this *HTTPRequest) doReverseProxy() {
if origin.Addr == nil {
err := errors.New(this.requestPath() + ": origin '" + strconv.FormatInt(origin.Id, 10) + "' does not has a address")
logs.Error("REQUEST_REVERSE_PROXY", err.Error())
this.write500(err)
this.write502(err)
return
}
this.RawReq.URL.Scheme = origin.Addr.Protocol.Primary().Scheme()
@@ -128,7 +128,7 @@ func (this *HTTPRequest) doReverseProxy() {
client, addr, err := SharedHTTPClientPool.Client(this, origin)
if err != nil {
logs.Error("REQUEST_REVERSE_PROXY", err.Error())
this.write500(err)
this.write502(err)
return
}
@@ -142,7 +142,7 @@ func (this *HTTPRequest) doReverseProxy() {
if !ok || httpErr.Err != context.Canceled {
// TODO 如果超过最大失败次数，则下线
this.write500(err)
this.write502(err)
logs.Println("REQUEST_REVERSE_PROXY", this.RawReq.URL.String()+"': "+err.Error())
} else {
// 是否为客户端方面的错误
@@ -156,7 +156,7 @@ func (this *HTTPRequest) doReverseProxy() {
}
if !isClientError {
this.write500(err)
this.write502(err)
}
}
if resp != nil && resp.Body != nil {