deploy: 7ae99ac — 更新 index.js

backend/src/index.js

@@ -15,7 +15,12 @@ const litellm = require("./litellm");
 const app = express();
 const PORT = parseInt(process.env.PORT || "3001", 10);
 
-// ─── Middleware ────────────────────────────────────────────────────────────
+// BUG FIX: Set trust proxy BEFORE rate limiter.
+// nginx runs in front of backend and sets X-Forwarded-For header.
+// Without trust proxy, express-rate-limit throws ERR_ERL_UNEXPECTED_X_FORWARDED_FOR
+// and logs a ValidationError on every single request.
+// Value 1 = trust first proxy hop (nginx on localhost).
+app.set("trust proxy", 1);
 
 app.use(helmet({ crossOriginResourcePolicy: false }));
 app.use(
@@ -33,7 +38,6 @@ app.use(
   })
 );
 
-// Rate limiting
 app.use(
   "/api/",
   rateLimit({
@@ -45,44 +49,28 @@ app.use(
   })
 );
 
-// ─── Routes ───────────────────────────────────────────────────────────────
-
-// BUG FIX: The simple stub was removed. statsRouter's /health checks LiteLLM + DB health.
 app.use("/api/models", modelsRouter);
 app.use("/api", statsRouter);
 
-// 404
 app.use((req, res) => {
   res.status(404).json({ success: false, error: "Not found" });
 });
 
-// Error handler
 app.use((err, req, res, _next) => {
   logger.error("Unhandled error", { error: err.message, stack: err.stack });
   res.status(500).json({ success: false, error: "Internal server error" });
 });
 
-// ─── Startup ──────────────────────────────────────────────────────────────
-
 async function start() {
-  // Initialize DB
   db.getDb();
   logger.info("Database initialized");
-
-  // BUG FIX: 5s hardcoded delay was unreliable — LiteLLM can take 30-60s to start.
-  // Use retry loop with backoff instead.
   syncModelsToLitellmWithRetry();
-
   app.listen(PORT, "0.0.0.0", () => {
-    logger.info(`AI Gateway Backend running on port ${PORT}`);
-    logger.info(`Gateway public URL: ${process.env.GATEWAY_PUBLIC_URL || "http://localhost"}`);
+    logger.info("AI Gateway Backend running on port " + PORT);
+    logger.info("Gateway public URL: " + (process.env.GATEWAY_PUBLIC_URL || "http://localhost"));
   });
 }
 
-/**
- * Retry wrapper: attempts sync with exponential backoff for up to ~5 minutes.
- * Handles the case where LiteLLM container starts slower than the backend.
- */
 async function syncModelsToLitellmWithRetry() {
   const MAX_ATTEMPTS = 10;
   const BASE_DELAY_MS = 5000;
@@ -93,33 +81,26 @@ async function syncModelsToLitellmWithRetry() {
       return;
     } catch (err) {
       const delay = Math.min(BASE_DELAY_MS * attempt, 30000);
-      logger.warn(`LiteLLM not ready (attempt ${attempt}/${MAX_ATTEMPTS}), retrying in ${delay}ms...`);
+      logger.warn("LiteLLM not ready (attempt " + attempt + "/" + MAX_ATTEMPTS + "), retrying in " + delay + "ms...");
       await new Promise((r) => setTimeout(r, delay));
     }
   }
-  logger.error("LiteLLM did not become ready after all retry attempts. Models not synced.");
+  logger.error("LiteLLM did not become ready after all retry attempts.");
 }
 
-/**
- * On startup, re-register all persisted models with LiteLLM.
- * This handles the case where LiteLLM was restarted and lost its in-memory state.
- */
 async function syncModelsToLitellm() {
   try {
     const models = db.listModels({ enabledOnly: true });
-    logger.info(`Syncing ${models.length} models to LiteLLM...`);
-
+    logger.info("Syncing " + models.length + " models to LiteLLM...");
     for (const model of models) {
       try {
-        // _apiKey is the raw (unmasked) API key set by deserializeModel
         const litellmId = await litellm.registerModel({ ...model, _apiKey: model._apiKey });
         db.updateModel(model.id, { litellmId });
-        logger.info(`Synced: ${model.name}`);
+        logger.info("Synced: " + model.name);
       } catch (err) {
-        logger.warn(`Failed to sync model ${model.name}: ${err.message}`);
+        logger.warn("Failed to sync model " + model.name + ": " + err.message);
       }
     }
-
     logger.info("Model sync complete");
   } catch (err) {
     logger.error("Model sync failed", { error: err.message });
@@ -129,4 +110,4 @@ async function syncModelsToLitellm() {
 start().catch((err) => {
   logger.error("Fatal startup error", { error: err.message });
   process.exit(1);
-});
+});