expression evaluate

internal/workflow/node-processor/context.go

@@ -0,0 +1,126 @@
+package nodeprocessor
+
+import (
+	"context"
+	"sync"
+)
+
+// 定义上下文键类型，避免键冲突
+type workflowContextKey string
+
+const (
+	nodeOutputsKey workflowContextKey = "node_outputs"
+)
+
+// 带互斥锁的节点输出容器
+type nodeOutputsContainer struct {
+	sync.RWMutex
+	outputs map[string]map[string]any
+}
+
+// 创建新的并发安全的节点输出容器
+func newNodeOutputsContainer() *nodeOutputsContainer {
+	return &nodeOutputsContainer{
+		outputs: make(map[string]map[string]any),
+	}
+}
+
+// 添加节点输出到上下文
+func AddNodeOutput(ctx context.Context, nodeId string, output map[string]any) context.Context {
+	container := getNodeOutputsContainer(ctx)
+	if container == nil {
+		container = newNodeOutputsContainer()
+	}
+
+	container.Lock()
+	defer container.Unlock()
+
+	// 创建输出的深拷贝以避免后续修改
+	outputCopy := make(map[string]any, len(output))
+	for k, v := range output {
+		outputCopy[k] = v
+	}
+
+	container.outputs[nodeId] = outputCopy
+	return context.WithValue(ctx, nodeOutputsKey, container)
+}
+
+// 从上下文获取节点输出
+func GetNodeOutput(ctx context.Context, nodeId string) map[string]any {
+	container := getNodeOutputsContainer(ctx)
+	if container == nil {
+		return nil
+	}
+
+	container.RLock()
+	defer container.RUnlock()
+
+	output, exists := container.outputs[nodeId]
+	if !exists {
+		return nil
+	}
+
+	outputCopy := make(map[string]any, len(output))
+	for k, v := range output {
+		outputCopy[k] = v
+	}
+
+	return outputCopy
+}
+
+// 获取特定节点的特定输出项
+func GetNodeOutputValue(ctx context.Context, nodeId string, key string) (any, bool) {
+	output := GetNodeOutput(ctx, nodeId)
+	if output == nil {
+		return nil, false
+	}
+
+	value, exists := output[key]
+	return value, exists
+}
+
+// 获取所有节点输出
+func GetNodeOutputs(ctx context.Context) map[string]map[string]any {
+	container := getNodeOutputsContainer(ctx)
+	if container == nil {
+		return nil
+	}
+
+	container.RLock()
+	defer container.RUnlock()
+
+	// 创建所有输出的深拷贝
+	allOutputs := make(map[string]map[string]any, len(container.outputs))
+	for nodeId, output := range container.outputs {
+		nodeCopy := make(map[string]any, len(output))
+		for k, v := range output {
+			nodeCopy[k] = v
+		}
+		allOutputs[nodeId] = nodeCopy
+	}
+
+	return allOutputs
+}
+
+// 获取节点输出容器
+func getNodeOutputsContainer(ctx context.Context) *nodeOutputsContainer {
+	value := ctx.Value(nodeOutputsKey)
+	if value == nil {
+		return nil
+	}
+	return value.(*nodeOutputsContainer)
+}
+
+// 检查节点是否有输出
+func HasNodeOutput(ctx context.Context, nodeId string) bool {
+	container := getNodeOutputsContainer(ctx)
+	if container == nil {
+		return false
+	}
+
+	container.RLock()
+	defer container.RUnlock()
+
+	_, exists := container.outputs[nodeId]
+	return exists
+}