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

反向代理中源站增加最大连接数、连接超时时间等参数

Browse Source
This commit is contained in:
刘祥超
2021-03-25 21:17:59 +08:00
parent d6058a9421
commit d6de02b031
2 changed files with 121 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
internal/rpc/services/service_origin.go
View File

@@ -6,6 +6,7 @@ import (
"errors"
"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/maps"
)
@@ -32,7 +33,32 @@ func (this *OriginService) CreateOrigin(ctx context.Context, req *pb.CreateOrigi
tx := this.NullTx()
originId, err := models.SharedOriginDAO.CreateOrigin(tx, adminId, userId, req.Name, string(addrMap.AsJSON()), req.Description, req.Weight, req.IsOn)
// 校验参数
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
}
}
originId, err := models.SharedOriginDAO.CreateOrigin(tx, adminId, userId, req.Name, string(addrMap.AsJSON()), req.Description, req.Weight, req.IsOn, connTimeout, readTimeout, idleTimeout, req.MaxConns, req.MaxIdleConns)
if err != nil {
return nil, err
}
@@ -61,7 +87,32 @@ func (this *OriginService) UpdateOrigin(ctx context.Context, req *pb.UpdateOrigi
tx := this.NullTx()
err = models.SharedOriginDAO.UpdateOrigin(tx, req.OriginId, req.Name, string(addrMap.AsJSON()), req.Description, req.Weight, req.IsOn)
// 校验参数
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.SharedOriginDAO.UpdateOrigin(tx, req.OriginId, req.Name, string(addrMap.AsJSON()), req.Description, req.Weight, req.IsOn, connTimeout, readTimeout, idleTimeout, req.MaxConns, req.MaxIdleConns)
if err != nil {
return nil, err
}