mirror of
https://github.com/oarkflow/mq.git
synced 2025-10-05 07:57:00 +08:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/oarkflow/mq"
|
|
"github.com/oarkflow/mq/dag"
|
|
"github.com/oarkflow/mq/examples/tasks"
|
|
)
|
|
|
|
var d *dag.DAG
|
|
|
|
func main() {
|
|
d = dag.New(mq.WithSyncMode(true), mq.WithTLS(true, "./certs/server.crt", "./certs/server.key"), mq.WithCAPath("./certs/ca.crt"))
|
|
d.AddNode("queue1", tasks.Node1, true)
|
|
d.AddNode("queue2", tasks.Node2)
|
|
d.AddNode("queue3", tasks.Node3)
|
|
d.AddNode("queue4", tasks.Node4)
|
|
|
|
d.AddNode("queue5", tasks.CheckCondition)
|
|
d.AddNode("queue6", tasks.Pass)
|
|
d.AddNode("queue7", tasks.Fail)
|
|
|
|
d.AddCondition("queue5", map[string]string{"pass": "queue6", "fail": "queue7"})
|
|
d.AddEdge("queue1", "queue2")
|
|
d.AddEdge("queue2", "queue4")
|
|
d.AddEdge("queue3", "queue5")
|
|
|
|
d.AddLoop("queue2", "queue3")
|
|
d.Prepare()
|
|
go func() {
|
|
d.Start(context.Background(), ":8081")
|
|
}()
|
|
go func() {
|
|
time.Sleep(3 * time.Second)
|
|
result := d.Send(context.Background(), []byte(`[{"user_id": 1}, {"user_id": 2}]`))
|
|
if result.Error != nil {
|
|
panic(result.Error)
|
|
}
|
|
fmt.Println("Response", string(result.Payload))
|
|
}()
|
|
|
|
time.Sleep(10 * time.Second)
|
|
/*d.Prepare()
|
|
http.HandleFunc("POST /publish", requestHandler("publish"))
|
|
http.HandleFunc("POST /request", requestHandler("request"))
|
|
err := d.Start(context.TODO(), ":8083")
|
|
if err != nil {
|
|
panic(err)
|
|
}*/
|
|
}
|
|
|
|
func requestHandler(requestType string) func(w http.ResponseWriter, r *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, "Invalid request method", http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
var payload []byte
|
|
if r.Body != nil {
|
|
defer r.Body.Close()
|
|
var err error
|
|
payload, err = io.ReadAll(r.Body)
|
|
if err != nil {
|
|
http.Error(w, "Failed to read request body", http.StatusBadRequest)
|
|
return
|
|
}
|
|
} else {
|
|
http.Error(w, "Empty request body", http.StatusBadRequest)
|
|
return
|
|
}
|
|
var rs mq.Result
|
|
if requestType == "request" {
|
|
rs = d.Request(context.Background(), payload)
|
|
} else {
|
|
rs = d.Send(context.Background(), payload)
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
result := map[string]any{
|
|
"message_id": rs.TaskID,
|
|
"payload": string(rs.Payload),
|
|
"error": rs.Error,
|
|
}
|
|
json.NewEncoder(w).Encode(result)
|
|
}
|
|
}
|