befly 3.8.29 → 3.8.31

package/lib/logger.ts

@@ -1,216 +1,95 @@
 /**
- * 日志系统 - Befly 项目专用
- * 直接集成环境变量，提供开箱即用的日志功能
+ * 日志系统 - 基于 pino 实现
  */
 
+import pino from 'pino';
 import { join } from 'pathe';
-import { appendFile, stat } from 'node:fs/promises';
-import type { LogLevel } from '../types/common.js';
-import type { LoggerConfig } from '../types/befly.js';
+
+import type { LoggerConfig } from 'befly-shared/types';
+
+let instance: pino.Logger | null = null;
+let mockInstance: pino.Logger | null = null;
+let config: LoggerConfig = {
+    debug: 0,
+    dir: './logs',
+    console: 1,
+    maxSize: 10
+};
 
 /**
- * 日志消息类型
+ * 配置日志
  */
-type LogMessage = string | number | boolean | null | undefined | Record<string, any> | any[];
+export function configure(cfg: LoggerConfig): void {
+    config = { ...config, ...cfg };
+    instance = null;
+}
 
 /**
- * 格式化日期时间
+ * 设置 Mock Logger（用于测试）
+ * @param mock - Mock pino 实例，传 null 清除 mock
  */
-function formatDate(): string {
-    const now = new Date();
-    const year = now.getFullYear();
-    const month = String(now.getMonth() + 1).padStart(2, '0');
-    const day = String(now.getDate()).padStart(2, '0');
-    const hours = String(now.getHours()).padStart(2, '0');
-    const minutes = String(now.getMinutes()).padStart(2, '0');
-    const seconds = String(now.getSeconds()).padStart(2, '0');
-    return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}`;
+export function setMockLogger(mock: pino.Logger | null): void {
+    mockInstance = mock;
 }
 
 /**
- * 日志器类
+ * 获取 pino 日志实例
  */
-export class Logger {
-    /** 当前使用的日志文件缓存 */
-    private static currentFiles: Map<string, string> = new Map();
-
-    /** 日志配置 */
-    private static config: LoggerConfig = {
-        debug: 1,
-        excludeFields: 'password,token,secret',
-        dir: './logs',
-        console: 1,
-        maxSize: 10 * 1024 * 1024
-    };
-
-    /**
-     * 配置日志器
-     * @param config - 日志配置
-     */
-    static configure(config: LoggerConfig) {
-        this.config = { ...this.config, ...config };
-    }
-
-    /**
-     * 记录日志
-     * @param level - 日志级别
-     * @param message - 日志消息
-     */
-    static async log(level: LogLevel, message: LogMessage): Promise<void> {
-        // debug 日志特殊处理：仅当 LOG_DEBUG=1 时才记录
-        if (level === 'debug' && this.config.debug !== 1) return;
-
-        // 格式化消息
-        const timestamp = formatDate();
-
-        // 处理消息内容
-        let content = '';
-        if (typeof message === 'object' && message !== null && Object.keys(message).length > 0) {
-            content = JSON.stringify(message, null, 0).replace(/\\"/g, '"');
-        } else {
-            content = String(message);
+export function getLogger(): pino.Logger {
+    // 优先返回 mock 实例（用于测试）
+    if (mockInstance) return mockInstance;
+
+    if (instance) return instance;
+
+    const level = config.debug === 1 ? 'debug' : 'info';
+    const targets: pino.TransportTargetOptions[] = [];
+
+    // 文件输出
+    targets.push({
+        target: 'pino-roll',
+        level: level,
+        options: {
+            file: join(config.dir || './logs', 'app'),
+            frequency: 'daily',
+            size: `${config.maxSize || 10}m`,
+            mkdir: true,
+            dateFormat: 'yyyy-MM-dd'
         }
-
-        // 格式化日志消息
-        const levelStr = level.toUpperCase().padStart(5);
-        const logMessage = `[${timestamp}] ${levelStr} - ${content}`;
-
-        // 控制台输出
-        if (this.config.console === 1) {
-            console.log(logMessage);
-        }
-
-        // 文件输出
-        await this.writeToFile(logMessage, level);
+    });
+
+    // 控制台输出
+    if (config.console === 1) {
+        targets.push({
+            target: 'pino/file',
+            level: level,
+            options: { destination: 1 }
+        });
     }
 
-    /**
-     * 记录成功日志（使用 info 级别）
-     * @param message - 日志消息
-     */
-    static async success(message: LogMessage): Promise<void> {
-        await this.log('info', message);
-    }
-
-    /**
-     * 写入日志文件
-     * @param message - 格式化后的消息
-     * @param level - 日志级别
-     */
-    static async writeToFile(message: string, level: LogLevel = 'info'): Promise<void> {
-        try {
-            const logDir = this.config.dir || './logs';
-            // 确定文件前缀
-            const prefix = level === 'debug' ? 'debug' : new Date().toISOString().split('T')[0];
-
-            // 检查缓存的当前文件是否仍然可用
-            let currentLogFile = this.currentFiles.get(prefix);
-
-            if (currentLogFile) {
-                try {
-                    const stats = await stat(currentLogFile);
-                    if (stats.size >= (this.config.maxSize || 10 * 1024 * 1024)) {
-                        this.currentFiles.delete(prefix);
-                        currentLogFile = undefined;
-                    }
-                } catch {
-                    this.currentFiles.delete(prefix);
-                    currentLogFile = undefined;
-                }
-            }
-
-            // 查找或创建新文件
-            if (!currentLogFile) {
-                const glob = new Bun.Glob(`${prefix}.*.log`);
-                const files = await Array.fromAsync(glob.scan(this.config.dir || 'logs'));
-
-                // 按索引排序并查找可用文件
-                const getIndex = (f: string) => parseInt(f.match(/\.(\d+)\.log$/)?.[1] || '0');
-                files.sort((a, b) => getIndex(a) - getIndex(b));
-
-                let foundFile = false;
-                for (let i = files.length - 1; i >= 0; i--) {
-                    const filePath = join(this.config.dir || 'logs', files[i]);
-                    try {
-                        const stats = await stat(filePath);
-                        // 检查文件大小
-                        if (stats.size < (this.config.maxSize || 10 * 1024 * 1024)) {
-                            currentLogFile = filePath;
-                            foundFile = true;
-                            break;
-                        }
-                    } catch {
-                        continue;
-                    }
-                }
-
-                // 没有可用文件,创建新文件
-                if (!foundFile) {
-                    const maxIndex = files.length > 0 ? Math.max(...files.map(getIndex)) : -1;
-                    currentLogFile = join(this.config.dir || 'logs', `${prefix}.${maxIndex + 1}.log`);
-                }
+    instance = pino({
+        level: level,
+        transport: { targets: targets }
+    });
 
-                this.currentFiles.set(prefix, currentLogFile);
-            }
-
-            await appendFile(currentLogFile, message + '\n', 'utf8');
-        } catch (error: any) {
-            console.error('写入日志文件失败:', error?.message || error);
-        }
-    }
-
-    /**
-     * 记录错误日志
-     * @param name - 错误名称/位置
-     * @param error - 错误对象或消息
-     */
-    static async error(name: string, error?: any): Promise<void> {
-        if (!error) {
-            return this.log('error', name);
-        }
-
-        // 构建错误消息
-        const parts = [name];
-        if (error?.message || error?.stack) {
-            if (error.message) parts.push(error.message);
-            if (error.stack) parts.push('\n' + error.stack);
-        } else {
-            const errorStr = typeof error === 'object' ? JSON.stringify(error) : String(error);
-            parts.push(errorStr);
-        }
-
-        await this.log('error', parts.join(' - '));
-    }
-
-    /**
-     * 记录警告日志
-     * @param message - 日志消息
-     */
-    static async warn(message: LogMessage): Promise<void> {
-        await this.log('warn', message);
-    }
-
-    /**
-     * 记录信息日志
-     * @param message - 日志消息
-     */
-    static async info(message: LogMessage): Promise<void> {
-        await this.log('info', message);
-    }
-
-    /**
-     * 记录调试日志
-     * 受 enableDebug 配置控制，仅当 enableDebug=true 时才记录
-     * @param message - 日志消息
-     */
-    static async debug(message: LogMessage): Promise<void> {
-        await this.log('debug', message);
-    }
-
-    /**
-     * 清除文件缓存
-     */
-    static clearCache(): void {
-        this.currentFiles.clear();
-    }
+    return instance;
 }
+
+/**
+ * 日志实例（延迟初始化）
+ */
+export const Logger = {
+    get info() {
+        return getLogger().info.bind(getLogger());
+    },
+    get warn() {
+        return getLogger().warn.bind(getLogger());
+    },
+    get error() {
+        return getLogger().error.bind(getLogger());
+    },
+    get debug() {
+        return getLogger().debug.bind(getLogger());
+    },
+    configure: configure,
+    setMock: setMockLogger
+};

package/lib/redisHelper.ts

@@ -6,7 +6,7 @@
 import { SQL, RedisClient } from 'bun';
 import { Logger } from './logger.js';
 import { Connect } from './connect.js';
-import type { KeyValue } from '../types/common.js';
+import type { KeyValue } from 'befly-shared/types';
 
 /**
  * Redis 助手类
@@ -79,6 +79,11 @@ export class RedisHelper {
         }
     }
 
+    // ==================== ID 生成 ====================
+    // 注意：ID 生成功能强依赖 Redis 原子操作（INCR/INCRBY）保证分布式唯一性
+    // 主要被 DbHelper.insData/insBatch 使用
+    // 如未来有其他 ID 生成需求，可考虑提取到独立模块
+
     /**
      * 生成基于时间的唯一 ID
      * 格式: 秒级时间戳(10位) + 4位自增 = 14位纯数字
@@ -209,6 +214,25 @@ export class RedisHelper {
         }
     }
 
+    /**
+     * 批量获取剩余过期时间（利用 Bun Redis 自动管道优化）
+     * @param keys - 键名数组
+     * @returns TTL 数组（-2 表示键不存在，-1 表示无过期时间）
+     */
+    async ttlBatch(keys: string[]): Promise<number[]> {
+        if (keys.length === 0) {
+            return [];
+        }
+
+        try {
+            const results = await Promise.all(keys.map((key) => this.ttl(key)));
+            return results;
+        } catch (error: any) {
+            Logger.error('Redis ttlBatch 错误', error);
+            return keys.map(() => -1);
+        }
+    }
+
     /**
      * 向 Set 中添加一个或多个成员
      * @param key - 键名
@@ -273,6 +297,44 @@ export class RedisHelper {
         }
     }
 
+    /**
+     * 批量向多个 Set 添加成员（利用 Bun Redis 自动管道优化）
+     * @param items - [{ key, members }] 数组
+     * @returns 成功添加的总成员数量
+     */
+    async saddBatch(items: Array<{ key: string; members: string[] }>): Promise<number> {
+        if (items.length === 0) {
+            return 0;
+        }
+
+        try {
+            const results = await Promise.all(items.map((item) => this.sadd(item.key, item.members)));
+            return results.reduce((sum, count) => sum + count, 0);
+        } catch (error: any) {
+            Logger.error('Redis saddBatch 错误', error);
+            return 0;
+        }
+    }
+
+    /**
+     * 批量检查成员是否在 Set 中（利用 Bun Redis 自动管道优化）
+     * @param items - [{ key, member }] 数组
+     * @returns 布尔数组（true 表示存在，false 表示不存在）
+     */
+    async sismemberBatch(items: Array<{ key: string; member: string }>): Promise<boolean[]> {
+        if (items.length === 0) {
+            return [];
+        }
+
+        try {
+            const results = await Promise.all(items.map((item) => this.sismember(item.key, item.member)));
+            return results.map((r) => r > 0);
+        } catch (error: any) {
+            Logger.error('Redis sismemberBatch 错误', error);
+            return items.map(() => false);
+        }
+    }
+
     /**
      * 删除键
      * @param key - 键名
@@ -288,6 +350,106 @@ export class RedisHelper {
         }
     }
 
+    /**
+     * 批量删除键（利用 Bun Redis 自动管道优化）
+     * @param keys - 键名数组
+     * @returns 成功删除的键数量
+     */
+    async delBatch(keys: string[]): Promise<number> {
+        if (keys.length === 0) {
+            return 0;
+        }
+
+        try {
+            const results = await Promise.all(
+                keys.map((key) => {
+                    const pkey = `${this.prefix}${key}`;
+                    return this.client.del(pkey);
+                })
+            );
+            return results.reduce((sum, count) => sum + count, 0);
+        } catch (error: any) {
+            Logger.error('Redis delBatch 错误', error);
+            return 0;
+        }
+    }
+
+    /**
+     * 批量设置对象（利用 Bun Redis 自动管道优化）
+     * @param items - 键值对数组 [{ key, value, ttl? }]
+     * @returns 成功设置的数量
+     */
+    async setBatch<T = any>(items: Array<{ key: string; value: T; ttl?: number | null }>): Promise<number> {
+        if (items.length === 0) {
+            return 0;
+        }
+
+        try {
+            const results = await Promise.all(items.map((item) => this.setObject(item.key, item.value, item.ttl ?? null)));
+            return results.filter((r) => r !== null).length;
+        } catch (error: any) {
+            Logger.error('Redis setBatch 错误', error);
+            return 0;
+        }
+    }
+
+    /**
+     * 批量获取对象（利用 Bun Redis 自动管道优化）
+     * @param keys - 键名数组
+     * @returns 对象数组（不存在的键返回 null）
+     */
+    async getBatch<T = any>(keys: string[]): Promise<Array<T | null>> {
+        if (keys.length === 0) {
+            return [];
+        }
+
+        try {
+            const results = await Promise.all(keys.map((key) => this.getObject<T>(key)));
+            return results;
+        } catch (error: any) {
+            Logger.error('Redis getBatch 错误', error);
+            return keys.map(() => null);
+        }
+    }
+
+    /**
+     * 批量检查键是否存在（利用 Bun Redis 自动管道优化）
+     * @param keys - 键名数组
+     * @returns 布尔数组（true 表示存在，false 表示不存在）
+     */
+    async existsBatch(keys: string[]): Promise<boolean[]> {
+        if (keys.length === 0) {
+            return [];
+        }
+
+        try {
+            const results = await Promise.all(keys.map((key) => this.exists(key)));
+            return results.map((r) => r > 0);
+        } catch (error: any) {
+            Logger.error('Redis existsBatch 错误', error);
+            return keys.map(() => false);
+        }
+    }
+
+    /**
+     * 批量设置过期时间（利用 Bun Redis 自动管道优化）
+     * @param items - 键名和过期时间数组 [{ key, seconds }]
+     * @returns 成功设置的数量
+     */
+    async expireBatch(items: Array<{ key: string; seconds: number }>): Promise<number> {
+        if (items.length === 0) {
+            return 0;
+        }
+
+        try {
+            const results = await Promise.all(items.map((item) => this.expire(item.key, item.seconds)));
+            return results.filter((r) => r > 0).length;
+        } catch (error: any) {
+            Logger.error('Redis expireBatch 错误', error);
+            return 0;
+        }
+    }
+
     /**
      * 测试 Redis 连接
      * @returns ping 响应结果

package/lib/sqlBuilder.ts

@@ -3,7 +3,8 @@
  * 提供链式 API 构建 SQL 查询
  */
 
-import type { WhereConditions, SqlValue, OrderByField } from '../types/common.js';
+import type { WhereConditions, OrderByField } from '../types/common.js';
+import type { SqlValue } from 'befly-shared/types';
 
 /**
  * SQL 构建器类