Update egg-api.js

egg-api.js

@@ -305,6 +305,47 @@ app.get('/processGif', async (req, res) => {
         res.status(500).send('内部服务器错误');
     }
 });
+
+const apiStats = {
+    '/process-image': 0,
+    '/processGif': 0
+};
+
+// 中间件记录API请求次数
+app.use((req, res, next) => {
+    if (apiStats[req.path] !== undefined) {
+        apiStats[req.path]++;
+    }
+    next();
+});
+
+// 提供API请求统计数据的接口
+app.get('/api-stats', (req, res) => {
+    res.json(apiStats);
+});
+
+// 每天重置统计数据
+setInterval(() => {
+    for (const api in apiStats) {
+        apiStats[api] = 0;
+    }
+}, 24 * 60 * 60 * 1000); // 24小时
+
+// 每月重置统计数据
+const resetStatsMonthly = () => {
+    const now = new Date();
+    const firstDayOfNextMonth = new Date(now.getFullYear(), now.getMonth() + 1, 1);
+    const timeUntilNextMonth = firstDayOfNextMonth - now;
+    setTimeout(() => {
+        for (const api in apiStats) {
+            apiStats[api] = 0;
+        }
+        resetStatsMonthly();
+    }, timeUntilNextMonth);
+};
+
+resetStatsMonthly();
+
 app.use(serveStatic(__dirname)); // 添加这一行，设置静态文件服务器
 app.get('/', (req, res) => {
     res.sendFile(path.join(__dirname, 'index.html')); // 修改这一行，发送HTML文件
@@ -312,4 +353,4 @@ app.get('/', (req, res) => {
 // 启动 Express 服务器
 app.listen(PORT, () => {
     console.log(`服务器运行在端口 ${PORT}`);
-});
+});