Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/batch_image_queue.go

@@ -0,0 +1,64 @@
+package service
+
+import (
+	"context"
+	"net/http"
+	"strings"
+	"time"
+
+	infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
+)
+
+var (
+	ErrBatchImageQueueEmpty          = infraerrors.New(http.StatusNotFound, "BATCH_IMAGE_QUEUE_EMPTY", "batch image queue is empty")
+	ErrBatchImageAlreadyQueued       = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_ALREADY_QUEUED", "batch image job is already queued")
+	ErrBatchImageLockNotAcquired     = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_LOCK_NOT_ACQUIRED", "batch image job lock was not acquired")
+	ErrInvalidBatchImageQueuePayload = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_QUEUE_INVALID_PAYLOAD", "invalid batch image queue payload")
+)
+
+type ReservedBatchImageJob struct {
+	BatchID string
+}
+
+type BatchImageJobLock interface {
+	Release(ctx context.Context) error
+}
+
+type BatchImageQueue interface {
+	Enqueue(ctx context.Context, batchID string) error
+	Reserve(ctx context.Context, blockTimeout time.Duration) (ReservedBatchImageJob, error)
+	RequeueAfter(ctx context.Context, batchID string, delay time.Duration) error
+	Ack(ctx context.Context, batchID string) error
+	Heartbeat(ctx context.Context, batchID string) error
+	MoveDueDelayedToReady(ctx context.Context, limit int) (int, error)
+	RecoverStaleActive(ctx context.Context, staleAfter time.Duration, limit int) (int, error)
+	TryAcquireJobLock(ctx context.Context, batchID string, ttl time.Duration) (BatchImageJobLock, bool, error)
+}
+
+type BatchImageService struct {
+	repo  BatchImageRepository
+	queue BatchImageQueue
+}
+
+func NewBatchImageService(repo BatchImageRepository, queue BatchImageQueue) *BatchImageService {
+	return &BatchImageService{repo: repo, queue: queue}
+}
+
+func (s *BatchImageService) EnqueueBatchImageJob(ctx context.Context, batchID string) error {
+	if !IsValidBatchImageID(batchID) {
+		return ErrInvalidBatchImageQueuePayload
+	}
+	if s == nil || s.queue == nil {
+		return infraerrors.New(http.StatusInternalServerError, "BATCH_IMAGE_QUEUE_NOT_CONFIGURED", "batch image queue is not configured")
+	}
+	if s.repo != nil {
+		if _, err := s.repo.GetBatchImageJobByBatchID(ctx, batchID); err != nil {
+			return err
+		}
+	}
+	return s.queue.Enqueue(ctx, batchID)
+}
+
+func IsValidBatchImageID(batchID string) bool {
+	return strings.HasPrefix(batchID, "imgbatch_") && len(batchID) > len("imgbatch_")
+}