@@ -3,6 +3,7 @@ package asyncjob
3
3
import (
4
4
"context"
5
5
"fmt"
6
+ "github.com/Azure/go-asynctask"
6
7
"testing"
7
8
)
8
9
@@ -56,19 +57,20 @@ func (sjb *SqlJobBuilder) BuildJob(bCtx context.Context, errorInjections map[str
56
57
job := NewJob ("sqlSummaryJob" )
57
58
jobLib := & SqlSummaryJobLib {ErrorInjection : errorInjections }
58
59
59
- serverNameParamTask := InputParam (bCtx , job , "param_serverName " , & sjb .ServerName )
60
+ serverNameParamTask := InputParam (bCtx , job , "serverName " , & sjb .ServerName )
60
61
connTsk , _ := StepAfter (bCtx , job , "getConnection" , serverNameParamTask , jobLib .GetConnection )
61
62
62
- table1ParamTsk := InputParam (bCtx , job , "param_table1 " , & sjb .Table1 )
63
+ table1ParamTsk := InputParam (bCtx , job , "table1 " , & sjb .Table1 )
63
64
table1ClientTsk , _ := StepAfterBoth (bCtx , job , "getTableClient1" , connTsk , table1ParamTsk , jobLib .GetTableClient )
64
- query1ParamTsk := InputParam (bCtx , job , "param_query1 " , & sjb .Query1 )
65
+ query1ParamTsk := InputParam (bCtx , job , "query1 " , & sjb .Query1 )
65
66
qery1ResultTsk , _ := StepAfterBoth (bCtx , job , "queryTable1" , table1ClientTsk , query1ParamTsk , jobLib .ExecuteQuery )
66
67
67
- table2ParamTsk := InputParam (bCtx , job , "param_table2 " , & sjb .Table2 )
68
+ table2ParamTsk := InputParam (bCtx , job , "table2 " , & sjb .Table2 )
68
69
table2ClientTsk , _ := StepAfterBoth (bCtx , job , "getTableClient2" , connTsk , table2ParamTsk , jobLib .GetTableClient )
69
- query2ParamTsk := InputParam (bCtx , job , "param_query2 " , & sjb .Query2 )
70
+ query2ParamTsk := InputParam (bCtx , job , "query2 " , & sjb .Query2 )
70
71
qery2ResultTsk , _ := StepAfterBoth (bCtx , job , "queryTable2" , table2ClientTsk , query2ParamTsk , jobLib .ExecuteQuery )
71
72
72
- StepAfterBoth (bCtx , job , "summarize" , qery1ResultTsk , qery2ResultTsk , jobLib .SummarizeQueryResult )
73
+ summaryTsk , _ := StepAfterBoth (bCtx , job , "summarize" , qery1ResultTsk , qery2ResultTsk , jobLib .SummarizeQueryResult )
74
+ AddStep (bCtx , job , "emailNotification" , asynctask .ActionToFunc (jobLib .EmailNotification ), ExecuteAfter (summaryTsk ))
73
75
return job
74
76
}
0 commit comments