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
2020-11-17 17:11:40 +08:00
parent 3fe9b4dcde
commit 05611f98f3
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/db/models/origin_dao.go
View File

@@ -91,12 +91,16 @@ func (this *OriginDAO) FindOriginName(id int64) (string, error) {
}
// 创建源站
func (this *OriginDAO) CreateOrigin(name string, addrJSON string, description string) (originId int64, err error) {
func (this *OriginDAO) CreateOrigin(name string, addrJSON string, description string, weight int32) (originId int64, err error) {
op := NewOriginOperator()
op.IsOn = true
op.Name = name
op.Addr = addrJSON
op.Description = description
if weight < 0 {
weight = 0
}
op.Weight = weight
op.State = OriginStateEnabled
_, err = this.Save(op)
if err != nil {
@@ -106,7 +110,7 @@ func (this *OriginDAO) CreateOrigin(name string, addrJSON string, description st
}
// 修改源站
func (this *OriginDAO) UpdateOrigin(originId int64, name string, addrJSON string, description string) error {
func (this *OriginDAO) UpdateOrigin(originId int64, name string, addrJSON string, description string, weight int32) error {
if originId <= 0 {
return errors.New("invalid originId")
}
@@ -115,6 +119,10 @@ func (this *OriginDAO) UpdateOrigin(originId int64, name string, addrJSON string
op.Name = name
op.Addr = addrJSON
op.Description = description
if weight < 0 {
weight = 0
}
op.Weight = weight
op.Version = dbs.SQL("version+1")
_, err := this.Save(op)
return err