jsql-neo 1.0.0 → 1.1.0

package/lib/database.js

@@ -28,6 +28,7 @@ class Database {
         this._dirty = false;
         this._autoSaveTimer = null;
         this._loading = false;     // 加载中，不触发自动保存
+        this._transaction = null;   // 事务快照 { tables, autoIncrements }
 
         // 从文件加载
         if (!this._memoryMode && fs.existsSync(this._filePath)) {
@@ -96,6 +97,103 @@ class Database {
         return Object.keys(this._tables);
     }
 
+    // ============================================================
+    // 事务
+    // ============================================================
+
+    /**
+     * 开始事务
+     * 保存当前数据快照，后续操作可回滚
+     */
+    begin() {
+        if (this._transaction) {
+            throw new Error('Transaction already in progress');
+        }
+        this._transaction = {
+            rows: {},
+            autoIncrements: {}
+        };
+        for (const [name, table] of Object.entries(this._tables)) {
+            this._transaction.rows[name] = table._rows.map(r => ({ ...r }));
+            this._transaction.autoIncrements[name] = table._autoIncrement;
+        }
+    }
+
+    /**
+     * 提交事务
+     */
+    commit() {
+        if (!this._transaction) {
+            throw new Error('No transaction in progress');
+        }
+        this._transaction = null;
+        this._markDirty();
+    }
+
+    /**
+     * 回滚事务
+     */
+    rollback() {
+        if (!this._transaction) {
+            throw new Error('No transaction in progress');
+        }
+        for (const [name, table] of Object.entries(this._tables)) {
+            if (this._transaction.rows[name]) {
+                table._rows = this._transaction.rows[name];
+                table._autoIncrement = this._transaction.autoIncrements[name];
+                // 重建索引
+                for (const field of Object.keys(table._indexes)) {
+                    table.createIndex(field);
+                }
+            }
+        }
+        this._transaction = null;
+    }
+
+    /**
+     * 是否在事务中
+     */
+    inTransaction() {
+        return !!this._transaction;
+    }
+
+    // ============================================================
+    // 备份与恢复
+    // ============================================================
+
+    /**
+     * 备份到文件
+     */
+    backup(filePath) {
+        const data = this.export();
+        const json = this._options.pretty
+            ? JSON.stringify(data, null, 2)
+            : JSON.stringify(data);
+        const dir = path.dirname(filePath);
+        if (!fs.existsSync(dir)) {
+            fs.mkdirSync(dir, { recursive: true });
+        }
+        fs.writeFileSync(filePath, json, 'utf8');
+        return filePath;
+    }
+
+    /**
+     * 从文件恢复（覆盖当前数据）
+     */
+    restore(filePath) {
+        if (!fs.existsSync(filePath)) {
+            throw new Error(`Backup file not found: ${filePath}`);
+        }
+        const json = fs.readFileSync(filePath, 'utf8');
+        const data = JSON.parse(json);
+        for (const [name, rows] of Object.entries(data)) {
+            if (this._tables[name]) {
+                this._tables[name]._loadRows(rows);
+            }
+        }
+        this._markDirty();
+    }
+
     // ============================================================
     // 持久化
     // ============================================================

package/lib/table.js

@@ -13,6 +13,14 @@ class Table {
         this._indexes = {};       // 字段名 -> Map<value, [rowIndex, ...]>
         this._autoIncrement = 0;  // 自增计数器
         this._dirty = false;      // 是否有未保存的修改
+        this._hooks = {             // 事件钩子
+            beforeInsert: [],
+            afterInsert: [],
+            beforeUpdate: [],
+            afterUpdate: [],
+            beforeRemove: [],
+            afterRemove: []
+        };
 
         // 解析主键和自增字段
         this._primaryKey = null;
@@ -31,6 +39,11 @@ class Table {
     // ============================================================
 
     insert(data) {
+        // 触发 beforeInsert 钩子
+        for (const hook of this._hooks.beforeInsert) {
+            data = hook({ ...data }) || data;
+        }
+
         // 校验必填字段
         for (const [field, def] of Object.entries(this._schema)) {
             if (def.required && (data[field] === undefined || data[field] === null)) {
@@ -67,6 +80,12 @@ class Table {
 
         this._dirty = true;
         this._db._markDirty();
+
+        // 触发 afterInsert 钩子
+        for (const hook of this._hooks.afterInsert) {
+            hook(data);
+        }
+
         return data;
     }
 
@@ -139,6 +158,11 @@ class Table {
     // ============================================================
 
     update(query, updates) {
+        // 触发 beforeUpdate 钩子
+        for (const hook of this._hooks.beforeUpdate) {
+            updates = hook(query, { ...updates }) || updates;
+        }
+
         const matched = this._applyFilter(this._rows, query);
         let count = 0;
 
@@ -172,6 +196,11 @@ class Table {
         if (count > 0) {
             this._dirty = true;
             this._db._markDirty();
+
+            // 触发 afterUpdate 钩子
+            for (const hook of this._hooks.afterUpdate) {
+                hook(query, updates, count);
+            }
         }
         return count;
     }
@@ -188,6 +217,12 @@ class Table {
     // ============================================================
 
     remove(query = {}) {
+        // 触发 beforeRemove 钩子
+        for (const hook of this._hooks.beforeRemove) {
+            const result = hook(query);
+            if (result === false) return 0; // 钩子取消删除
+        }
+
         if (Object.keys(query).length === 0) {
             // 删除全部
             const count = this._rows.length;
@@ -225,6 +260,11 @@ class Table {
         if (toRemove.length > 0) {
             this._dirty = true;
             this._db._markDirty();
+
+            // 触发 afterRemove 钩子
+            for (const hook of this._hooks.afterRemove) {
+                hook(query, toRemove.length);
+            }
         }
         return toRemove.length;
     }
@@ -258,6 +298,35 @@ class Table {
         delete this._indexes[field];
     }
 
+    // ============================================================
+    // 事件钩子
+    // ============================================================
+
+    /**
+     * 注册事件钩子
+     * @param {string} event - 'beforeInsert' | 'afterInsert' | 'beforeUpdate' | 'afterUpdate' | 'beforeRemove' | 'afterRemove'
+     * @param {function} callback
+     *   - beforeInsert(data) => 返回修改后的 data
+     *   - afterInsert(data)
+     *   - beforeUpdate(query, updates) => 返回修改后的 updates
+     *   - afterUpdate(query, updates, count)
+     *   - beforeRemove(query) => 返回 false 阻止删除
+     *   - afterRemove(query, count)
+     */
+    on(event, callback) {
+        if (this._hooks[event]) {
+            this._hooks[event].push(callback);
+        }
+        return this;
+    }
+
+    off(event, callback) {
+        if (this._hooks[event]) {
+            this._hooks[event] = this._hooks[event].filter(cb => cb !== callback);
+        }
+        return this;
+    }
+
     // ============================================================
     // 内部方法
     // ============================================================
@@ -386,6 +455,7 @@ class Query {
         this._limitCount = null;
         this._offsetCount = 0;
         this._selectFields = null;
+        this._groupBy = null;
     }
 
     orderBy(field, dir = 'asc') {
@@ -409,6 +479,11 @@ class Query {
         return this;
     }
 
+    groupBy(field) {
+        this._groupBy = field;
+        return this;
+    }
+
     get() {
         let rows = this._table._applyFilter(this._table._rows, this._conditions);
 
@@ -456,6 +531,72 @@ class Query {
         return rows.length;
     }
 
+    sum(field) {
+        const rows = this._applyGroupedRows();
+        return rows.reduce((acc, r) => acc + (Number(r[field]) || 0), 0);
+    }
+
+    avg(field) {
+        const rows = this._applyGroupedRows();
+        if (rows.length === 0) return 0;
+        return this.sum(field) / rows.length;
+    }
+
+    min(field) {
+        const rows = this._applyGroupedRows();
+        if (rows.length === 0) return null;
+        return rows.reduce((min, r) => {
+            const v = Number(r[field]);
+            return (min === null || v < min) ? v : min;
+        }, null);
+    }
+
+    max(field) {
+        const rows = this._applyGroupedRows();
+        if (rows.length === 0) return null;
+        return rows.reduce((max, r) => {
+            const v = Number(r[field]);
+            return (max === null || v > max) ? v : max;
+        }, null);
+    }
+
+    /**
+     * 获取分组统计结果
+     * @returns {Array<{ _group: value, _count, _sum, _avg, _min, _max }>}
+     */
+    groupStats(field) {
+        if (!this._groupBy) throw new Error('groupStats() requires groupBy() to be called first');
+        const rows = this._table._applyFilter(this._table._rows, this._conditions);
+        const groups = {};
+
+        for (const row of rows) {
+            const key = String(row[this._groupBy] ?? '__null__');
+            if (!groups[key]) {
+                groups[key] = { _group: row[this._groupBy], _count: 0, _sum: 0, _min: null, _max: null, _rows: [] };
+            }
+            const g = groups[key];
+            g._count++;
+            const v = Number(row[field]) || 0;
+            g._sum += v;
+            g._min = g._min === null ? v : Math.min(g._min, v);
+            g._max = g._max === null ? v : Math.max(g._max, v);
+            g._rows.push(row);
+        }
+
+        const result = Object.values(groups);
+        result.forEach(g => {
+            g._avg = g._count > 0 ? g._sum / g._count : 0;
+            delete g._rows;
+        });
+
+        return result;
+    }
+
+    _applyGroupedRows() {
+        let rows = this._table._applyFilter(this._table._rows, this._conditions);
+        return rows;
+    }
+
     update(updates) {
         return this._table.update(this._conditions, updates);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "jsql-neo",
-    "version": "1.0.0",
+    "version": "1.1.0",
     "description": "JSQL — Pure JavaScript embedded database. SQL-like API, zero native dependencies, file-based or in-memory storage.",
     "main": "index.js",
     "files": [