package/mq
1
0
Fork 0
mirror of https://github.com/oarkflow/mq.git synced 2025-09-27 04:15:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update: dependencies

Browse Source
This commit is contained in:
Oarkflow
2024-11-23 10:51:22 +05:45
parent 6b87422a1a
commit 1c84e18d0c
30 changed files with 2291 additions and 2296 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
examples/dag.go
View File

@@ -4,14 +4,13 @@ import (
"context"
"encoding/json"
"fmt"
v2 "github.com/oarkflow/mq/dag/v2"
"github.com/oarkflow/mq"
"github.com/oarkflow/mq/dag"
"github.com/oarkflow/mq/examples/tasks"
)
func main() {
f := v2.NewDAG("Sample DAG", "sample-dag", func(taskID string, result mq.Result) {
f := dag.NewDAG("Sample DAG", "sample-dag", func(taskID string, result mq.Result) {
fmt.Printf("Final result for task %s: %s\n", taskID, string(result.Payload))
}, mq.WithSyncMode(true))
f.SetNotifyResponse(func(ctx context.Context, result mq.Result) error {
@@ -25,39 +24,38 @@ func main() {
if err != nil {
panic(err)
}
f.Start(context.Background(), ":8083")
f.Start(context.Background(), ":8082")
sendData(f)
}
func subDAG() *v2.DAG {
f := v2.NewDAG("Sub DAG", "sub-dag", func(taskID string, result mq.Result) {
func subDAG() *dag.DAG {
f := dag.NewDAG("Sub DAG", "sub-dag", func(taskID string, result mq.Result) {
fmt.Printf("Final result for task %s: %s\n", taskID, string(result.Payload))
}, mq.WithSyncMode(true))
f.
AddNode(v2.Function, "Store data", "store:data", &tasks.StoreData{Operation: v2.Operation{Type: "process"}}, true).
AddNode(v2.Function, "Send SMS", "send:sms", &tasks.SendSms{Operation: v2.Operation{Type: "process"}}).
AddNode(v2.Function, "Notification", "notification", &tasks.InAppNotification{Operation: v2.Operation{Type: "process"}}).
AddEdge(v2.Simple, "Store Payload to send sms", "store:data", "send:sms").
AddEdge(v2.Simple, "Store Payload to notification", "send:sms", "notification")
AddNode(dag.Function, "Store data", "store:data", &tasks.StoreData{Operation: dag.Operation{Type: "process"}}, true).
AddNode(dag.Function, "Send SMS", "send:sms", &tasks.SendSms{Operation: dag.Operation{Type: "process"}}).
AddNode(dag.Function, "Notification", "notification", &tasks.InAppNotification{Operation: dag.Operation{Type: "process"}}).
AddEdge(dag.Simple, "Store Payload to send sms", "store:data", "send:sms").
AddEdge(dag.Simple, "Store Payload to notification", "send:sms", "notification")
return f
}
func setup(f *v2.DAG) {
func setup(f *dag.DAG) {
f.
AddNode(v2.Function, "Email Delivery", "email:deliver", &tasks.EmailDelivery{Operation: v2.Operation{Type: "process"}}).
AddNode(v2.Function, "Prepare Email", "prepare:email", &tasks.PrepareEmail{Operation: v2.Operation{Type: "process"}}).
AddNode(v2.Function, "Get Input", "get:input", &tasks.GetData{Operation: v2.Operation{Type: "input"}}, true).
AddNode(v2.Function, "Final Payload", "final", &tasks.Final{Operation: v2.Operation{Type: "page"}}).
AddNode(v2.Function, "Iterator Processor", "loop", &tasks.Loop{Operation: v2.Operation{Type: "loop"}}).
AddNode(v2.Function, "Condition", "condition", &tasks.Condition{Operation: v2.Operation{Type: "condition"}}).
AddNode(dag.Function, "Email Delivery", "email:deliver", &tasks.EmailDelivery{Operation: dag.Operation{Type: "process"}}).
AddNode(dag.Function, "Prepare Email", "prepare:email", &tasks.PrepareEmail{Operation: dag.Operation{Type: "process"}}).
AddNode(dag.Function, "Get Input", "get:input", &tasks.GetData{Operation: dag.Operation{Type: "input"}}, true).
AddNode(dag.Function, "Iterator Processor", "loop", &tasks.Loop{Operation: dag.Operation{Type: "loop"}}).
AddNode(dag.Function, "Condition", "condition", &tasks.Condition{Operation: dag.Operation{Type: "condition"}}).
AddDAGNode("Persistent", "persistent", subDAG()).
AddEdge(v2.Simple, "Get input to loop", "get:input", "loop").
AddEdge(v2.Iterator, "Loop to prepare email", "loop", "prepare:email").
AddEdge(v2.Simple, "Prepare Email to condition", "prepare:email", "condition").
AddEdge(dag.Simple, "Get input to loop", "get:input", "loop").
AddEdge(dag.Iterator, "Loop to prepare email", "loop", "prepare:email").
AddEdge(dag.Simple, "Prepare Email to condition", "prepare:email", "condition").
AddCondition("condition", map[string]string{"pass": "email:deliver", "fail": "persistent"})
}
func sendData(f *v2.DAG) {
func sendData(f *dag.DAG) {
data := []map[string]any{
{"phone": "+123456789", "email": "abc.xyz@gmail.com"}, {"phone": "+98765412", "email": "xyz.abc@gmail.com"},
}