Function testRouter

examples/workflow_complete/main.go:161–245

测试 3: Router 路由

(ctx context.Context)

// 测试 3: Router 路由
func testRouter(ctx context.Context) {
	// 创建路由目标步骤
	routeA := workflow.NewFunctionStep("route_a", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		return &workflow.StepOutput{Content: "路由A执行", Metadata: make(map[string]any)}, nil
	})

	routeB := workflow.NewFunctionStep("route_b", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		return &workflow.StepOutput{Content: "路由B执行", Metadata: make(map[string]any)}, nil
	})

	finalStep := workflow.NewFunctionStep("final", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		return &workflow.StepOutput{
			Content:  fmt.Sprintf("最终结果: %v", input.PreviousStepContent),
			Metadata: make(map[string]any),
		}, nil
	})

	// 测试 SimpleRouter
	fmt.Println("  测试 SimpleRouter:")
	simpleRouter := workflow.SimpleRouter("simple_router",
		func(input *workflow.StepInput) string {
			if inputStr, ok := input.Input.(string); ok {
				if len(inputStr) > 10 {
					return "route_a"
				}
			}
			return "route_b"
		},
		map[string]workflow.Step{
			"route_a": routeA,
			"route_b": routeB,
		},
	)

	wf1 := workflow.New("SimpleRouterTest").AddStep(simpleRouter)
	input1 := &workflow.WorkflowInput{Input: "short"}

	reader1 := wf1.Execute(ctx, input1)
	for {
		event, err := reader1.Recv()
		if err != nil {
			if errors.Is(err, io.EOF) {
				break
			}
			fmt.Printf("    ❌ 错误: %v\n", err)
			continue
		}
		if event.Type == workflow.EventWorkflowCompleted {
			fmt.Printf("    ✅ SimpleRouter 完成: %v\n",
				event.Data.(map[string]any)["output"])
		}
	}

	// 测试 ChainRouter
	fmt.Println("  测试 ChainRouter:")
	chainRouter := workflow.ChainRouter("chain_router",
		func(input *workflow.StepInput) []string {
			return []string{"route_a", "final"}