65 lines
2.1 KiB
Go
65 lines
2.1 KiB
Go
|
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
|
|||
|
|
}
|