From 31a9654d2b352d10ba0aea5ba859e83ad2641e1e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A2=AB=E9=81=97=E5=BF=98=E7=9A=84=E8=AE=B0=E5=BF=86?=
 <32097237+YIWANG-sketch@users.noreply.github.com>
Date: Mon, 9 Dec 2024 01:26:55 +0800
Subject: [PATCH] request log

---
 you_providers/logger.mjs | 152 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 you_providers/logger.mjs

diff --git a/you_providers/logger.mjs b/you_providers/logger.mjs
new file mode 100644
index 0000000..d40c2b4
--- /dev/null
+++ b/you_providers/logger.mjs
@@ -0,0 +1,152 @@
+import path from "path";
+import fs from "fs";
+import { fileURLToPath } from 'url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+class Logger {
+    constructor() {
+        this.logFilePath = path.join(__dirname, 'requests.log');
+        this.statistics = {};
+        this.monthStart = this.getMonthStart();
+        this.loadStatistics();
+    }
+
+    getMonthStart() {
+        const now = new Date();
+        // 每月第一天
+        const monthStart = new Date(now.getFullYear(), now.getMonth(), 1);
+        monthStart.setHours(0, 0, 0, 0);
+        return monthStart;
+    }
+
+    // 加载日志
+    loadStatistics() {
+        if (fs.existsSync(this.logFilePath)) {
+            const data = fs.readFileSync(this.logFilePath, 'utf-8');
+            const entries = data.split('\n').filter(line => line.trim());
+            const validEntries = [];
+
+            for (const line of entries) {
+                try {
+                    const logEntry = JSON.parse(line);
+                    if (logEntry && logEntry.email && logEntry.time && logEntry.mode && logEntry.model !== undefined && logEntry.completed !== undefined && logEntry.unusualQueryVolume !== undefined) {
+                        validEntries.push(logEntry);
+                    } else {
+                        console.warn(`忽略格式不正确的日志: ${line}`);
+                    }
+                } catch (e) {
+                    console.warn(`无法解析的日志，已忽略: ${line}`);
+                }
+            }
+
+            // 处理有效日志
+            for (const logEntry of validEntries) {
+                const logDate = new Date(logEntry.time);
+                const email = logEntry.email || 'unknown';
+                // 初始化邮箱
+                if (!this.statistics[email]) {
+                    this.statistics[email] = {
+                        requests: [],
+                        totalRequests: 0,
+                        defaultModeCount: 0,
+                        customModeCount: 0,
+                        modelCount: {},
+                    };
+                }
+                if (logDate >= this.monthStart) {
+                    this.statistics[email].requests.push(logEntry);
+                    this.updateStatistics(email, logEntry);
+                }
+            }
+
+            // 每个邮箱时间排序
+            for (const email in this.statistics) {
+                this.statistics[email].requests.sort((a, b) => new Date(b.time) - new Date(a.time));
+            }
+
+            // 理无效数据
+            const cleanedData = validEntries.map(entry => JSON.stringify(entry)).join('\n') + '\n';
+            fs.writeFileSync(this.logFilePath, cleanedData);
+        }
+    }
+
+    // 更新统计
+    updateStatistics(email, logEntry) {
+        const stats = this.statistics[email];
+        stats.totalRequests++;
+        if (logEntry.mode === 'default') {
+            stats.defaultModeCount++;
+        } else if (logEntry.mode === 'custom') {
+            stats.customModeCount++;
+        }
+
+        if (logEntry.model) {
+            if (!stats.modelCount[logEntry.model]) {
+                stats.modelCount[logEntry.model] = 0;
+            }
+            stats.modelCount[logEntry.model]++;
+        }
+    }
+
+    // 记录请求日志
+    logRequest({ email, time, mode, model, completed, unusualQueryVolume }) {
+        const logEntry = {
+            email,  // 用户邮箱
+            time,  // 请求时间
+            mode,  // 请求模式
+            model,  // 请求模型
+            completed,  // 是否完成
+            unusualQueryVolume,  // 是否异常请求量
+        };
+
+        // 写入日志
+        fs.appendFileSync(this.logFilePath, JSON.stringify(logEntry) + '\n');
+
+        // 当前月份请求更新统计
+        const logDate = new Date(time);
+        if (logDate >= this.monthStart) {
+            const userEmail = email || 'unknown';
+            if (!this.statistics[userEmail]) {
+                this.statistics[userEmail] = {
+                    requests: [],
+                    totalRequests: 0,
+                    defaultModeCount: 0,
+                    customModeCount: 0,
+                    modelCount: {},
+                };
+            }
+            this.statistics[userEmail].requests.push(logEntry);
+            // 对请求按照时间排序
+            this.statistics[userEmail].requests.sort((a, b) => new Date(b.time) - new Date(a.time));
+            this.updateStatistics(userEmail, logEntry);
+        }
+    }
+
+    // 输出当前月份的统计信息
+    printStatistics() {
+        console.log('===== 请求统计信息 =====');
+        console.log(`当前月份: ${this.monthStart.toLocaleDateString('zh-CN', {
+            year: 'numeric',
+            month: 'long',
+            day: 'numeric',
+        })}`);
+        const emails = Object.keys(this.statistics).sort();
+        for (const email of emails) {
+            const stats = this.statistics[email];
+            console.log(`用户邮箱: ${email}`);
+            console.log(`总请求次数: ${stats.totalRequests}`);
+            console.log(`default 请求次数: ${stats.defaultModeCount}`);
+            console.log(`custom 请求次数: ${stats.customModeCount}`);
+            console.log('各模型请求次数:');
+            for (const [model, count] of Object.entries(stats.modelCount)) {
+                console.log(`  - ${model}: ${count}`);
+            }
+            console.log('------------------------------');
+        }
+        console.log('================================');
+    }
+}
+
+export default Logger;
\ No newline at end of file