mirror of
https://github.com/aceld/kis-flow.git
synced 2025-01-23 07:30:23 +08:00
116 lines
2.4 KiB
Go
116 lines
2.4 KiB
Go
package test
|
||
|
||
import (
|
||
"context"
|
||
"github.com/aceld/kis-flow/common"
|
||
"github.com/aceld/kis-flow/config"
|
||
"github.com/aceld/kis-flow/file"
|
||
"github.com/aceld/kis-flow/flow"
|
||
"github.com/aceld/kis-flow/kis"
|
||
"github.com/aceld/kis-flow/test/faas"
|
||
"github.com/aceld/kis-flow/test/proto"
|
||
"testing"
|
||
)
|
||
|
||
func TestAutoInjectParamWithConfig(t *testing.T) {
|
||
ctx := context.Background()
|
||
|
||
kis.Pool().FaaS("AvgStuScore", faas.AvgStuScore)
|
||
kis.Pool().FaaS("PrintStuAvgScore", faas.PrintStuAvgScore)
|
||
|
||
// 1. 加载配置文件并构建Flow
|
||
if err := file.ConfigImportYaml("load_conf/"); err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
// 2. 获取Flow
|
||
flow1 := kis.Pool().GetFlow("StuAvg")
|
||
if flow1 == nil {
|
||
panic("flow1 is nil")
|
||
}
|
||
|
||
// 3. 提交原始数据
|
||
_ = flow1.CommitRow(&faas.AvgStuScoreIn{
|
||
StuScores: proto.StuScores{
|
||
StuId: 100,
|
||
Score1: 1,
|
||
Score2: 2,
|
||
Score3: 3,
|
||
},
|
||
})
|
||
_ = flow1.CommitRow(faas.AvgStuScoreIn{
|
||
StuScores: proto.StuScores{
|
||
StuId: 100,
|
||
Score1: 1,
|
||
Score2: 2,
|
||
Score3: 3,
|
||
},
|
||
})
|
||
|
||
// 提交原始数据（json字符串）
|
||
_ = flow1.CommitRow(`{"stu_id":101}`)
|
||
|
||
// 4. 执行flow1
|
||
if err := flow1.Run(ctx); err != nil {
|
||
panic(err)
|
||
}
|
||
}
|
||
|
||
func TestAutoInjectParam(t *testing.T) {
|
||
ctx := context.Background()
|
||
|
||
kis.Pool().FaaS("AvgStuScore", faas.AvgStuScore)
|
||
kis.Pool().FaaS("PrintStuAvgScore", faas.PrintStuAvgScore)
|
||
|
||
source1 := config.KisSource{
|
||
Name: "Test",
|
||
Must: []string{},
|
||
}
|
||
|
||
avgStuScoreConfig := config.NewFuncConfig("AvgStuScore", common.C, &source1, nil)
|
||
if avgStuScoreConfig == nil {
|
||
panic("AvgStuScore is nil")
|
||
}
|
||
|
||
printStuAvgScoreConfig := config.NewFuncConfig("PrintStuAvgScore", common.C, &source1, nil)
|
||
if printStuAvgScoreConfig == nil {
|
||
panic("printStuAvgScoreConfig is nil")
|
||
}
|
||
|
||
myFlowConfig1 := config.NewFlowConfig("cal_stu_avg_score", common.FlowEnable)
|
||
|
||
flow1 := flow.NewKisFlow(myFlowConfig1)
|
||
|
||
// 4. 拼接Functioin 到 Flow 上
|
||
if err := flow1.Link(avgStuScoreConfig, nil); err != nil {
|
||
panic(err)
|
||
}
|
||
if err := flow1.Link(printStuAvgScoreConfig, nil); err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
// 3. 提交原始数据
|
||
_ = flow1.CommitRow(&faas.AvgStuScoreIn{
|
||
StuScores: proto.StuScores{
|
||
StuId: 100,
|
||
Score1: 1,
|
||
Score2: 2,
|
||
Score3: 3,
|
||
},
|
||
})
|
||
_ = flow1.CommitRow(`{"stu_id":101}`)
|
||
_ = flow1.CommitRow(faas.AvgStuScoreIn{
|
||
StuScores: proto.StuScores{
|
||
StuId: 100,
|
||
Score1: 1,
|
||
Score2: 2,
|
||
Score3: 3,
|
||
},
|
||
})
|
||
|
||
// 4. 执行flow1
|
||
if err := flow1.Run(ctx); err != nil {
|
||
panic(err)
|
||
}
|
||
}
|