Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-18 18:34:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
boyce
2025-09-23 10:20:03 +08:00
parent 3bcce31a86
commit 77e2986ffb
8 changed files with 181 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
util/blueprint/node.go
View File

@@ -37,7 +37,7 @@ func (en *execNode) Next() *execNode {
return en.nextNode[en.nextIdx]
}
func (en *execNode) exec(gr *graph) (int, error) {
func (en *execNode) exec(gr *Graph) (int, error) {
e, ok := en.execNode.(IExecNode)
if !ok {
return -1, fmt.Errorf("exec node %s not exec", en.execNode.GetName())
@@ -55,7 +55,7 @@ func (en *execNode) exec(gr *graph) (int, error) {
return e.Exec()
}
func (en *execNode) doSetInPort(gr *graph, index int, inPort IPort) error {
func (en *execNode) doSetInPort(gr *Graph, index int, inPort IPort) error {
// 找到当前Node的InPort的index的前一个结点
preNode := en.preInPort[index]
// 如果前一个结点为空，则填充默认值
@@ -79,10 +79,10 @@ func (en *execNode) doSetInPort(gr *graph, index int, inPort IPort) error {
}
// 如果前一个结点没有执行过，则递归执行前一个结点
return preNode.node.Do(gr)
return preNode.node.Do(gr, nil)
}
func (en *execNode) Do(gr *graph) error {
func (en *execNode) Do(gr *Graph, outPortArgs ...any) error {
// 重新初始化上下文
inPorts, outPorts := en.execNode.CloneInOutPort()
gr.context[en.Id] = &ExecContext{
@@ -90,6 +90,16 @@ func (en *execNode) Do(gr *graph) error {
OutputPorts: outPorts,
}
for i := 0; i < len(outPortArgs); i++ {
if i >= len(outPorts) {
return fmt.Errorf("args %d not found in node %s", i, en.execNode.GetName())
}
if err := outPorts[i].setAnyVale(outPortArgs[i]); err != nil {
return fmt.Errorf("args %d set value error: %w", i, err)
}
}
// 处理InPort结点值
var err error
for index := range inPorts {
@@ -111,7 +121,7 @@ func (en *execNode) Do(gr *graph) error {
return err
}
if nextIndex == -1 {
if nextIndex == -1 || en.nextNode == nil {
return nil
}