65 lines
2.3 KiB
Go
65 lines
2.3 KiB
Go
|
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_")
|
||
|
|
}
|