origin/util/blueprint/blueprint.go

package blueprint

import (
	"fmt"
	"sync/atomic"
)

type Blueprint struct {
	execNodes []IExecNode // 注册的定义执行结点

	execPool  ExecPool
	graphPool GraphPool

	blueprintModule IBlueprintModule
	mapGraph        map[int64]IGraph
	seedID          int64
	cancelTimer     func(*uint64) bool
}

func (bm *Blueprint) RegExecNode(execNode IExecNode) {
	bm.execNodes = append(bm.execNodes, execNode)
}

func (bm *Blueprint) regSysNode() {
	bm.RegExecNode(&AddInt{})
	bm.RegExecNode(&SubInt{})
	bm.RegExecNode(&MulInt{})
	bm.RegExecNode(&DivInt{})
	bm.RegExecNode(&ModInt{})
	bm.RegExecNode(&RandNumber{})

	bm.RegExecNode(&Entrance_ArrayParam{})
	bm.RegExecNode(&Entrance_IntParam{})
	bm.RegExecNode(&Entrance_Timer{})
	bm.RegExecNode(&DebugOutput{})
	bm.RegExecNode(&Sequence{})
	bm.RegExecNode(&Foreach{})
	bm.RegExecNode(&ForeachIntArray{})

	bm.RegExecNode(&GetArrayInt{})
	bm.RegExecNode(&GetArrayString{})
	bm.RegExecNode(&GetArrayLen{})
	bm.RegExecNode(&CreateIntArray{})
	bm.RegExecNode(&CreateStringArray{})
	bm.RegExecNode(&AppendIntegerToArray{})
	bm.RegExecNode(&AppendStringToArray{})

	bm.RegExecNode(&BoolIf{})
	bm.RegExecNode(&GreaterThanInteger{})
	bm.RegExecNode(&LessThanInteger{})
	bm.RegExecNode(&EqualInteger{})
	bm.RegExecNode(&RangeCompare{})
	bm.RegExecNode(&EqualSwitch{})
	bm.RegExecNode(&Probability{})
	bm.RegExecNode(&CreateTimer{})
}

func (bm *Blueprint) Init(execDefFilePath string, graphFilePath string, blueprintModule IBlueprintModule, cancelTimer func(*uint64) bool) error {
	// 加载配置结点生成名字对应的innerExecNode
	err := bm.execPool.Load(execDefFilePath)
	if err != nil {
		return err
	}

	// 注册系统执行结点
	bm.regSysNode()

	// 将注册的实际执行结点与innerExecNode进行关联
	for _, e := range bm.execNodes {
		if !bm.execPool.Register(e) {
			return fmt.Errorf("register exec failed,exec:%s", e.GetName())
		}
	}

	// 加载所有的vgf蓝图文件
	err = bm.graphPool.Load(&bm.execPool, graphFilePath, blueprintModule)
	if err != nil {
		return err
	}
	
	bm.cancelTimer = cancelTimer
	bm.blueprintModule = blueprintModule
	bm.mapGraph = make(map[int64]IGraph, 128)

	return nil
}

func (bm *Blueprint) Create(graphName string) int64 {
	if graphName == "" {
		return 0
	}

	graphID := atomic.AddInt64(&bm.seedID, 1)
	gr := bm.graphPool.Create(graphName, graphID)
	if gr == nil {
		return 0
	}

	bm.mapGraph[graphID] = gr
	return graphID
}

func (bm *Blueprint) TriggerEvent(graphID int64, eventID int64, args ...any) error {
	graph := bm.mapGraph[graphID]
	if graph == nil {
		return fmt.Errorf("can not find graph:%d", graphID)
	}

	_, err := graph.Do(eventID, args...)
	return err
}

func (bm *Blueprint) Do(graphID int64, entranceID int64, args ...any) (Port_Array, error) {
	graph := bm.mapGraph[graphID]
	if graph == nil {
		return nil, fmt.Errorf("can not find graph:%d", graphID)
	}

	return graph.Do(entranceID, args...)
}

func (bm *Blueprint) ReleaseGraph(graphID int64) {
	if graphID == 0 {
		return
	}

	defer delete(bm.mapGraph, graphID)
	graph := bm.mapGraph[graphID]
	if graph == nil {
		return
	}

	graph.Release()
}

func (bm *Blueprint) CancelTimerId(graphID int64, timerId *uint64) bool {
	tId := *timerId
	bm.cancelTimer(timerId)

	graph := bm.mapGraph[graphID]
	if graph == nil {
		return false
	}

	gr, ok := graph.(*Graph)
	if !ok {
		return false
	}

	delete(gr.mapTimerID, tId)
	return true
}