LightFlow：一个基于函数式编程的任务编排框架

go get github.com/Bilibotter/light-flow/flow

package main

import (
    "fmt"
    "github.com/Bilibotter/light-flow/flow"
)

func First(step flow.Step) (any, error) {
    if input, exist := step.Get("input"); exist {
        fmt.Printf("[Step: %s] get input: %v\n", step.Name(), input)
    }
    step.Set("key", "value")
    return "result", nil
}

func Second(step flow.Step) (any, error) {
    if value, exist := step.Get("key"); exist {
        fmt.Printf("[Step: %s] get key: %v\n", step.Name(), value)
    }
    if result, exist := step.Result(step.Dependents()[0]); exist {
        fmt.Printf("[Step: %s] get result: %v\n", step.Name(), result)
    }
    return nil, nil
}

func ErrorStep(step flow.Step) (any, error) {
    if value, exist := step.Get("key"); exist {
        fmt.Printf("[Step: %s] get key: %v\n", step.Name(), value)
    } else {
        fmt.Printf("[Step: %s] cannot get key \n", step.Name())
    }
    return nil, fmt.Errorf("execute failed")
}

func ErrorHandler(step flow.Step) (bool, error) {
    if step.Has(flow.Failed) {
        fmt.Printf("[Step: %s] has failed\n", step.Name())
    } else {
        fmt.Printf("[Step: %s] success\n", step.Name())
    }
    return true, nil
}

func init() {
    process := flow.FlowWithProcess("Example")
    process.Follow(First, Second)
    process.Follow(ErrorStep)
    process.AfterStep(true, ErrorHandler)
}

func main() {
    flow.DoneFlow("Example", map[string]any{"input": "Hello world"})
}

[Step: First] get input: Hello world
[Step: First] success
[Step: Second] get key: value
[Step: Second] get result: result
[Step: Second] success
[Step: ErrorStep] cannot get key 
[Step: ErrorStep] has failed