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

源站可以自定义回源主机名

Browse Source
This commit is contained in:
GoEdgeLab
2022-03-17 17:03:52 +08:00
parent 59cda8534b
commit dbeb40a11a
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
internal/nodes/http_request_reverse_proxy.go
View File

@@ -21,22 +21,22 @@ func (this *HTTPRequest) doReverseProxy() {
} }
// 对URL的处理 // 对URL的处理
stripPrefix := this.reverseProxy.StripPrefix var stripPrefix = this.reverseProxy.StripPrefix
requestURI := this.reverseProxy.RequestURI var requestURI = this.reverseProxy.RequestURI
requestURIHasVariables := this.reverseProxy.RequestURIHasVariables() var requestURIHasVariables = this.reverseProxy.RequestURIHasVariables()
var requestHost = "" var requestHost = ""
if this.reverseProxy.RequestHostType == serverconfigs.RequestHostTypeCustomized { if this.reverseProxy.RequestHostType == serverconfigs.RequestHostTypeCustomized {
requestHost = this.reverseProxy.RequestHost requestHost = this.reverseProxy.RequestHost
} }
requestHostHasVariables := this.reverseProxy.RequestHostHasVariables() var requestHostHasVariables = this.reverseProxy.RequestHostHasVariables()
// 源站 // 源站
requestCall := shared.NewRequestCall() var requestCall = shared.NewRequestCall()
requestCall.Request = this.RawReq requestCall.Request = this.RawReq
requestCall.Formatter = this.Format requestCall.Formatter = this.Format
requestCall.Domain = this.ReqHost requestCall.Domain = this.ReqHost
origin := this.reverseProxy.NextOrigin(requestCall) var origin = this.reverseProxy.NextOrigin(requestCall)
requestCall.CallResponseCallbacks(this.writer) requestCall.CallResponseCallbacks(this.writer)
if origin == nil { if origin == nil {
err := errors.New(this.URL() + ": no available origin sites for reverse proxy") err := errors.New(this.URL() + ": no available origin sites for reverse proxy")
@@ -89,9 +89,9 @@ func (this *HTTPRequest) doReverseProxy() {
} }
// 处理RequestURI中的问号 // 处理RequestURI中的问号
questionMark := strings.LastIndex(this.uri, "?") var questionMark = strings.LastIndex(this.uri, "?")
if questionMark > 0 { if questionMark > 0 {
path := this.uri[:questionMark] var path = this.uri[:questionMark]
if strings.Contains(path, "?") { if strings.Contains(path, "?") {
this.uri = path + "&" + this.uri[questionMark+1:] this.uri = path + "&" + this.uri[questionMark+1:]
} }
@@ -102,7 +102,7 @@ func (this *HTTPRequest) doReverseProxy() {
} }
// 获取源站地址 // 获取源站地址
originAddr := origin.Addr.PickAddress() var originAddr = origin.Addr.PickAddress()
if origin.Addr.HostHasVariables() { if origin.Addr.HostHasVariables() {
originAddr = this.Format(originAddr) originAddr = this.Format(originAddr)
} }
@@ -113,7 +113,7 @@ func (this *HTTPRequest) doReverseProxy() {
if requestHostHasVariables { if requestHostHasVariables {
this.RawReq.Host = this.Format(requestHost) this.RawReq.Host = this.Format(requestHost)
} else { } else {
this.RawReq.Host = this.reverseProxy.RequestHost this.RawReq.Host = requestHost
} }
this.RawReq.URL.Host = this.RawReq.Host this.RawReq.URL.Host = this.RawReq.Host
} else if this.reverseProxy.RequestHostType == serverconfigs.RequestHostTypeOrigin { } else if this.reverseProxy.RequestHostType == serverconfigs.RequestHostTypeOrigin {
@@ -132,7 +132,7 @@ func (this *HTTPRequest) doReverseProxy() {
} }
// 重组请求URL // 重组请求URL
questionMark := strings.Index(this.uri, "?") var questionMark = strings.Index(this.uri, "?")
if questionMark > -1 { if questionMark > -1 {
this.RawReq.URL.Path = this.uri[:questionMark] this.RawReq.URL.Path = this.uri[:questionMark]
this.RawReq.URL.RawQuery = this.uri[questionMark+1:] this.RawReq.URL.RawQuery = this.uri[questionMark+1:]
@@ -199,7 +199,7 @@ func (this *HTTPRequest) doReverseProxy() {
} }
} else { } else {
// 是否为客户端方面的错误 // 是否为客户端方面的错误
isClientError := false var isClientError = false
if ok { if ok {
if httpErr.Err == context.Canceled { if httpErr.Err == context.Canceled {
// 如果是服务器端主动关闭，则无需提示 // 如果是服务器端主动关闭，则无需提示
@@ -254,7 +254,7 @@ func (this *HTTPRequest) doReverseProxy() {
if this.web.Charset != nil && this.web.Charset.IsOn && len(this.web.Charset.Charset) > 0 { if this.web.Charset != nil && this.web.Charset.IsOn && len(this.web.Charset.Charset) > 0 {
contentTypes, ok := resp.Header["Content-Type"] contentTypes, ok := resp.Header["Content-Type"]
if ok && len(contentTypes) > 0 { if ok && len(contentTypes) > 0 {
contentType := contentTypes[0] var contentType = contentTypes[0]
if _, found := textMimeMap[contentType]; found { if _, found := textMimeMap[contentType]; found {
resp.Header["Content-Type"][0] = contentType + "; charset=" + this.web.Charset.Charset resp.Header["Content-Type"][0] = contentType + "; charset=" + this.web.Charset.Charset
} }
@@ -306,7 +306,7 @@ func (this *HTTPRequest) doReverseProxy() {
} }
pool.Put(buf) pool.Put(buf)
closeErr := resp.Body.Close() var closeErr = resp.Body.Close()
if closeErr != nil { if closeErr != nil {
if !this.canIgnore(closeErr) { if !this.canIgnore(closeErr) {
remotelogs.Warn("HTTP_REQUEST_REVERSE_PROXY", closeErr.Error()) remotelogs.Warn("HTTP_REQUEST_REVERSE_PROXY", closeErr.Error())