mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-09-24 16:05:44 +08:00
shaw
80a229bce5
修复 PR #3768 批量图像 MVP 合并后审计报告中的全部问题： 结算与计费（高危）： - 所有 SETTLEMENT_* 失败（超冻结/计数非法/manifest 冲突/定价缺失/扣费失败） 统一计入 retry_count 并在耗尽时释放冻结转 failed，消灭 settling 无限 requeue 导致的冻结余额永久锁死 - 耗尽出口的释放指纹统一为 RequestHash，与 processor/Cancel/recovery 一致； release 遇同 request id 指纹冲突视为幂等成功，治愈历史毒消息 - 管理端校验 hold_multiplier >= discount_multiplier，定价快照对存量脏数据钳制 - 释放前校验 per-job hold claim（dedup+归档表），杜绝幻影释放 索引对账（高危）： - provider 输出与提交 custom_id 集对账：未知条目丢弃并记事件， 漏项补 PROVIDER_RESULT_MISSING 失败行，保证 success+fail == item_count 提交与恢复（高危）： - 提交前转 uploading 并在 provider.Submit 期间心跳刷新 updated_at； 恢复扫描改为原子复核（FailStaleUnsubmittedBatchImageJob）， 消灭慢提交被误杀退款而上游任务照常计费的孤儿场景 - 上游任务创建成功但本地状态推进失败时，尽力取消上游并清理输入 - recovery 释放失败时入队交由 worker releaseTerminalHold 兜底重试 队列与并发（中危）： - Enqueue（SetNX+LPush）与 Reserve（BRPop+ZAdd）均改为 Lua 原子脚本， 消灭崩溃窗口导致 job 脱离队列、被 7 天 inflight 键锁死 - 锁冲突按 LockConflictDelay 重新入队（原直接丢弃需等 10 分钟 stale 恢复） - 处理期间心跳：active zset 续期（ZAddXX 防幽灵成员）+ 锁 TTL 续期 - ReplaceBatchImageItemsForJob 增加 indexing 状态守卫，防掉队 worker 重写账目 存量回归（中危）： - image-only 定价条目（仅图片价无 token 价）恢复 token 计费 fail-closed， 不再按 $0 计费；图片计费路径不受影响 - 鉴权余额门槛恢复 balance <= 0 语义，MinimumBalanceReserve 不再作硬 403 加固： - ZIP max_items 钳制到管理员上限；Submit 补齐 Platform==Gemini 校验； gemini downloadUri 跟随前做 host 白名单校验 - 批量客户端改用共享 httpclient（拨号/TLS/响应头超时有界） - 审计点名的忽略错误（MarkDownloaded/SettlementFailed/AppendEvent 等）改为记日志
415 lines
20 KiB
Go
415 lines
20 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"crypto/rand"
|
||
"encoding/hex"
|
||
"net/http"
|
||
"time"
|
||
|
||
infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
|
||
)
|
||
|
||
const (
|
||
BatchImageProviderGeminiAPI = "gemini_api"
|
||
BatchImageProviderVertex = "vertex"
|
||
)
|
||
|
||
const (
|
||
BatchImageJobStatusCreated = "created"
|
||
BatchImageJobStatusUploading = "uploading"
|
||
BatchImageJobStatusSubmitted = "submitted"
|
||
BatchImageJobStatusRunning = "running"
|
||
BatchImageJobStatusIndexing = "indexing"
|
||
BatchImageJobStatusSettling = "settling"
|
||
BatchImageJobStatusCompleted = "completed"
|
||
BatchImageJobStatusFailed = "failed"
|
||
BatchImageJobStatusCancelled = "cancelled"
|
||
BatchImageJobStatusOutputDeleted = "output_deleted"
|
||
)
|
||
|
||
const (
|
||
BatchImageItemStatusPending = "pending"
|
||
BatchImageItemStatusSuccess = "success"
|
||
BatchImageItemStatusFailed = "failed"
|
||
BatchImageItemStatusCancelled = "cancelled"
|
||
)
|
||
|
||
var (
|
||
ErrBatchImageJobNotFound = infraerrors.New(http.StatusNotFound, "BATCH_IMAGE_JOB_NOT_FOUND", "batch image job not found")
|
||
ErrBatchImageJobExists = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_JOB_EXISTS", "batch image job already exists")
|
||
ErrBatchImageItemExists = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_ITEM_EXISTS", "batch image item already exists")
|
||
|
||
ErrBatchImageInvalidTransition = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_INVALID_TRANSITION", "invalid batch image job status transition")
|
||
ErrBatchImageInvalidProvider = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_INVALID_PROVIDER", "invalid batch image provider")
|
||
|
||
ErrBatchImageMissingProviderJobName = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_MISSING_PROVIDER_JOB_NAME", "batch image provider job name is missing")
|
||
ErrBatchImageMissingAccountID = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_MISSING_ACCOUNT_ID", "batch image account id is missing")
|
||
ErrBatchImageUnsupportedProvider = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_UNSUPPORTED_PROVIDER", "unsupported batch image provider")
|
||
ErrBatchImageIndexOutputMissing = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_INDEX_OUTPUT_MISSING", "batch image provider output is missing")
|
||
ErrBatchImageIndexParseFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_INDEX_PARSE_FAILED", "batch image provider output parse failed")
|
||
ErrBatchImageIndexNoResultLines = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_INDEX_NO_RESULT_LINES", "batch image provider output has no result lines")
|
||
ErrBatchImageDuplicateCustomID = infraerrors.New(http.StatusBadGateway, "DUPLICATE_CUSTOM_ID_IN_OUTPUT", "batch image provider output contains duplicate custom id")
|
||
ErrBatchImageIndexStateConflict = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_INDEX_STATE_CONFLICT", "batch image job is no longer in indexing state")
|
||
|
||
ErrBatchImageSettlementInvalidStatus = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_SETTLEMENT_INVALID_STATUS", "batch image job is not ready for settlement")
|
||
ErrBatchImageSettlementManifestConflict = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_SETTLEMENT_MANIFEST_CONFLICT", "batch image settlement manifest hash conflict")
|
||
ErrBatchImageSettlementPricingMissing = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_SETTLEMENT_PRICING_MISSING", "batch image settlement pricing is missing")
|
||
ErrBatchImageSettlementBillingFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_SETTLEMENT_BILLING_FAILED", "batch image settlement billing failed")
|
||
ErrBatchImageAlreadySettled = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_ALREADY_SETTLED", "batch image job is already settled")
|
||
ErrBatchImageSettlementMissingAPIKeyID = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_SETTLEMENT_MISSING_API_KEY_ID", "batch image settlement api key id is missing")
|
||
ErrBatchImageSettlementMissingAccountID = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_SETTLEMENT_MISSING_ACCOUNT_ID", "batch image settlement account id is missing")
|
||
ErrBatchImageSettlementInvalidCounts = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_SETTLEMENT_INVALID_COUNTS", "batch image settlement counts are invalid")
|
||
ErrBatchImageSettlementCostExceedsHold = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_SETTLEMENT_COST_EXCEEDS_HOLD", "batch image settlement cost exceeds held balance")
|
||
ErrBatchImageBillingHoldFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_BILLING_HOLD_FAILED", "batch image balance hold failed")
|
||
ErrBatchImageInsufficientBalance = infraerrors.New(http.StatusPaymentRequired, "BATCH_IMAGE_INSUFFICIENT_BALANCE", "insufficient balance for batch image hold")
|
||
|
||
ErrBatchImageDisabled = infraerrors.New(http.StatusNotFound, "BATCH_IMAGE_DISABLED", "batch image API is disabled")
|
||
ErrBatchImageGroupDisabled = infraerrors.New(http.StatusForbidden, "BATCH_IMAGE_GROUP_DISABLED", "batch image API is disabled for this group")
|
||
ErrBatchImageInvalidModel = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_INVALID_MODEL", "batch image model is required")
|
||
ErrBatchImageNoAccountAvailable = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_NO_ACCOUNT_AVAILABLE", "no compatible batch image account is available")
|
||
ErrBatchImageInvalidItems = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_INVALID_ITEMS", "batch image items are invalid")
|
||
ErrBatchImageDuplicateCustomIDInRequest = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_DUPLICATE_CUSTOM_ID", "batch image custom ids must be unique")
|
||
ErrBatchImagePromptTooLong = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_PROMPT_TOO_LONG", "batch image prompt is too long")
|
||
ErrBatchImageInvalidReferenceImage = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_INVALID_REFERENCE_IMAGE", "batch image reference image is invalid")
|
||
ErrBatchImageTooManyReferenceImages = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_TOO_MANY_REFERENCE_IMAGES", "too many batch image reference images for this model")
|
||
ErrBatchImageReferenceImagesTooLarge = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_REFERENCE_IMAGES_TOO_LARGE", "batch image reference images are too large")
|
||
ErrBatchImageTooManyOutputImages = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_TOO_MANY_OUTPUT_IMAGES", "too many batch image output images")
|
||
ErrBatchImageProviderSubmitFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_PROVIDER_SUBMIT_FAILED", "batch image provider submit failed")
|
||
ErrBatchImageQueueFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_QUEUE_FAILED", "batch image queue failed")
|
||
ErrBatchImageIdempotencyConflict = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_IDEMPOTENCY_CONFLICT", "idempotency key reused with different batch image request")
|
||
ErrBatchImageCancelFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_CANCEL_FAILED", "batch image cancel failed")
|
||
ErrBatchImageVertexGCSBucketMissing = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_VERTEX_GCS_BUCKET_MISSING", "Vertex managed GCS bucket is not configured")
|
||
|
||
ErrBatchImageNotReady = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_NOT_READY", "batch image job is not completed")
|
||
ErrBatchImageOutputDeleted = infraerrors.New(http.StatusGone, "BATCH_IMAGE_OUTPUT_DELETED", "batch image output has been deleted")
|
||
ErrBatchImageItemNotFound = infraerrors.New(http.StatusNotFound, "BATCH_IMAGE_ITEM_NOT_FOUND", "batch image item not found")
|
||
ErrBatchImageItemFailed = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_ITEM_FAILED", "batch image item did not succeed")
|
||
ErrBatchImageResultMissing = infraerrors.New(http.StatusInternalServerError, "BATCH_IMAGE_RESULT_MISSING", "batch image result is missing")
|
||
ErrBatchImageDownloadLimited = infraerrors.New(http.StatusTooManyRequests, "BATCH_IMAGE_DOWNLOAD_LIMITED", "too many batch image downloads")
|
||
ErrBatchImageDownloadFailed = infraerrors.New(http.StatusInternalServerError, "BATCH_IMAGE_DOWNLOAD_FAILED", "batch image download failed")
|
||
ErrBatchImageDownloadTooLarge = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_DOWNLOAD_TOO_LARGE", "batch image download is too large")
|
||
ErrBatchImageItemImageIndexOutOfRange = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_ITEM_IMAGE_INDEX_OUT_OF_RANGE", "batch image item image index is out of range")
|
||
ErrBatchImageZipTooManyItems = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_ZIP_TOO_MANY_ITEMS", "batch image ZIP contains too many items; use single item downloads")
|
||
ErrBatchImageOutputDeleteNotReady = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_OUTPUT_DELETE_NOT_READY", "batch image output can only be deleted after completion")
|
||
ErrBatchImageRecordDeleteNotReady = infraerrors.New(http.StatusConflict, "BATCH_IMAGE_RECORD_DELETE_NOT_READY", "batch image record can only be deleted after the job finishes")
|
||
ErrBatchImageCleanupFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_CLEANUP_FAILED", "batch image cleanup failed")
|
||
ErrBatchImageCleanupUnsafePath = infraerrors.New(http.StatusBadRequest, "BATCH_IMAGE_CLEANUP_UNSAFE_PATH", "batch image cleanup path is unsafe")
|
||
ErrBatchImageProviderCleanupFailed = infraerrors.New(http.StatusBadGateway, "BATCH_IMAGE_PROVIDER_CLEANUP_FAILED", "batch image provider cleanup failed")
|
||
)
|
||
|
||
type BatchImageJob struct {
|
||
ID int64
|
||
BatchID string
|
||
UserID int64
|
||
APIKeyID *int64
|
||
AccountID *int64
|
||
Provider string
|
||
Model string
|
||
TaskName string
|
||
ParentBatchID *string
|
||
Status string
|
||
ProviderJobName *string
|
||
ProviderInputRef *string
|
||
ProviderOutputRef *string
|
||
GCSInputURI *string
|
||
GCSOutputURI *string
|
||
|
||
ItemCount int
|
||
SuccessCount int
|
||
FailCount int
|
||
CancelledCount int
|
||
|
||
EstimatedCost float64
|
||
HoldAmount *float64
|
||
ActualCost *float64
|
||
BaseUnitPrice float64
|
||
GroupRateMultiplier float64
|
||
AccountRateMultiplier float64
|
||
BatchDiscountMultiplier float64
|
||
HoldMultiplier float64
|
||
BillableUnitPrice float64
|
||
HoldUnitPrice float64
|
||
PricingSnapshotVersion int
|
||
Currency string
|
||
HoldID *string
|
||
|
||
IdempotencyKey *string
|
||
RequestHash *string
|
||
ManifestHash *string
|
||
|
||
RetryCount int
|
||
Version int
|
||
|
||
OutputExpiresAt *time.Time
|
||
InputDeletedAt *time.Time
|
||
OutputDeletedAt *time.Time
|
||
DownloadedAt *time.Time
|
||
UserDeletedAt *time.Time
|
||
|
||
LastErrorCode *string
|
||
LastErrorMessage *string
|
||
|
||
CreatedAt time.Time
|
||
UpdatedAt time.Time
|
||
SubmittedAt *time.Time
|
||
StartedAt *time.Time
|
||
FinishedAt *time.Time
|
||
SettledAt *time.Time
|
||
}
|
||
|
||
type CreateBatchImageJobParams struct {
|
||
BatchID string
|
||
UserID int64
|
||
APIKeyID *int64
|
||
AccountID *int64
|
||
Provider string
|
||
Model string
|
||
TaskName string
|
||
ParentBatchID *string
|
||
Status string
|
||
ProviderJobName *string
|
||
ProviderInputRef *string
|
||
ProviderOutputRef *string
|
||
GCSInputURI *string
|
||
GCSOutputURI *string
|
||
|
||
ItemCount int
|
||
SuccessCount int
|
||
FailCount int
|
||
CancelledCount int
|
||
|
||
EstimatedCost float64
|
||
HoldAmount *float64
|
||
ActualCost *float64
|
||
BaseUnitPrice float64
|
||
GroupRateMultiplier float64
|
||
AccountRateMultiplier float64
|
||
BatchDiscountMultiplier float64
|
||
HoldMultiplier float64
|
||
BillableUnitPrice float64
|
||
HoldUnitPrice float64
|
||
PricingSnapshotVersion int
|
||
Currency string
|
||
HoldID *string
|
||
|
||
IdempotencyKey *string
|
||
RequestHash *string
|
||
ManifestHash *string
|
||
|
||
RetryCount int
|
||
|
||
OutputExpiresAt *time.Time
|
||
}
|
||
|
||
type BatchImageItem struct {
|
||
ID int64
|
||
JobID string
|
||
CustomID string
|
||
Status string
|
||
RequestHash *string
|
||
PromptPreview *string
|
||
ProviderSourceObject *string
|
||
SourceLineNumber *int
|
||
SourceByteOffset *int64
|
||
SourceByteLength *int64
|
||
MimeType *string
|
||
FileExtension *string
|
||
ImageCount int
|
||
ErrorCode *string
|
||
ErrorMessage *string
|
||
BilledAmount *float64
|
||
CreatedAt time.Time
|
||
IndexedAt *time.Time
|
||
}
|
||
|
||
type CreateBatchImageItemParams struct {
|
||
JobID string
|
||
CustomID string
|
||
Status string
|
||
RequestHash *string
|
||
PromptPreview *string
|
||
ProviderSourceObject *string
|
||
SourceLineNumber *int
|
||
SourceByteOffset *int64
|
||
SourceByteLength *int64
|
||
MimeType *string
|
||
FileExtension *string
|
||
ImageCount int
|
||
ErrorCode *string
|
||
ErrorMessage *string
|
||
BilledAmount *float64
|
||
IndexedAt *time.Time
|
||
}
|
||
|
||
type BatchImageItemFilter struct {
|
||
Status string
|
||
Limit int
|
||
Offset int
|
||
}
|
||
|
||
type BatchImageJobFilter struct {
|
||
Status string
|
||
TaskNameLike string
|
||
Downloaded *bool
|
||
CreatedAfter *time.Time
|
||
CreatedBefore *time.Time
|
||
ExcludeDeleted bool
|
||
Limit int
|
||
Offset int
|
||
}
|
||
|
||
type BatchImageCounts struct {
|
||
SuccessCount int
|
||
FailCount int
|
||
}
|
||
|
||
type UpdateBatchImageJobProviderSubmitParams struct {
|
||
BatchID string
|
||
ProviderJobName string
|
||
ProviderInputRef string
|
||
ProviderOutputRef string
|
||
GCSInputURI string
|
||
GCSOutputURI string
|
||
EventPayload any
|
||
}
|
||
|
||
type BatchImageTransitionOptions struct {
|
||
EventType string
|
||
EventPayload any
|
||
ErrorCode *string
|
||
ErrorMessage *string
|
||
Now *time.Time
|
||
}
|
||
|
||
type MarkBatchImageJobSettledParams struct {
|
||
BatchID string
|
||
ActualCost float64
|
||
ManifestHash string
|
||
EventPayload any
|
||
Now *time.Time
|
||
OutputExpiresAt *time.Time
|
||
}
|
||
|
||
type BatchImageEvent struct {
|
||
ID int64
|
||
JobID string
|
||
EventType string
|
||
Payload []byte
|
||
EventHash *string
|
||
CreatedAt time.Time
|
||
}
|
||
|
||
type BatchImageRepository interface {
|
||
CreateBatchImageJob(ctx context.Context, params CreateBatchImageJobParams) (*BatchImageJob, error)
|
||
GetBatchImageJobByBatchID(ctx context.Context, batchID string) (*BatchImageJob, error)
|
||
GetBatchImageJobByIdempotencyKey(ctx context.Context, userID, apiKeyID int64, key string) (*BatchImageJob, error)
|
||
GetBatchImageJobByBatchIDForOwner(ctx context.Context, userID, apiKeyID int64, batchID string) (*BatchImageJob, error)
|
||
GetBatchImageJobByID(ctx context.Context, id int64) (*BatchImageJob, error)
|
||
ListBatchImageJobsForOwner(ctx context.Context, userID, apiKeyID int64, filter BatchImageJobFilter) ([]*BatchImageJob, error)
|
||
TransitionBatchImageJobStatus(ctx context.Context, batchID, toStatus string, opts BatchImageTransitionOptions) error
|
||
// TouchBatchImageJobSubmitting 刷新未提交（created/uploading）job 的 updated_at，
|
||
// 作为慢提交期间的心跳，防止被 stale 恢复扫描误杀。
|
||
TouchBatchImageJobSubmitting(ctx context.Context, batchID string) error
|
||
// FailStaleUnsubmittedBatchImageJob 原子地将仍处于 created/uploading 且
|
||
// provider_job_name 为空、updated_at 早于 cutoff 的 job 转为 failed。
|
||
// 返回 false 表示 job 已被并发推进（如已提交成功），调用方不得释放冻结。
|
||
FailStaleUnsubmittedBatchImageJob(ctx context.Context, batchID string, cutoff time.Time, code, message string) (bool, error)
|
||
UpdateBatchImageJobProviderOutputRef(ctx context.Context, batchID, providerOutputRef string) error
|
||
UpdateBatchImageJobProviderSubmit(ctx context.Context, params UpdateBatchImageJobProviderSubmitParams) error
|
||
RecordBatchImageJobSubmitFailure(ctx context.Context, batchID, code, message string, markFailed bool) error
|
||
MarkBatchImageJobSettled(ctx context.Context, params MarkBatchImageJobSettledParams) error
|
||
SetBatchImageJobSettlementFailed(ctx context.Context, batchID, code, message string) (int, error)
|
||
CreateBatchImageItem(ctx context.Context, params CreateBatchImageItemParams) (*BatchImageItem, error)
|
||
BulkCreateBatchImageItems(ctx context.Context, params []CreateBatchImageItemParams) error
|
||
ReplaceBatchImageItemsForJob(ctx context.Context, batchID string, items []CreateBatchImageItemParams, counts BatchImageCounts) error
|
||
ListBatchImageItems(ctx context.Context, batchID string, filter BatchImageItemFilter) ([]*BatchImageItem, error)
|
||
ListBatchImageItemsForOwner(ctx context.Context, userID, apiKeyID int64, batchID string, filter BatchImageItemFilter) ([]*BatchImageItem, error)
|
||
GetBatchImageJobForDownload(ctx context.Context, userID, apiKeyID int64, batchID string) (*BatchImageJob, error)
|
||
GetBatchImageItemForDownload(ctx context.Context, batchID, customID string) (*BatchImageItem, error)
|
||
ListBatchImageItemsForDownload(ctx context.Context, batchID string, status string, limit int) ([]*BatchImageItem, error)
|
||
ListBatchImageJobsDueForInputCleanup(ctx context.Context, cutoff time.Time, limit int) ([]*BatchImageJob, error)
|
||
ListBatchImageJobsDueForOutputCleanup(ctx context.Context, now time.Time, limit int) ([]*BatchImageJob, error)
|
||
ListStaleUnsubmittedBatchImageJobs(ctx context.Context, cutoff time.Time, limit int) ([]*BatchImageJob, error)
|
||
MarkBatchImageInputDeleted(ctx context.Context, batchID string, deletedAt time.Time) error
|
||
MarkBatchImageOutputDeleted(ctx context.Context, batchID string, deletedAt time.Time) error
|
||
MarkBatchImageDownloaded(ctx context.Context, batchID string, downloadedAt time.Time) error
|
||
MarkBatchImageJobUserDeleted(ctx context.Context, userID, apiKeyID int64, batchID string, deletedAt time.Time) error
|
||
SetBatchImageOutputExpiresAt(ctx context.Context, batchID string, expiresAt time.Time) error
|
||
RecordBatchImageCleanupFailure(ctx context.Context, batchID, code, message string) error
|
||
AppendBatchImageEvent(ctx context.Context, batchID, eventType string, payload any) error
|
||
}
|
||
|
||
func NewBatchImageID() (string, error) {
|
||
var b [16]byte
|
||
if _, err := rand.Read(b[:]); err != nil {
|
||
return "", err
|
||
}
|
||
return "imgbatch_" + hex.EncodeToString(b[:]), nil
|
||
}
|
||
|
||
func IsSupportedBatchImageProvider(provider string) bool {
|
||
switch provider {
|
||
case BatchImageProviderGeminiAPI, BatchImageProviderVertex:
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func IsTerminalBatchImageJobStatus(status string) bool {
|
||
switch status {
|
||
case BatchImageJobStatusCompleted, BatchImageJobStatusFailed, BatchImageJobStatusCancelled, BatchImageJobStatusOutputDeleted:
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func CanTransitionBatchImageJob(from, to string) bool {
|
||
if from == "" || to == "" {
|
||
return false
|
||
}
|
||
if IsTerminalBatchImageJobStatus(from) {
|
||
return to == BatchImageJobStatusOutputDeleted &&
|
||
from != BatchImageJobStatusOutputDeleted &&
|
||
(from == BatchImageJobStatusCompleted || from == BatchImageJobStatusFailed || from == BatchImageJobStatusCancelled)
|
||
}
|
||
if to == BatchImageJobStatusFailed {
|
||
return true
|
||
}
|
||
|
||
allowed := map[string]map[string]struct{}{
|
||
BatchImageJobStatusCreated: {
|
||
BatchImageJobStatusUploading: {},
|
||
BatchImageJobStatusSubmitted: {},
|
||
BatchImageJobStatusCancelled: {},
|
||
},
|
||
BatchImageJobStatusUploading: {
|
||
BatchImageJobStatusSubmitted: {},
|
||
BatchImageJobStatusCancelled: {},
|
||
},
|
||
BatchImageJobStatusSubmitted: {
|
||
BatchImageJobStatusRunning: {},
|
||
BatchImageJobStatusIndexing: {},
|
||
BatchImageJobStatusFailed: {},
|
||
BatchImageJobStatusCancelled: {},
|
||
},
|
||
BatchImageJobStatusRunning: {
|
||
BatchImageJobStatusRunning: {},
|
||
BatchImageJobStatusIndexing: {},
|
||
BatchImageJobStatusFailed: {},
|
||
BatchImageJobStatusCancelled: {},
|
||
},
|
||
BatchImageJobStatusIndexing: {
|
||
BatchImageJobStatusSettling: {},
|
||
BatchImageJobStatusFailed: {},
|
||
},
|
||
BatchImageJobStatusSettling: {
|
||
BatchImageJobStatusCompleted: {},
|
||
},
|
||
}
|
||
_, ok := allowed[from][to]
|
||
return ok
|
||
}
|