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

反向代理可以整体设置源站默认超时时间等参数

Browse Source
This commit is contained in:
GoEdgeLab
2021-03-26 22:11:05 +08:00
parent 73dbf38448
commit 43db1e6411
3 changed files with 102 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
internal/rpc/services/service_reverse_proxy.go
View File

@@ -5,6 +5,7 @@ import (
"encoding/json"
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
"github.com/iwind/TeaGo/types"
)
@@ -214,7 +215,32 @@ func (this *ReverseProxyService) UpdateReverseProxy(ctx context.Context, req *pb
tx := this.NullTx()
err = models.SharedReverseProxyDAO.UpdateReverseProxy(tx, req.ReverseProxyId, types.Int8(req.RequestHostType), req.RequestHost, req.RequestURI, req.StripPrefix, req.AutoFlush, req.AddHeaders)
// 校验参数
var connTimeout = &shared.TimeDuration{}
if len(req.ConnTimeoutJSON) > 0 {
err = json.Unmarshal(req.ConnTimeoutJSON, connTimeout)
if err != nil {
return nil, err
}
}
var readTimeout = &shared.TimeDuration{}
if len(req.ReadTimeoutJSON) > 0 {
err = json.Unmarshal(req.ReadTimeoutJSON, readTimeout)
if err != nil {
return nil, err
}
}
var idleTimeout = &shared.TimeDuration{}
if len(req.IdleTimeoutJSON) > 0 {
err = json.Unmarshal(req.IdleTimeoutJSON, idleTimeout)
if err != nil {
return nil, err
}
}
err = models.SharedReverseProxyDAO.UpdateReverseProxy(tx, req.ReverseProxyId, types.Int8(req.RequestHostType), req.RequestHost, req.RequestURI, req.StripPrefix, req.AutoFlush, req.AddHeaders, connTimeout, readTimeout, idleTimeout, req.MaxConns, req.MaxIdleConns)
if err != nil {
return nil, err
}