feat: separate broker

examples/dag.go

@@ -2,67 +2,33 @@ package main
 
 import (
 	"context"
-	"encoding/json"
-	"io"
-	"net/http"
-
-	"github.com/oarkflow/mq"
+	"fmt"
 	"github.com/oarkflow/mq/dag"
 	"github.com/oarkflow/mq/examples/tasks"
+	"time"
 )
 
 var d *dag.DAG
 
 func main() {
-	d = dag.New(mq.WithTLS(true, "./certs/server.crt", "./certs/server.key"), mq.WithCAPath("./certs/ca.crt"))
-	d.AddNode("queue1", tasks.Node1)
-	d.AddNode("queue2", tasks.Node2)
-	d.AddNode("queue3", tasks.Node3)
-	d.AddNode("queue4", tasks.Node4)
+	d = dag.New()
+	d.AddNode("queue1", tasks.CheckCondition, true)
+	d.AddNode("queue2", tasks.Pass)
+	d.AddNode("queue3", tasks.Fail)
 
-	d.AddEdge("queue1", "queue2")
-	d.AddLoop("queue2", "queue3")
-	d.AddEdge("queue2", "queue4")
+	d.AddCondition("queue1", map[string]string{"pass": "queue2", "fail": "queue3"})
 	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)
-	}
-}
+	go func() {
+		d.Start(context.Background(), ":8081")
+	}()
+	go func() {
+		time.Sleep(3 * time.Second)
+		result := d.Send(context.Background(), []byte(`{"user_id": 1}`))
+		if result.Error != nil {
+			panic(result.Error)
+		}
+		fmt.Println("Response", string(result.Payload))
+	}()
 
-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.MessageID,
-			"payload":    string(rs.Payload),
-			"error":      rs.Error,
-		}
-		json.NewEncoder(w).Encode(result)
-	}
+	time.Sleep(10 * time.Second)
 }