feat: 点赞相关rabbitMQ

backend/internal/middleware/rabbitmq/likeMQ.go

@@ -0,0 +1,67 @@
+package rabbitmq
+
+import (
+	"context"
+	"errors"
+	"time"
+)
+
+type LikeMQ struct {
+	*RabbitMQ
+}
+
+const (
+	likeExchange   = "like.events"
+	likeQueue      = "like.events"
+	likeBindingKey = "like.*"
+
+	likeLikeRK   = "like.like"
+	likeUnlikeRK = "like.unlike"
+)
+
+type LikeEvent struct {
+	EventID    string    `json:"event_id"`
+	Action     string    `json:"action"`
+	UserID     uint      `json:"user_id"`
+	VideoID    uint      `json:"video_id"`
+	OccurredAt time.Time `json:"occurred_at"`
+}
+
+func NewLikeMQ(base *RabbitMQ) (*LikeMQ, error) {
+	if base == nil {
+		return nil, errors.New("rabbitmq base is nil")
+	}
+	if err := base.DeclareTopic(likeExchange, likeQueue, likeBindingKey); err != nil {
+		return nil, err
+	}
+	return &LikeMQ{RabbitMQ: base}, nil
+}
+
+func (l *LikeMQ) Like(ctx context.Context, userID, videoID uint) error {
+	return l.publish(ctx, "like", likeLikeRK, userID, videoID)
+}
+
+func (l *LikeMQ) Unlike(ctx context.Context, userID, videoID uint) error {
+	return l.publish(ctx, "unlike", likeUnlikeRK, userID, videoID)
+}
+
+func (l *LikeMQ) publish(ctx context.Context, action, routingKey string, userID, videoID uint) error {
+	if l == nil || l.RabbitMQ == nil {
+		return errors.New("like mq is not initialized")
+	}
+	if userID == 0 || videoID == 0 {
+		return errors.New("userID and videoID are required")
+	}
+	id, err := newEventID(16)
+	if err != nil {
+		return err
+	}
+	event := LikeEvent{
+		EventID:    id,
+		Action:     action,
+		UserID:     userID,
+		VideoID:    videoID,
+		OccurredAt: time.Now(),
+	}
+	return l.PublishJSON(ctx, likeExchange, routingKey, event)
+}