mm_sqlite 1.5.3 → 1.5.4

package/index.js

@@ -257,6 +257,79 @@ SQLite.prototype.exec = async function (sql, params = [], timeout) {
   return ret;
 };
 
+/**
+ * 执行多语句SQL脚本（专用于导入）
+ * 逐条执行DML语句统计变更行数，其他语句使用exec执行
+ * @param {string} sql - SQL脚本内容，可含多条以分号分隔的语句
+ * @returns {Promise<object>} 执行结果统计对象
+ * @throws {TypeError} 当sql参数无效时
+ */
+SQLite.prototype.runSql = async function (sql) {
+  // 参数校验
+  if (typeof sql !== 'string' || sql.trim() === '') {
+    throw new TypeError('sql must be non-empty string');
+  }
+
+  // 注意：同 scope 下为单例共享，this.error/this.sql 并发时以最后一次写入为准
+  this.error = null;
+  this.sql = '';
+  let conn = null;
+  let ret = { success: false, statements: 0, affected_rows: 0, error: null };
+  try {
+    // 获取连接
+    conn = await this.getConn();
+    this.sql = sql;
+
+    // 拆分语句列表
+    let stmts = splitStatements(sql);
+    ret.statements = stmts.length;
+    // 逐条执行，DML语句统计变更行数
+    for (let stmt_sql of stmts) {
+      if (/^(INSERT|UPDATE|DELETE|REPLACE)\b/i.test(stmt_sql)) {
+        let info = conn.prepare(stmt_sql).run();
+        ret.affected_rows += info.changes || 0;
+      } else {
+        conn.exec(stmt_sql);
+      }
+    }
+    ret.success = true;
+  } catch (error) {
+    console.error('SQL执行失败', error);
+    this.sql = sql;
+    this.error = {
+      code: error.code || error.errno,
+      message: error.message
+    };
+    ret.error = error.message;
+  } finally {
+    await releaseConn(conn);
+  }
+  return ret;
+};
+
+/**
+ * 拆分SQL脚本为语句列表
+ * 去除注释后按分号分割，过滤空语句
+ * @private
+ * @param {string} sql - SQL脚本内容
+ * @returns {Array} 语句列表
+ */
+function splitStatements(sql) {
+  // 去除行注释和块注释
+  let clean_sql = sql
+    .replace(/--[^\n]*/g, '')
+    .replace(/\/\*[\s\S]*?\*\//g, '')
+    .trim();
+  let stmts = [];
+  for (let part of clean_sql.split(';')) {
+    let stmt = part.trim();
+    if (stmt !== '') {
+      stmts.push(stmt);
+    }
+  }
+  return stmts;
+}
+
 /**
  * 读取整张表的数据
  * @param {string} table - 表名

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "mm_sqlite",
-    "version": "1.5.3",
+    "version": "1.5.4",
     "description": "高性能SQLite数据库操作模块，提供与mm_mysql完全兼容的API接口，支持异步操作、事务处理和批量操作",
     "main": "index.js",
     "scripts": {