feat: separate broker

This commit is contained in:
Oarkflow
2024-09-26 19:10:36 +05:45
parent 5db7f5706a
commit 4a67eeefe0
6 changed files with 188 additions and 171 deletions

View File

@@ -3,19 +3,18 @@ package main
import (
"context"
"fmt"
"github.com/oarkflow/mq/broker"
"github.com/oarkflow/mq"
)
func main() {
consumer := broker.NewConsumer(":8080")
consumer.RegisterHandler("queue1", func(ctx context.Context, task broker.Task) broker.Result {
consumer := mq.NewConsumer(":8080")
consumer.RegisterHandler("queue1", func(ctx context.Context, task mq.Task) mq.Result {
fmt.Println("Handling task for queue1:", task.ID)
return broker.Result{Payload: task.Payload, MessageID: task.ID}
return mq.Result{Payload: task.Payload, MessageID: task.ID}
})
consumer.RegisterHandler("queue2", func(ctx context.Context, task broker.Task) broker.Result {
consumer.RegisterHandler("queue2", func(ctx context.Context, task mq.Task) mq.Result {
fmt.Println("Handling task for queue2:", task.ID)
return broker.Result{Payload: task.Payload, MessageID: task.ID}
return mq.Result{Payload: task.Payload, MessageID: task.ID}
})
consumer.Consume(context.Background(), "queue2", "queue1")
}