TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-12-19 01:46:35 +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.IsOn = true
op.IsOn = isOn
op.Name = name
op.Addr = addrJSON
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 {
return errors.New("invalid originId")
}
@@ -123,6 +123,7 @@ func (this *OriginDAO) UpdateOrigin(originId int64, name string, addrJSON string
weight = 0
}
op.Weight = weight
op.IsOn = isOn
op.Version = dbs.SQL("version+1")
_, err := this.Save(op)
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,
"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 {
return nil, err
}
@@ -53,7 +53,7 @@ func (this *OriginService) UpdateOrigin(ctx context.Context, req *pb.UpdateOrigi
"portRange": req.Addr.PortRange,
"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 {
return nil, err
}