TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-12-16 07:46:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

[反向代理]增加请求主机名类型选择

Browse Source
This commit is contained in:
刘祥超
2020-11-30 22:27:50 +08:00
parent 021ef3dd84
commit 49d3c1b586
3 changed files with 28 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/nodes/http_client_pool.go
View File

@@ -36,17 +36,12 @@ func NewHTTPClientPool() *HTTPClientPool {
}
// 根据地址获取客户端
func (this *HTTPClientPool) Client(req *HTTPRequest, origin *serverconfigs.OriginConfig) (rawClient *http.Client, realAddr string, err error) {
func (this *HTTPClientPool) Client(origin *serverconfigs.OriginConfig, originAddr string) (rawClient *http.Client, err error) {
if origin.Addr == nil {
return nil, "", errors.New("origin addr should not be empty (originId:" + strconv.FormatInt(origin.Id, 10) + ")")
return nil, errors.New("origin addr should not be empty (originId:" + strconv.FormatInt(origin.Id, 10) + ")")
}
key := origin.UniqueKey()
originAddr := origin.Addr.PickAddress()
if origin.Addr.HostHasVariables() {
originAddr = req.Format(originAddr)
}
key += "@" + originAddr
key := origin.UniqueKey() + "@" + originAddr
this.locker.Lock()
defer this.locker.Unlock()
@@ -54,7 +49,7 @@ func (this *HTTPClientPool) Client(req *HTTPRequest, origin *serverconfigs.Origi
client, found := this.clientsMap[key]
if found {
client.UpdateAccessTime()
return client.RawClient(), originAddr, nil
return client.RawClient(), nil
}
maxConnections := origin.MaxConns
@@ -128,7 +123,7 @@ func (this *HTTPClientPool) Client(req *HTTPRequest, origin *serverconfigs.Origi
this.clientsMap[key] = NewHTTPClient(rawClient)
return rawClient, originAddr, nil
return rawClient, nil
}
// 清理不使用的Client