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

[反向代理]源站可以设置名称、描述、是否启用

Browse Source
This commit is contained in:
刘祥超
2020-11-30 20:00:44 +08:00
parent 118926e55e
commit e2a0fcbd8c
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -91,9 +91,9 @@ func (this *OriginDAO) FindOriginName(id int64) (string, error) {
} }
// 创建源站 // 创建源站
func (this *OriginDAO) CreateOrigin(name string, addrJSON string, description string, weight int32) (originId int64, err error) { func (this *OriginDAO) CreateOrigin(name string, addrJSON string, description string, weight int32, isOn bool) (originId int64, err error) {
op := NewOriginOperator() op := NewOriginOperator()
op.IsOn = true op.IsOn = isOn
op.Name = name op.Name = name
op.Addr = addrJSON op.Addr = addrJSON
op.Description = description op.Description = description
@@ -110,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, weight int32) error { func (this *OriginDAO) UpdateOrigin(originId int64, name string, addrJSON string, description string, weight int32, isOn bool) error {
if originId <= 0 { if originId <= 0 {
return errors.New("invalid originId") return errors.New("invalid originId")
} }
@@ -123,6 +123,7 @@ func (this *OriginDAO) UpdateOrigin(originId int64, name string, addrJSON string
weight = 0 weight = 0
} }
op.Weight = weight op.Weight = weight
op.IsOn = isOn
op.Version = dbs.SQL("version+1") op.Version = dbs.SQL("version+1")
_, err := this.Save(op) _, err := this.Save(op)
return err return err

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

@@ -30,7 +30,7 @@ func (this *OriginService) CreateOrigin(ctx context.Context, req *pb.CreateOrigi
"portRange": req.Addr.PortRange, "portRange": req.Addr.PortRange,
"host": req.Addr.Host, "host": req.Addr.Host,
} }
originId, err := models.SharedOriginDAO.CreateOrigin(req.Name, string(addrMap.AsJSON()), req.Description, req.Weight) originId, err := models.SharedOriginDAO.CreateOrigin(req.Name, string(addrMap.AsJSON()), req.Description, req.Weight, req.IsOn)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -53,7 +53,7 @@ func (this *OriginService) UpdateOrigin(ctx context.Context, req *pb.UpdateOrigi
"portRange": req.Addr.PortRange, "portRange": req.Addr.PortRange,
"host": req.Addr.Host, "host": req.Addr.Host,
} }
err = models.SharedOriginDAO.UpdateOrigin(req.OriginId, req.Name, string(addrMap.AsJSON()), req.Description, req.Weight) err = models.SharedOriginDAO.UpdateOrigin(req.OriginId, req.Name, string(addrMap.AsJSON()), req.Description, req.Weight, req.IsOn)
if err != nil { if err != nil {
return nil, err return nil, err
} }