wangbo
6bab0f0749
为 PostgreSQL 连接增加可配置的事务空闲与锁等待超时，并在请求取消或 Worker 退出后使用独立有界上下文回滚事务。\n\n恢复过期任务时按批次使用 SKIP LOCKED，周期重建缺失或终态 River Job；锁等待超时只在确认尚未提交上游时安全重排，避免重复执行和重复结算。\n\n验证：完整 Go 测试、go vet、生产 Kustomize 渲染、gofmt 与 git diff --check 均通过。
91 lines
3.1 KiB
Go
91 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/config"
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/httpapi"
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/store"
|
|
)
|
|
|
|
// @title EasyAI AI Gateway API
|
|
// @version 0.1.0
|
|
// @description EasyAI AI Gateway 的本地鉴权、平台模型管理、定价、运行策略、钱包和 AI 任务接口。
|
|
// @description 受保护接口使用 Authorization: Bearer <JWT 或 API Key>，管理接口只接受 JWT 用户凭证。
|
|
// @BasePath /
|
|
// @schemes http https
|
|
// @securityDefinitions.apikey BearerAuth
|
|
// @in header
|
|
// @name Authorization
|
|
// @description Bearer JWT 或 API Key。
|
|
func main() {
|
|
cfg := config.Load()
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
|
|
Level: cfg.LogLevel,
|
|
}))
|
|
if err := cfg.Validate(); err != nil {
|
|
logger.Error("invalid gateway configuration", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
db, err := store.ConnectWithPoolOptions(ctx, cfg.DatabaseURL, store.PostgresPoolOptions{
|
|
MaxConns: cfg.DatabaseMaxConns,
|
|
IdleInTransactionTimeout: time.Duration(cfg.DatabaseIdleInTransactionTimeoutSeconds) * time.Second,
|
|
LockTimeout: time.Duration(cfg.DatabaseLockTimeoutSeconds) * time.Second,
|
|
})
|
|
if err != nil {
|
|
stop()
|
|
logger.Error("connect postgres failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
defer db.Close()
|
|
// Cancel background LISTEN and worker goroutines before closing the pool.
|
|
// This also prevents a startup panic from waiting forever in pgxpool.Close.
|
|
defer stop()
|
|
if recovery, err := db.RecoverInterruptedRuntimeState(ctx); err != nil {
|
|
logger.Error("recover interrupted runtime state failed", "error", err)
|
|
os.Exit(1)
|
|
} else if recovery.ReleasedConcurrencyLeases > 0 || recovery.ReleasedRateReservations > 0 || recovery.FailedAttempts > 0 || recovery.FailedTasks > 0 || recovery.RequeuedAsyncTasks > 0 || recovery.CleanedTaskAdmissions > 0 {
|
|
logger.Warn("interrupted runtime state recovered",
|
|
"releasedConcurrencyLeases", recovery.ReleasedConcurrencyLeases,
|
|
"releasedRateReservations", recovery.ReleasedRateReservations,
|
|
"failedAttempts", recovery.FailedAttempts,
|
|
"failedTasks", recovery.FailedTasks,
|
|
"requeuedAsyncTasks", recovery.RequeuedAsyncTasks,
|
|
"cleanedTaskAdmissions", recovery.CleanedTaskAdmissions,
|
|
)
|
|
}
|
|
|
|
server := &http.Server{
|
|
Addr: cfg.HTTPAddr,
|
|
Handler: httpapi.NewServerWithContext(ctx, cfg, db, logger),
|
|
ReadHeaderTimeout: 10 * time.Second,
|
|
}
|
|
|
|
go func() {
|
|
logger.Info("easyai ai gateway api started", "addr", cfg.HTTPAddr)
|
|
if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
logger.Error("http server failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
<-ctx.Done()
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
|
|
defer cancel()
|
|
if err := server.Shutdown(shutdownCtx); err != nil {
|
|
logger.Error("http shutdown failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
logger.Info("easyai ai gateway api stopped")
|
|
}
|