wangbo
fd3b6bf042
River 初始化遇到瞬时数据库超时时最多重试三次，并确保启动异常时先取消后台 LISTEN 和 Worker 上下文再关闭连接池，避免不就绪 Pod 卡在退出路径。Worker 维护循环每 15 秒执行幂等运行时恢复，使旧实例中断任务在执行租约到期后自动重排并继续触发孤儿 Job 救援。验证通过完整 Go 测试、go vet、竞态检查和迁移安全检查。
87 lines
2.9 KiB
Go
87 lines
2.9 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.ConnectWithMaxConns(ctx, cfg.DatabaseURL, cfg.DatabaseMaxConns)
|
|
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")
|
|
}
|