You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
2.1 KiB
64 lines
2.1 KiB
package service |
|
|
|
import ( |
|
"context" |
|
|
|
"github.com/jinzhu/gorm" |
|
"go-common/app/admin/main/aegis/model/net" |
|
"go-common/library/log" |
|
) |
|
|
|
func (s *Service) addFlowResources(tx *gorm.DB, netID int64, rids []int64, flowID int64, state int8) (err error) { |
|
fr := &net.FlowResource{ |
|
FlowID: flowID, |
|
State: state, |
|
NetID: netID, |
|
} |
|
for _, rid := range rids { |
|
fr.ID = 0 |
|
fr.RID = rid |
|
if err = s.gorm.AddItem(context.TODO(), tx, fr); err != nil { |
|
log.Error("addFlowResources s.gorm.AddItem error(%v) fr(%+v)", err, fr) |
|
return |
|
} |
|
} |
|
|
|
return |
|
} |
|
|
|
/** |
|
* updateFlowResources 正常流转到新节点 |
|
* 指定网 & 指定资源 & 新节点 |
|
* 不需要指定现状flowid，不同资源可能不在同一现状节点上 |
|
* 已被取消运行的资源现状，不能被更新 |
|
*/ |
|
func (s *Service) updateFlowResources(c context.Context, tx *gorm.DB, netID int64, rid int64, newFlowID int64) (err error) { |
|
var ( |
|
frs []*net.FlowResource |
|
) |
|
|
|
//资源的运行现状 |
|
if frs, err = s.gorm.FRByNetRID(c, []int64{netID}, []int64{rid}, false); err != nil { |
|
log.Error("updateFlowResources s.gorm.FRByNetRID error(%v)", err) |
|
return |
|
} |
|
if len(frs) == 0 { |
|
if err = s.addFlowResources(tx, netID, []int64{rid}, newFlowID, net.FRStateRunning); err != nil { |
|
log.Error("updateFlowResources s.addFlowResources error(%v)", err) |
|
} |
|
return |
|
} |
|
|
|
//确定state & 记录数目：注意单线->并发和并发->单线的转折点 |
|
//todo--只有新节点，如何确定要不要并发拆分或并发合并？--并发要怎么存储（在配置时候确认并发拆分点、并发分支、并发结合点）？ |
|
//todo--需要上游transitionid确认上游dir.order吗？能出现transition->flow的多对多的情况吗（不能）？ |
|
//todo--单独通过flow去确定是在单线上还是并发线上？ |
|
for _, item := range frs { |
|
if err = s.gorm.UpdateFields(context.TODO(), tx, net.TableFlowResource, item.ID, |
|
map[string]interface{}{"flow_id": newFlowID, "state": net.FRStateRunning}); err != nil { |
|
log.Error("updateFlowResources s.gorm.UpdateFields error(%v)", err) |
|
return |
|
} |
|
} |
|
return |
|
}
|
|
|