mirror of
https://github.com/oarkflow/mq.git
synced 2025-09-27 04:15:52 +08:00
77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/oarkflow/mq"
|
|
"github.com/oarkflow/mq/dag"
|
|
"github.com/oarkflow/mq/examples/tasks"
|
|
)
|
|
|
|
func main() {
|
|
aSync()
|
|
}
|
|
|
|
func subDAG() *dag.DAG {
|
|
f := dag.NewDAG("Sub DAG", "sub-dag", mq.WithCleanTaskOnComplete(), mq.WithSyncMode(true))
|
|
f.
|
|
AddNode("Store data", "store:data", &tasks.StoreData{Operation: dag.Operation{Type: "process"}}, true).
|
|
AddNode("Send SMS", "send:sms", &tasks.SendSms{Operation: dag.Operation{Type: "process"}}).
|
|
AddNode("Notification", "notification", &tasks.InAppNotification{Operation: dag.Operation{Type: "process"}}).
|
|
AddEdge("Store Data to send sms and notification", "store:data", "send:sms", "notification")
|
|
return f
|
|
}
|
|
|
|
func setup(f *dag.DAG) {
|
|
f.
|
|
AddNode("Email Delivery", "email:deliver", &tasks.EmailDelivery{Operation: dag.Operation{Type: "process"}}).
|
|
AddNode("Prepare Email", "prepare:email", &tasks.PrepareEmail{Operation: dag.Operation{Type: "process"}}).
|
|
AddNode("Get Input", "get:input", &tasks.GetData{Operation: dag.Operation{Type: "input"}}, true).
|
|
AddNode("Iterator Processor", "loop", &tasks.Loop{Operation: dag.Operation{Type: "loop"}}).
|
|
AddNode("Condition", "condition", &tasks.Condition{Operation: dag.Operation{Type: "condition"}}).
|
|
AddDAGNode("Persistent", "persistent", subDAG()).
|
|
AddCondition("condition", map[dag.When]dag.Then{"pass": "email:deliver", "fail": "persistent"}).
|
|
AddEdge("Get input to loop", "get:input", "loop").
|
|
AddIterator("Loop to prepare email", "loop", "prepare:email").
|
|
AddEdge("Prepare Email to condition", "prepare:email", "condition")
|
|
}
|
|
|
|
func sendData(f *dag.DAG) {
|
|
data := []map[string]any{
|
|
{"phone": "+123456789", "email": "abc.xyz@gmail.com"}, {"phone": "+98765412", "email": "xyz.abc@gmail.com"},
|
|
}
|
|
bt, _ := json.Marshal(data)
|
|
result := f.Process(context.Background(), bt)
|
|
fmt.Println(string(result.Payload))
|
|
}
|
|
|
|
func aSync() {
|
|
f := dag.NewDAG("Sample DAG", "sample-dag", mq.WithCleanTaskOnComplete())
|
|
|
|
f.SetNotifyResponse(func(ctx context.Context, result mq.Result) error {
|
|
if f.Notifier != nil {
|
|
f.Notifier.ToRoom("global", "final-message", result)
|
|
}
|
|
return nil
|
|
})
|
|
f.ReportNodeResult(func(result mq.Result) {
|
|
if f.Notifier != nil {
|
|
f.Notifier.ToRoom("global", "message", result)
|
|
}
|
|
log.Printf("DAG - FINAL_RESPONSE ~> TaskID: %s, Payload: %s, Topic: %s, Error: %v, Latency: %s", result.TaskID, result.Payload, result.Topic, result.Error, result.Latency)
|
|
})
|
|
setup(f)
|
|
err := f.Validate()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = f.Start(context.TODO(), ":8083")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|