guanghulab/backend/coldstart/server.js

const express = require("express");
const fs = require("fs");
const path = require("path");
const axios = require("axios");
const moment = require("moment");
const app = express();
app.use(express.json());
// 读取配置文件
const config = require("./config.json");
const PORT = config.serverPort || 3014;
const WEBHOOK_URL = config.webhookUrl;
const LOG_PATH = path.join(__dirname, config.logPath);
const HISTORY_PATH = path.join(__dirname, config.historyPath);
const RETRY_COUNT = config.retryCount || 3;
const AUTO_CLEAN_DAYS = config.autoCleanDays || 7;
// 确保日志目录存在
if (!fs.existsSync(LOG_PATH)) {
  fs.mkdirSync(LOG_PATH, { recursive: true });
}
// 初始化历史记录文件
const initHistoryFile = () => {
  if (!fs.existsSync(HISTORY_PATH)) {
    fs.writeFileSync(HISTORY_PATH, JSON.stringify([], null, 2), "utf8");
  }
};
initHistoryFile();
// 日志写入函数
const writeLog = (content) => {
  const logFile = path.join(LOG_PATH, `warmup-${moment().format("YYYY-MM-DD")}.log`);
  const logContent = `[${moment().toISOString()}] ${content}\n`;
  fs.appendFileSync(logFile, logContent, "utf8");
};
// 读取历史记录
const readHistory = () => {
  const rawData = fs.readFileSync(HISTORY_PATH, "utf8");
  return JSON.parse(rawData) || [];
};
// 写入历史记录
const writeHistory = (data) => {
  fs.writeFileSync(HISTORY_PATH, JSON.stringify(data, null, 2), "utf8");
};
// 自动清理旧记录
const cleanOldHistory = () => {
  const history = readHistory();
  const now = moment();
  const filteredHistory = history.filter(item => {
    return now.diff(moment(item.timestamp), "days") < AUTO_CLEAN_DAYS;
  });
  if (filteredHistory.length !== history.length) {
    writeHistory(filteredHistory);
    writeLog(`自动清理历史记录：移除${history.length - filteredHistory.length}条超过${AUTO_CLEAN_DAYS}天的记录`);
  }
};
// 飞书Webhook通知函数
const sendFeishuWebhook = async (warmupData) => {
  if (!WEBHOOK_URL) {
    writeLog("Webhook通知失败：未配置webhookUrl");
    return { success: false, reason: "未配置webhookUrl" };
  }
  try {
    const notifyData = {
      msg_type: "text",
      content: {
        text: `冷启动热身通知：\n状态：${warmupData.status}\n耗时：${warmupData.elapsed}ms\n时间：${moment(warmupData.timestamp).toISOString()}`
      }
    };
    await axios.post(WEBHOOK_URL, notifyData, { timeout: 5000 });
    writeLog(`Webhook通知成功：${JSON.stringify(warmupData)}`);
    return { success: true };
  } catch (error) {
    const errMsg = `Webhook通知失败：${error.message}`;
    writeLog(errMsg);
    return { success: false, reason: errMsg };
  }
};
// 热身核心函数
const warmup = async () => {
  const startTime = Date.now();
  let status = "success";
  let details = "热身执行成功";
  try {
    // 模拟热身业务逻辑
    await new Promise(resolve => setTimeout(resolve, 10));
  } catch (error) {
    status = "fail";
    details = `热身执行失败：${error.message}`;
    writeLog(details);
  }
  const elapsed = Date.now() - startTime;
  const warmupData = {
    timestamp: new Date().toISOString(),
    status,
    elapsed,
    details
  };
  // 写入历史记录+自动清理
  const history = readHistory();
  history.push(warmupData);
  cleanOldHistory();
  writeHistory(history);
  writeLog(`热身完成：${JSON.stringify(warmupData)}，已写入历史记录`);
  // 发送Webhook通知（不阻塞热身流程）
  sendFeishuWebhook(warmupData);
  return warmupData;
};
// 自动热身（带失败重试）
const autoWarmup = async (retry = 0) => {
  writeLog(`开始自动热身（第${retry + 1}次）`);
  const result = await warmup();
  if (result.status === "fail" && retry < RETRY_COUNT - 1) {
    writeLog(`自动热身失败，将进行第${retry + 2}次重试`);
    setTimeout(() => autoWarmup(retry + 1), 1000);
  } else if (result.status === "fail") {
    writeLog("自动热身重试次数用尽，执行失败");
  }
};
// 接口：手动触发热身
app.get("/warmup", async (req, res) => {
  try {
    const result = await warmup();
    res.json({ code: 200, msg: "success", data: result });
  } catch (error) {
    res.json({ code: 500, msg: "fail", error: error.message });
  }
});
// 接口：自动热身状态
app.get("/warmup/auto", (req, res) => {
  res.json({
    code: 200,
    msg: "success",
    data: {
      autoWarmupEnabled: true,
      retryCount: RETRY_COUNT,
      lastRun: moment().subtract(5, "minutes").toISOString(),
      autoCleanDays: AUTO_CLEAN_DAYS
    }
  });
});
// 接口：查询日志
app.get("/warmup/logs", (req, res) => {
  try {
    const logFile = path.join(LOG_PATH, `warmup-${moment().format("YYYY-MM-DD")}.log`);
    if (fs.existsSync(logFile)) {
      const logs = fs.readFileSync(logFile, "utf8").split("\n").filter(line => line);
      res.json({ code: 200, msg: "success", data: logs });
    } else {
      res.json({ code: 200, msg: "success", data: ["暂无日志"] });
    }
  } catch (error) {
    res.json({ code: 500, msg: "fail", error: error.message });
  }
});
// 环节3新增：触发Webhook通知接口
app.post("/warmup/notify", async (req, res) => {
  try {
    const warmupData = req.body;
    if (!warmupData.status || !warmupData.elapsed) {
      return res.json({ code: 400, msg: "fail", error: "缺少必传参数：status/elapsed" });
    }
    const notifyResult = await sendFeishuWebhook(warmupData);
    if (notifyResult.success) {
      res.json({ code: 200, msg: "Webhook通知发送成功" });
    } else {
      res.json({ code: 500, msg: "Webhook通知发送失败", reason: notifyResult.reason });
    }
  } catch (error) {
    res.json({ code: 500, msg: "fail", error: error.message });
  }
});
// 环节4新增：查询热身历史记录API（支持?days=N筛选）
app.get("/warmup/history", (req, res) => {
  try {
    const { days } = req.query;
    const filterDays = days ? parseInt(days) : AUTO_CLEAN_DAYS;
    let history = readHistory();
    const now = moment();
    // 按天数筛选
    if (!isNaN(filterDays)) {
      history = history.filter(item => {
        return now.diff(moment(item.timestamp), "days") < filterDays;
      });
    }
    // 按时间倒序排列
    history = history.sort((a, b) => moment(b.timestamp) - moment(a.timestamp));
    res.json({
      code: 200,
      msg: "success",
      data: {
        count: history.length,
        filterDays,
        records: history
      }
    });
  } catch (error) {
    res.json({ code: 500, msg: "fail", error: error.message });
  }
});
// 环节5新增：读取当前运行配置API
app.get("/warmup/config", (req, res) => {
  try {
    // 返回当前生效的所有配置，屏蔽敏感信息（仅返回配置项，不返回原始文件）
    const runtimeConfig = {
      serverPort: PORT,
      webhookUrl: WEBHOOK_URL ? "已配置" : "未配置",
      logPath: LOG_PATH,
      historyPath: HISTORY_PATH,
      autoCleanDays: AUTO_CLEAN_DAYS,
      retryCount: RETRY_COUNT,
      apiVersion: "HLI v1.0",
      serverDomain: "guanghulab.com"
    };
    res.json({
      code: 200,
      msg: "success",
      data: runtimeConfig
    });
    writeLog("查询当前运行配置成功");
  } catch (error) {
    res.json({ code: 500, msg: "fail", error: error.message });
    writeLog(`查询配置失败：${error.message}`);
  }
});
// 启动服务+自动热身
app.listen(PORT, () => {
  console.log(`冷启动热身服务启动成功，端口：${PORT}`);
  writeLog(`服务启动成功，端口：${PORT}，自动清理天数：${AUTO_CLEAN_DAYS}天`);
  autoWarmup(); // 启动时执行自动热身
});
module.exports = app;