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

HTTP源站有多个L2节点时发生错误时会主动尝试下一个L2节点

Browse Source
This commit is contained in:
刘祥超
2022-08-03 20:30:59 +08:00
parent bc706237ef
commit ceadcfece9
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
internal/nodes/http_request_ln.go
View File

@@ -16,6 +16,6 @@ func (this *HTTPRequest) checkLnRequest() bool {
return false return false
} }
func (this *HTTPRequest) getLnOrigin() *serverconfigs.OriginConfig { func (this *HTTPRequest) getLnOrigin(excludingNodeIds []int64) (originConfig *serverconfigs.OriginConfig, lnNodeId int64, hasMultipleNodes bool) {
return nil return nil, 0, false
} }

13
internal/nodes/http_request_reverse_proxy.go
View File

@@ -24,20 +24,24 @@ func (this *HTTPRequest) doReverseProxy() {
var retries = 3 var retries = 3
var failedOriginIds []int64 var failedOriginIds []int64
var failedLnNodeIds []int64
for i := 0; i < retries; i++ { for i := 0; i < retries; i++ {
originId, shouldRetry := this.doOriginRequest(failedOriginIds, i == 0, i == retries-1) originId, lnNodeId, shouldRetry := this.doOriginRequest(failedOriginIds, failedLnNodeIds, i == 0, i == retries-1)
if !shouldRetry { if !shouldRetry {
break break
} }
if originId > 0 { if originId > 0 {
failedOriginIds = append(failedOriginIds, originId) failedOriginIds = append(failedOriginIds, originId)
} }
if lnNodeId > 0 {
failedLnNodeIds = append(failedLnNodeIds, lnNodeId)
}
} }
} }
// 请求源站 // 请求源站
func (this *HTTPRequest) doOriginRequest(failedOriginIds []int64, isFirstTry bool, isLastRetry bool) (originId int64, shouldRetry bool) { func (this *HTTPRequest) doOriginRequest(failedOriginIds []int64, failedLnNodeIds []int64, isFirstTry bool, isLastRetry bool) (originId int64, lnNodeId int64, shouldRetry bool) {
// 对URL的处理 // 对URL的处理
var stripPrefix = this.reverseProxy.StripPrefix var stripPrefix = this.reverseProxy.StripPrefix
var requestURI = this.reverseProxy.RequestURI var requestURI = this.reverseProxy.RequestURI
@@ -59,8 +63,9 @@ func (this *HTTPRequest) doOriginRequest(failedOriginIds []int64, isFirstTry boo
var origin *serverconfigs.OriginConfig var origin *serverconfigs.OriginConfig
// 二级节点 // 二级节点
var hasMultipleLnNodes = false
if this.cacheRef != nil { if this.cacheRef != nil {
origin = this.getLnOrigin() origin, lnNodeId, hasMultipleLnNodes = this.getLnOrigin(failedLnNodeIds)
if origin != nil { if origin != nil {
// 强制变更原来访问的域名 // 强制变更原来访问的域名
requestHost = this.ReqHost requestHost = this.ReqHost
@@ -262,7 +267,7 @@ func (this *HTTPRequest) doOriginRequest(failedOriginIds []int64, isFirstTry boo
}) })
// 是否需要重试 // 是否需要重试
if originId > 0 && !isLastRetry { if (originId > 0 || (lnNodeId > 0 && hasMultipleLnNodes)) && !isLastRetry {
shouldRetry = true shouldRetry = true
this.uri = oldURI // 恢复备份 this.uri = oldURI // 恢复备份