52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package generation
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"infinite-canvas/backend/internal/model"
|
|
"infinite-canvas/backend/internal/protocol"
|
|
)
|
|
|
|
// MediaResolver 水合参考媒体（资源 URL / data URL），由组合根注入，避免 generation → service 回环。
|
|
type MediaResolver interface {
|
|
HydrateGenerationMedia(userID string, input *Input, preferURL bool) error
|
|
}
|
|
|
|
// TaskProgress 写入任务进度/日志。
|
|
type TaskProgress interface {
|
|
SyncProviderTaskProgress(taskID string, responseBody []byte)
|
|
Log(userID, taskID, level, message, payload string) error
|
|
}
|
|
|
|
// BillingHooks 生成路径上的计费终态钩子。
|
|
type BillingHooks interface {
|
|
MarkBillingUncertain(taskID, reason string) error
|
|
Settle(taskID string) error
|
|
Refund(taskID, reason string) error
|
|
RestoreRefunded(taskID, reason string) error
|
|
}
|
|
|
|
// ChannelLimiter 渠道并发槽与熔断结果。
|
|
type ChannelLimiter interface {
|
|
AcquireChannelSlot(ctx context.Context, channelID string) (release func(), err error)
|
|
RecordChannelResult(channelID string, success bool)
|
|
}
|
|
|
|
// MediaPersister 持久化生成结果与资源 URL。
|
|
type MediaPersister interface {
|
|
PersistGeneratedMediaResult(ctx context.Context, userID, taskID string, result map[string]interface{}) (map[string]interface{}, error)
|
|
DirectResourceURL(resource *model.Resource, expiresAt time.Time) (string, error)
|
|
}
|
|
|
|
// Deps 是 Engine 的外部端口集合；禁止持有组合根或回环到 service。
|
|
type Deps struct {
|
|
Media MediaResolver
|
|
Progress TaskProgress
|
|
Billing BillingHooks
|
|
Channel ChannelLimiter
|
|
Persist MediaPersister
|
|
// Registry 可选；为空时 Engine 使用官方插件包回退表。
|
|
Registry *protocol.Registry
|
|
}
|