fix(worker): 将 Gemini 同步请求交由异步队列执行

将非流式 Gemini generateContent 改为 River Worker 执行，API 通过批量状态查询等待完成，避免高并发同步请求耗尽 API 数据库连接池。\n\n生成图片在持久化时保存带哈希的资产引用，响应恢复时下载并校验后重建 Base64，数据库不保存媒体原文。请求体物化增加前置内存门禁，并扩展双 API/Worker PostgreSQL 压力测试。\n\n验证：go test ./... -count=1；go vet ./...；pnpm openapi；迁移安全检查；64 与 256 请求双角色异步 Gemini 压力测试。
2026-07-31 05:49:45 +08:00
parent 352e23e099
commit 4af86b22ee
12 changed files with 516 additions and 48 deletions
@@ -25,28 +25,32 @@ import (
 )

 type Service struct {
-	cfg                  config.Config
-	store                *store.Store
-	logger               *slog.Logger
-	clients              map[string]clients.Client
-	scriptExecutor       *scriptengine.Executor
-	httpClients          *httpClientCache
-	riverMu              sync.RWMutex
-	riverControlClient   *river.Client[pgx.Tx]
-	riverExecutionClient asyncExecutionClient
-	riverDrainingClients map[asyncExecutionClient]struct{}
-	riverWorkerCapacity  int
-	workerInstanceID     string
-	asyncCapacityLoader  func(context.Context, int) (store.AsyncWorkerCapacitySnapshot, error)
-	admissionWakeMu      sync.Mutex
-	admissionWake        chan struct{}
-	asyncAdmissionWake   chan struct{}
-	admissionTaskWaiters map[string]*admissionTaskWaiter
-	admissionListener    sync.Once
-	asyncClientFactory   func(int) (asyncExecutionClient, error)
-	mediaResultSlots     chan struct{}
-	directOSS            *directOSSUploader
-	billingMetrics       billingMetricsObserver
+	cfg                    config.Config
+	store                  *store.Store
+	logger                 *slog.Logger
+	clients                map[string]clients.Client
+	scriptExecutor         *scriptengine.Executor
+	httpClients            *httpClientCache
+	riverMu                sync.RWMutex
+	riverControlClient     *river.Client[pgx.Tx]
+	riverExecutionClient   asyncExecutionClient
+	riverDrainingClients   map[asyncExecutionClient]struct{}
+	riverWorkerCapacity    int
+	workerInstanceID       string
+	asyncCapacityLoader    func(context.Context, int) (store.AsyncWorkerCapacitySnapshot, error)
+	admissionWakeMu        sync.Mutex
+	admissionWake          chan struct{}
+	asyncAdmissionWake     chan struct{}
+	admissionTaskWaiters   map[string]*admissionTaskWaiter
+	admissionListener      sync.Once
+	asyncClientFactory     func(int) (asyncExecutionClient, error)
+	mediaResultSlots       chan struct{}
+	directOSS              *directOSSUploader
+	taskCompletionOnce     sync.Once
+	taskCompletionMu       sync.Mutex
+	taskCompletionWaiters  map[string]map[chan struct{}]struct{}
+	taskCompletionPollWake chan struct{}
+	billingMetrics         billingMetricsObserver
 }

 type billingMetricsObserver interface {
@@ -149,13 +153,15 @@ func New(cfg config.Config, db *store.Store, logger *slog.Logger, observers ...b
 			"universal":             clients.UniversalClient{HTTPClient: httpClients.none, ScriptExecutor: scriptExecutor},
 			"simulation":            clients.SimulationClient{},
 		},
-		httpClients:          httpClients,
-		workerInstanceID:     asyncWorkerID(),
-		admissionWake:        make(chan struct{}, 4096),
-		asyncAdmissionWake:   make(chan struct{}, 1),
-		admissionTaskWaiters: map[string]*admissionTaskWaiter{},
-		mediaResultSlots:     make(chan struct{}, cfg.MediaMaterializationConcurrency),
-		directOSS:            newDirectOSSUploader(cfg),
+		httpClients:            httpClients,
+		workerInstanceID:       asyncWorkerID(),
+		admissionWake:          make(chan struct{}, 4096),
+		asyncAdmissionWake:     make(chan struct{}, 1),
+		admissionTaskWaiters:   map[string]*admissionTaskWaiter{},
+		mediaResultSlots:       make(chan struct{}, cfg.MediaMaterializationConcurrency),
+		directOSS:              newDirectOSSUploader(cfg),
+		taskCompletionWaiters:  map[string]map[chan struct{}]struct{}{},
+		taskCompletionPollWake: make(chan struct{}, 1),
 	}
 	if len(observers) > 0 {
 		service.billingMetrics = observers[0]