Initial commit: DeepRes Go backend — bot + worker + embed service

internal/queue/queue.go

@@ -0,0 +1,63 @@
+package queue
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"github.com/emil/deepres/internal/models"
+	"github.com/nats-io/nats.go"
+)
+
+type Queue struct {
+	conn *nats.Conn
+}
+
+func New(url string) (*Queue, error) {
+	conn, err := nats.Connect(url, nats.Timeout(5*time.Second))
+	if err != nil {
+		return nil, fmt.Errorf("connect to nats: %w", err)
+	}
+	return &Queue{conn: conn}, nil
+}
+
+func (q *Queue) Close() { q.conn.Close() }
+
+func (q *Queue) PublishFast(ctx context.Context, req *models.ResearchRequest) error {
+	data, err := json.Marshal(req)
+	if err != nil {
+		return fmt.Errorf("marshal fast request: %w", err)
+	}
+	return q.conn.Publish("research.fast", data)
+}
+
+func (q *Queue) PublishDeep(ctx context.Context, req *models.ResearchRequest) error {
+	data, err := json.Marshal(req)
+	if err != nil {
+		return fmt.Errorf("marshal deep request: %w", err)
+	}
+	return q.conn.Publish("research.deep", data)
+}
+
+func (q *Queue) SubscribeFast(handler func(*models.ResearchRequest)) (*nats.Subscription, error) {
+	sub, err := q.conn.QueueSubscribe("research.fast", "workers", func(msg *nats.Msg) {
+		var req models.ResearchRequest
+		if err := json.Unmarshal(msg.Data, &req); err != nil {
+			return
+		}
+		handler(&req)
+	})
+	return sub, err
+}
+
+func (q *Queue) SubscribeDeep(handler func(*models.ResearchRequest)) (*nats.Subscription, error) {
+	sub, err := q.conn.QueueSubscribe("research.deep", "workers", func(msg *nats.Msg) {
+		var req models.ResearchRequest
+		if err := json.Unmarshal(msg.Data, &req); err != nil {
+			return
+		}
+		handler(&req)
+	})
+	return sub, err
+}