mm_mysql 2.0.2 → 2.0.4

package/db.js

@@ -7,10 +7,12 @@ const Sql = require('./sql');
 class DB extends Sql {
 	/**
 	 * @description 数据库管理器
-	 * @param {String} database 数据库名称
+	 * @param {Object} mysql - MySQL实例
 	 */
 	constructor(mysql) {
 		super(mysql.run, mysql.exec);
+		// 保存mysql实例引用
+		this._mysql = mysql;
 		// 事务中
 		this.task = 0;
 
@@ -27,6 +29,21 @@ class DB extends Sql {
 	}
 }
 
+/**
+ * 创建超时Promise
+ * @private
+ * @param {Number} timeout - 超时时间(毫秒)
+ * @param {String} message - 错误信息
+ * @returns {Promise<Error>}
+ */
+DB.prototype._createTimeoutPromise = function(timeout, message) {
+	return new Promise((_, reject) => {
+		setTimeout(() => {
+			reject(new Error(message));
+		}, timeout);
+	});
+};
+
 /**
  * 获取数据库名
  * @return {String} 数据库
@@ -42,400 +59,613 @@ DB.prototype.database = function() {
  * @return {Object} 返回管理器
  */
 DB.prototype.new = function(table, key) {
-	var db = this.parent().db();
+	const db = this.parent().db();
 	db.table = table;
 	if (key) {
 		db.key = key;
 	} else {
-		var arr = table.split("_");
+		const arr = table.split("_");
 		db.key = arr[arr.length - 1] + "_id";
 	}
 	return db;
 };
 
+/**
+ * 执行SQL查询
+ * @param {string} sql SQL语句
+ * @param {Array} params 参数数组
+ * @param {Number} timeout 超时时间(毫秒)
+ * @returns {Promise<Object>} 查询结果
+ */
+DB.prototype.run = async function(sql, params = [], timeout = 30000) {
+	if (!this._mysql) {
+		throw new Error('MySQL实例未初始化');
+	}
+	try {
+		// 使用Promise.race实现超时控制
+		return await Promise.race([
+			this._mysql.run(sql, params),
+			this._createTimeoutPromise(timeout, `查询执行超时: ${sql.substring(0, 100)}...`)
+		]);
+	} catch (error) {
+		$.log.error(`[DB] [run] 查询执行失败`, {
+			error: error.message,
+			sql: sql.substring(0, 200),
+			params: JSON.stringify(params.slice(0, 5))
+		});
+		throw error;
+	}
+};
+
+/**
+ * 执行SQL命令
+ * @param {string} sql SQL语句
+ * @param {Number} timeout 超时时间(毫秒)
+ * @returns {Promise<Object>} 执行结果
+ */
+DB.prototype.exec = async function(sql, timeout = 30000) {
+	if (!this._mysql) {
+		throw new Error('MySQL实例未初始化');
+	}
+	try {
+		// 使用Promise.race实现超时控制
+		return await Promise.race([
+			this._mysql.exec(sql),
+			this._createTimeoutPromise(timeout, `命令执行超时: ${sql.substring(0, 100)}...`)
+		]);
+	} catch (error) {
+		$.log.error(`[DB] [exec] 命令执行失败`, {
+			error: error.message,
+			sql: sql.substring(0, 200)
+		});
+		throw error;
+	}
+};
+
+/**
+ * 获取数据库连接
+ * @param {Number} timeout 超时时间(毫秒)
+ * @returns {Promise<Object>} 数据库连接对象
+ */
+DB.prototype.getConn = async function(timeout = 15000) {
+	if (!this._mysql) {
+		throw new Error('MySQL实例未初始化');
+	}
+	try {
+		// 使用Promise.race实现超时控制
+		return await Promise.race([
+			this._mysql.getConn(),
+			this._createTimeoutPromise(timeout, '获取数据库连接超时')
+		]);
+	} catch (error) {
+		$.log.error(`[DB] [getConn] 获取连接失败`, {
+			error: error.message
+		});
+		throw error;
+	}
+};
+
 /**
  * 事务开始
+ * @param {String} identifier 事务标识符
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise<Object>} 执行结果
  */
-DB.prototype.start = async function(identifier = "point_1") {
+DB.prototype.start = async function(identifier = "point_1", timeout = 15000) {
 	this.task = 1;
-	return await this.exec("SET AUTOCOMMIT=0;BEGIN;");
+	try {
+		return await this.exec("SET AUTOCOMMIT=0;BEGIN;", timeout);
+	} catch (err) {
+		this.task = 0;
+		$.log.error(`[DB] [start] 事务开始失败: ${err.message}`);
+		throw err;
+	}
 };
 
 /**
  * 提交
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise<Object>} 执行结果
  */
-DB.prototype.commit = async function() {
+DB.prototype.commit = async function(timeout = 15000) {
+	if (this.task !== 1) {
+		$.log.warn("[DB] [commit] 没有活跃的事务可提交");
+		return 0;
+	}
 	this.task = 2;
-	return this.exec("COMMIT;");
+	try {
+		const result = await this.exec("COMMIT;", timeout);
+		// 重置事务状态
+		this.task = 0;
+		this.task_sql = '';
+		return result;
+	} catch (err) {
+		$.log.error(`[DB] [commit] 事务提交失败: ${err.message}`);
+		// 尝试回滚
+		try {
+			await this.exec("ROLLBACK;", timeout);
+		} catch (rollbackErr) {
+			$.log.error(`[DB] [commit] 事务回滚也失败: ${rollbackErr.message}`);
+		}
+		this.task = 0;
+		this.task_sql = '';
+		throw err;
+	}
 };
 
 /**
  * 事务结束
  */
-DB.prototype.end = async function() {
-	this.task = 3;
+DB.prototype.end = function() {
+	// 确保事务状态被重置
+	this.task = 0;
+	this.task_sql = '';
 };
 
 /**
  * 滚回
+ * @param {String} identifier 事务标识符
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise<Object>} 执行结果
  */
-DB.prototype.back = async function(identifier = "point_1") {
+DB.prototype.back = async function(identifier = "point_1", timeout = 15000) {
+	if (this.task !== 1) {
+		$.log.warn("[DB] [back] 没有活跃的事务可回滚");
+		this.task = 0;
+		this.task_sql = '';
+		return 0;
+	}
 	this.task = 3;
-	await this.exec("ROLLBACK;");
-	return this.exec("SET AUTOCOMMIT=1;");
+	try {
+		await this.exec("ROLLBACK;", timeout);
+		const result = await this.exec("SET AUTOCOMMIT=1;", timeout);
+		// 重置事务状态
+		this.task = 0;
+		this.task_sql = '';
+		return result;
+	} catch (err) {
+		$.log.error(`[DB] [back] 事务回滚失败: ${err.message}`);
+		this.task = 0;
+		this.task_sql = '';
+		throw err;
+	}
 };
 
 /**
  * @description 获取所有表名
  * @param {String} table 表名关键词, 可以 *table*包含、*table后缀、table*前缀 查询
+ * @param {Number} timeout 超时时间(毫秒)
  * @return {Promise|Array} 表名数组
  */
-DB.prototype.tables = async function(table) {
-	var list = await this.run("show tables");
-	var key = 'Tables_in_' + this.database();
-	if (table) {
-		list = list.search(table, key);
+DB.prototype.tables = async function(table, timeout = 15000) {
+	try {
+		const list = await this.run("show tables", [], timeout);
+		const key = 'Tables_in_' + this.database();
+		if (table) {
+			// 简单的过滤实现
+			const filtered = [];
+			for (const item of list) {
+				const value = item[key];
+				if (value && value.includes(table)) {
+					filtered.push(item);
+				}
+			}
+			return filtered.map(item => item[key]);
+		}
+		// 使用原生方法提取表名
+		return list.map(item => item[key]);
+	} catch (err) {
+		$.log.error(`[DB] [tables] 获取表名失败: ${err.message}`);
+		return [];
 	}
-	return list.getArr(key);
 };
 
 /**
  * @description 获取所有表字段
  * @param {String} table 表名
+ * @param {String} field_name 字段名，可选
+ * @param {Number} timeout 超时时间(毫秒)
  * @return {Promise|Array} 字段信息列表
  */
-DB.prototype.fields = async function(table, field_name) {
-	if (!table) {
-		table = this.table;
-	}
-	var field =
-		'COLUMN_NAME as `name`,ORDINAL_POSITION as `cid`,COLUMN_DEFAULT as `dflt_value`,IS_NULLABLE as `notnull`,COLUMN_TYPE as `type`,COLUMN_KEY as `pk`,EXTRA as `auto`,COLUMN_COMMENT as `note`';
-	var sql = "select " + field + " from information_schema.COLUMNS where `table_name` = '" + table +
-		"' and `table_schema` = '" + this.database() + "'";
-	if (field_name) {
-		sql += " AND COLUMN_NAME='" + field_name + "'";
-	}
-	var list = await this.run(sql);
-	var len = list.length;
-	for (var i = 0; i < len; i++) {
-		list[i].pk = list[i].pk ? true : false;
-		list[i].notnull = list[i].notnull == 'NO' ? true : false;
-	}
-	return list;
-};
-
-/**
- * @description 设置类型
- * @param {String} field 字段名
- * @param {String} type 类型名，常用类型 mediumint, int, varchar, datetime
- * @param {String} value 默认值
- * @param {Boolean} not_null 是否非空字段 true为非空，false为可空
- * @param {Boolean} auto 自动
- * @return {String} 返回最终类型
- */
-DB.prototype.setType = function(field, type, value, not_null, auto) {
-	if (!type) {
-		type = 'int';
-	}
-	switch (type) {
-		case "str":
-		case "varchar":
-		case "string":
-			type = "varchar(255)";
-			if (not_null) {
-				type += " NOT NULL";
-			}
-			if (value) {
-				type += " DEFAULT '" + value + "'";
-			} else {
-				type += " DEFAULT ''";
-			}
-			break;
-		case "number":
-			type = "int(11) NOT NULL";
-			if (auto) {
-				type += " AUTO_INCREMENT";
-			} else if (value) {
-				type += " DEFAULT " + value;
-			} else {
-				type += " DEFAULT 0";
-			}
-			break;
-		case "bool":
-		case "tinyint":
-			type = "tinyint(1) UNSIGNED NOT NULL";
-			if (value) {
-				type += " DEFAULT " + value;
-			} else {
-				type += " DEFAULT 0";
-			}
-			break;
-		case "datetime":
-			if (!value) {
-				value = "1970-01-01 00:00:00";
-			}
-			type += " DEFAULT '" + value + "'";
-			break;
-		case "timestamp":
-			if (auto) {
-				if (field.indexOf('update') !== -1 || field.indexOf('last') !== -1) {
-					type += " DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP";
-				} else {
-					type += " DEFAULT CURRENT_TIMESTAMP";
-				}
-			} else {
-				type += " DEFAULT CURRENT_TIMESTAMP";
-			}
-			break;
-		case "date":
-			if (!value) {
-				value = "1970-01-01";
-			}
-			type += " NOT NULL DEFAULT '" + value + "'";
-			break;
-		case "time":
-			if (!value) {
-				value = "00:00:00";
-			}
-			type += " NOT NULL DEFAULT '" + value + "'";
-			break;
-		case "double":
-			if (type == "double") {
-				type = "double(10, 2)"
-			}
-			type += " NOT NULL";
-			if (value) {
-				type += " DEFAULT '" + value + "'";
-			} else {
-				type += " DEFAULT 0";
-			}
-			break;
-		case "float":
-			if (type == "float") {
-				type = "float(17, 8)"
-			}
-			type += " NOT NULL";
-			if (value) {
-				type += " DEFAULT '" + value + "'";
-			} else {
-				type += " DEFAULT 0";
-			}
-			break;
-		case "longtext":
-		case "text":
-			if (type == "text") {
-				type = "text NULL"
-			}
-			else if (type == "longtext") {
-				type = "longtext NULL"
-			}
-			break;
-		default:
-			if (type.indexOf('var') !== -1) {
-				if (not_null) {
-					type += " NOT NULL"
-				}
-				if (value) {
-					type += " DEFAULT '" + value + "'";
-				} else {
-					type += " DEFAULT ''";
-				}
-			} else {
-				type += " UNSIGNED NOT NULL";
-				if (auto) {
-					type += " AUTO_INCREMENT";
-				} else {
-					if (value) {
-						type += " DEFAULT '" + value + "'";
-					} else {
-						type += " DEFAULT 0";
-					}
-				}
-			}
-			break;
+DB.prototype.fields = async function(table, field_name, timeout = 15000) {
+	try {
+		const targetTable = table || this.table;
+		if (!targetTable) {
+			$.log.error("[DB] [fields] 未指定表名");
+			return [];
+		}
+		const field = 'COLUMN_NAME as `name`,ORDINAL_POSITION as `cid`,COLUMN_DEFAULT as `dflt_value`,IS_NULLABLE as `notnull`,COLUMN_TYPE as `type`,COLUMN_KEY as `pk`,EXTRA as `auto`,COLUMN_COMMENT as `note`';
+		let sql = "select " + field + " from information_schema.COLUMNS where `table_name` = '" + targetTable +
+			"' and `table_schema` = '" + this.database() + "'";
+		if (field_name) {
+			sql += " AND COLUMN_NAME='" + field_name + "'";
+		}
+		const list = await this.run(sql, [], timeout);
+		const len = list.length;
+		for (let i = 0; i < len; i++) {
+			list[i].pk = !!list[i].pk;
+			list[i].notnull = list[i].notnull === 'NO';
+		}
+		return list;
+	} catch (err) {
+		$.log.error(`[DB] [fields] 获取字段信息失败: ${err.message}`);
+		return [];
 	}
-	return type;
 };
 
 /**
  * @description 创建数据表
  * @param {String} table 表名
  * @param {String} field 主键字段名
- * @param {String} type  类型名，常用类型 mediumint, int, varchar
+ * @param {String} type 类型名，常用类型 mediumint, int, varchar
  * @param {Boolean} auto 是否自增字段, 默认为自增字段
  * @param {String} commit 注释
+ * @param {Number} timeout 超时时间(毫秒)
  * @return {Promise|Number} 创建成功返回1，失败返回0
  */
-DB.prototype.addTable = async function(table, field, type, auto, commit = '') {
-	if (!field) {
-		field = "id";
-	}
-	var sql = "CREATE TABLE IF NOT EXISTS `{0}` (`{1}` {2})".replace('{0}', table).replace(
-		'{1}', field).replace('{2}', this.setType(field, type, null, true, auto) + ' PRIMARY KEY');
-	if (commit) {
-		sql += " COMMENT = '" + commit + "';"
-	} else {
-		sql += ";"
+DB.prototype.addTable = async function(table, field, type = 'int', auto = true, commit = '', timeout = 15000) {
+	try {
+		if (!table || typeof table !== 'string') {
+			$.log.error("[DB] [addTable] 表名无效");
+			return 0;
+		}
+		if (!field) {
+			field = "id";
+		}
+		// 构建完整的字段定义
+		let fieldDef = `\`${field}\` ${type}(11)`;
+		fieldDef += " NOT NULL";
+		if (auto) {
+			fieldDef += " AUTO_INCREMENT";
+		}
+		if (commit) {
+			fieldDef += " COMMENT '" + commit + "'";
+		}
+		fieldDef += " PRIMARY KEY";
+		
+		let sql = "CREATE TABLE IF NOT EXISTS \`" + table + "\` (" + fieldDef + ")";
+		sql += " ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;";
+		
+		// 执行SQL并设置表名
+		const result = await this.exec(sql, timeout);
+		// 设置实例的表名属性，以便后续操作使用
+		this.table = table;
+		return result && result.affectedRows !== undefined ? 1 : 0;
+	} catch (err) {
+		$.log.error(`[DB] [addTable] 创建表失败: ${err.message}`);
+		return 0;
 	}
-	var bl = await this.exec(sql);
-	return bl;
 };
 
 /**
- * @description 清空数据表
- * @param {Boolean} reset 是否重置自增ID
- * @return {Promise|Number} 清空成功返回1，失败返回0
+ * @description 删除表
+ * @param {String} table 表名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
  */
-DB.prototype.clearTable = async function(reset = true, table = '') {
-	var sql = reset ? "TRUNCATE table `{0}`;" : "DELETE FROM `{0}`";
-	var bl = await this.exec(sql.replace('{0}', table || this.table));
-	return bl;
+DB.prototype.dropTable = function(table, timeout = 15000) {
+	if (!table || typeof table !== 'string') {
+		$.log.error("[DB] [dropTable] 表名无效");
+		return 0;
+	}
+	return this.exec("DROP TABLE IF EXISTS \`" + table + "\`;", timeout);
+};
+
+/**
+ * @description 修改表名
+ * @param {String} table 旧表名
+ * @param {String} new_table 新表名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
+ */
+DB.prototype.renameTable = function(table, new_table, timeout = 15000) {
+	if (!table || !new_table) {
+		$.log.error("[DB] [renameTable] 表名参数不完整");
+		return 0;
+	}
+	return this.exec("RENAME TABLE \`" + table + "\` TO \`" + new_table + "\`;", timeout);
 };
 
 /**
  * @description 添加字段
  * @param {String} field 字段名
- * @param {String} type 类型名，常用类型 mediumint, int, float, double, varchar, tinyint, text, date, datetime, time, timestamp
- * @param {String|Number} value 默认值
- * @param {Boolean} not_null 是否非空字段 true为非空，false为可空
- * @param {Boolean} auto 是否自动（如果为数字类型则自增增段，如果为时间类型则默认事件）
- * @param {Boolean} isKey 是否主键
+ * @param {String} type 类型名
+ * @param {String} value 默认值
+ * @param {Boolean} not_null 是否非空
+ * @param {Boolean} auto 是否自增
+ * @param {String} comment 注释
+ * @param {Number} timeout 超时时间(毫秒)
  * @return {Promise|Number} 添加成功返回1，失败返回0
  */
-DB.prototype.field_add = async function(field, type, value, not_null, auto, isKey) {
-	var sql =
-		"SELECT COUNT(*) as `count` FROM information_schema.COLUMNS WHERE TABLE_SCHEMA='{0}' AND table_name='{1}' AND COLUMN_NAME='{2}'";
-	sql = sql.replace('{0}', this.database()).replace('{1}', this.table).replace('{2}', field);
-	var arr = await this.run(sql);
-	if (arr && arr.length > 0) {
-		if (arr[0].count == 0) {
-			var type = this.setType(field, type, value, not_null, auto);
-			var sql = "ALTER Table `{0}` ADD `{1}` {2}";
-			sql = sql.replace('{0}', this.table).replace('{1}', field).replace('{2}', type);
-			if (isKey) {
-				sql += " , ADD PRIMARY KEY (`" + field + "`)";
+DB.prototype.field_add = async function(field, type, value = '', not_null = true, auto = false, comment = '', timeout = 15000) {
+	try {
+		// 确保表名已设置
+		if (!this.table || !field || !type) {
+			$.log.error("[DB] [field_add] 表名、字段名或类型未指定");
+			return 0;
+		}
+		
+		// 构建字段定义
+		let fieldDef = `\`${field}\` ${type}`;
+		
+		// 添加非空约束
+		if (not_null) {
+			fieldDef += " NOT NULL";
+		}
+		
+		// 添加自增属性
+		if (auto) {
+			fieldDef += " AUTO_INCREMENT";
+		}
+		
+		// 添加默认值
+		if (value !== undefined && value !== null && value !== '') {
+			if (typeof value === 'string') {
+				fieldDef += " DEFAULT '" + value + "'";
 			} else {
-				sql += ";";
+				fieldDef += " DEFAULT " + value;
 			}
-			return await this.exec(sql);
 		}
+		
+		// 添加注释
+		if (comment) {
+			fieldDef += " COMMENT '" + comment + "'";
+		}
+		
+		// 使用ADD COLUMN而不是CHANGE COLUMN
+		const sql = `ALTER TABLE \`${this.table}\` ADD COLUMN ${fieldDef};`;
+		const result = await this.exec(sql, timeout);
+		return result && result.affectedRows !== undefined ? 1 : 0;
+	} catch (err) {
+		$.log.error(`[DB] [field_add] 添加字段失败: ${err.message}`);
+		return 0;
 	}
-	return 0;
 };
 
 /**
- * @description 添加字段
+ * @description 修改字段
  * @param {String} field 字段名
- * @param {String} type 类型名，常用类型 mediumint, int, float, double, varchar, tinyint, text, date, datetime, time, timestamp
- * @param {String|Number} value 默认值
- * @param {Boolean} not_null 是否非空字段 true为非空，false为可空
- * @param {Boolean} auto 是否自动（如果为数字类型则自增增段，如果为时间类型则默认事件）
+ * @param {String} type 类型名
+ * @param {String} value 默认值
+ * @param {Boolean} not_null 是否非空
+ * @param {Boolean} auto 是否自增
  * @param {Boolean} isKey 是否主键
- * @param {String} new_name 新名称
- * @return {Promise|Number} 添加成功返回1，失败返回0
+ * @param {String} new_name 新字段名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 修改成功返回1，失败返回0
  */
-DB.prototype.field_set = async function(field, type, value, not_null, auto, isKey, new_name) {
-	var sql =
-		"SELECT COUNT(*) as `count` FROM information_schema.COLUMNS WHERE TABLE_SCHEMA='{0}' AND table_name='{1}' AND COLUMN_NAME='{2}'";
-	sql = sql.replace('{0}', this.database()).replace('{1}', this.table).replace('{2}', field);
-	var arr = await this.run(sql);
-	if (arr && arr.length > 0) {
-		if (arr[0].count == 0) {
+DB.prototype.field_set = async function(field, type, value = '', not_null = true, auto = false, isKey = false, new_name = '', timeout = 15000) {
+	try {
+		if (!field || !type || !this.table) {
+			$.log.error("[DB] [field_set] 参数不完整或表名未设置");
 			return 0;
 		}
+		const targetName = new_name || field;
+		let fieldDef = `\`${targetName}\` ${type}`;
+		if (not_null) {
+			fieldDef += " NOT NULL";
+		}
+		if (auto) {
+			fieldDef += " AUTO_INCREMENT";
+		}
+		if (value !== undefined && value !== null && value !== '') {
+			if (typeof value === 'string') {
+				fieldDef += " DEFAULT '" + value + "'";
+			} else {
+				fieldDef += " DEFAULT " + value;
+			}
+		}
+		let sql = `ALTER TABLE \`${this.table}\` CHANGE COLUMN \`${field}\` ${fieldDef}`;
+		if (isKey) {
+			sql += ", ADD PRIMARY KEY (\`" + targetName + "\`)";
+		}
+		sql += ";";
+		const result = await this.exec(sql, timeout);
+		return result && result.affectedRows !== undefined ? 1 : 0;
+	} catch (err) {
+		$.log.error(`[DB] [field_set] 修改字段失败: ${err.message}`);
+		return 0;
 	}
+};
 
-	var type = this.setType(field, type, value, not_null, auto);
-	if (type.has('text')) {
-		type = type.replace('NOT NULL', '').replace("DEFAULT ''", "");
+/**
+ * @description 修改字段类型
+ * @param {String} table 表名
+ * @param {String} field 字段名
+ * @param {String} type 字段类型
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
+ */
+DB.prototype.field_edit = function(table, field, type, timeout = 15000) {
+	if (!table || !field || !type) {
+		$.log.error("[DB] [field_edit] 参数不完整");
+		return 0;
 	}
+	return this.exec("ALTER TABLE \`" + table + "\` MODIFY COLUMN \`" + field + "\` " + type + ";", timeout);
+};
 
-	if (!new_name) {
-		new_name = field;
+/**
+ * @description 删除字段
+ * @param {String} table 表名
+ * @param {String} field 字段名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
+ */
+DB.prototype.field_del = function(table, field, timeout = 15000) {
+	if (!table || !field) {
+		$.log.error("[DB] [field_del] 参数不完整");
+		return 0;
 	}
+	return this.exec("ALTER TABLE \`" + table + "\` DROP COLUMN \`" + field + "\`;", timeout);
+};
 
-	sql = "ALTER TABLE `{0}` CHANGE COLUMN `{1}` `{2}` {3}";
-	sql = sql.replace('{0}', this.table).replace('{1}', field).replace('{2}', new_name).replaceAll('{3}', type);
-	if (isKey) {
-		sql += ", DROP PRIMARY KEY, ADD PRIMARY KEY (" + new_name + ") USING BTREE;"
-	} else {
-		sql += ";";
+/**
+ * @description 修改字段名
+ * @param {String} table 表名
+ * @param {String} field 旧字段名
+ * @param {String} new_field 新字段名
+ * @param {String} type 字段类型
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
+ */
+DB.prototype.field_rename = function(table, field, new_field, type, timeout = 15000) {
+	if (!table || !field || !new_field) {
+		$.log.error("[DB] [field_rename] 参数不完整");
+		return 0;
 	}
-	return await this.exec(sql);
+	return this.exec("ALTER TABLE \`" + table + "\` CHANGE \`" + field + "\` \`" + new_field + "\` " + type + ";", timeout);
 };
 
 /**
- * @description 删除字段
- * @param {String} field 字段名
- * @return {Promise|Number} 删除成功返回1，失败返回0
+ * @description 获取表创建语句
+ * @param {String} table 表名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|String} 创建语句
  */
-DB.prototype.field_del = async function(field) {
-	var sql =
-		"SELECT COUNT(*) as `count` FROM information_schema.COLUMNS WHERE TABLE_SCHEMA='{0}' AND table_name='{1}' AND COLUMN_NAME='{2}'";
-	sql = sql.replace('{0}', this.database()).replace('{1}', this.table).replace('{2}', field);
-	var arr = await this.run(sql);
-	if (arr && arr.length > 0) {
-		if (arr[0].count > 0) {
-			sql = "ALTER Table `{0}` DROP `{1}`;".replace('{0}', this.table).replace('{1}', field);
-			return await this.exec(sql);
+DB.prototype.getCreateTable = async function(table, timeout = 15000) {
+	if (!table) {
+		$.log.error("[DB] [getCreateTable] 表名不能为空");
+		return '';
+	}
+	try {
+		const list = await this.run("SHOW CREATE TABLE \`" + table + "\`;", [], timeout);
+		if (list && list.length > 0) {
+			return list[0]["Create Table"] || list[0]["create table"];
 		}
+		return '';
+	} catch (err) {
+		$.log.error(`[DB] [getCreateTable] 获取表结构失败: ${err.message}`);
+		return '';
 	}
-	return 0;
 };
 
 /**
- * @description 拼接字段信息SQL
- * @param {Object} fd 字段信息
- * @return {String} sql语段
+ * @description 获取表数据的SQL插入语句
+ * @param {String} table 表名
+ * @param {Number} batchSize 批量处理大小
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|String} SQL插入语句
  */
-DB.prototype.field_sql = function(fd) {
-	var sql = "`{0}`".replace('{0}', fd.name);
-	sql += " " + fd.type;
-	if (fd.notnull) {
-		sql += " NOT NULL";
-	}
-	if (fd.auto) {
-		if (fd.dflt_value) {
-			if (fd.dflt_value === '0000-00-00 00:00:00') {
-				fd.dflt_value = 'CURRENT_TIMESTAMP';
+DB.prototype.getTableData = async function(table, batchSize = 100, timeout = 60000) {
+	if (!table) {
+		$.log.error("[DB] [getTableData] 表名不能为空");
+		return '';
+	}
+	try {
+		// 获取表数据
+		const rows = await this.run(`SELECT * FROM \`${table}\``, [], timeout);
+		let sql = '';
+		
+		if (rows && rows.length > 0) {
+			sql += `-- 表数据: ${table}\n`;
+			
+			// 批量处理大数据量
+			for (let j = 0; j < rows.length; j += batchSize) {
+				const batch = rows.slice(j, j + batchSize);
+				for (const row of batch) {
+					try {
+						const rowValues = Object.values(row)
+							.map(value => {
+								if (value === null) return 'NULL';
+								if (typeof value === 'boolean') return value ? 1 : 0;
+								if (typeof value === 'number') return value;
+								if (value instanceof Date) {
+									return "'" + value.toISOString().slice(0, 19).replace('T', ' ') + "'";
+								}
+								// 处理字符串，转义特殊字符
+								if (typeof value === 'string') {
+									return "'" + value
+										.replace(/[\\']/g, '\\$&')
+										.replace(/\n/g, '\\n')
+										.replace(/\r/g, '\\r')
+										.replace(/\t/g, '\\t')
+										.replace(/\u0000/g, '\\0') + "'";
+								}
+								return "'" + String(value || '') + "'";
+							}).join(',');
+						sql += `INSERT INTO \`${table}\` VALUES (${rowValues});\n`;
+					} catch (rowErr) {
+						$.log.error(`[DB] [getTableData] 处理行数据失败: ${rowErr.message}`);
+					}
+				}
 			}
-			sql += " DEFAULT " + fd.dflt_value;
+			sql += '\n';
 		}
-		sql += " " + fd.auto;
-	} else if (fd.dflt_value) {
-		if (fd.dflt_value === '0000-00-00 00:00:00') {
-			fd.dflt_value = '1970-01-01 00:00:00';
-		}
-		sql += " DEFAULT " + fd.dflt_value;
+		return sql;
+	} catch (err) {
+		$.log.error(`[DB] [getTableData] 获取表数据失败: ${err.message}`);
+		return '';
 	}
-	if (fd.note) {
-		sql += " COMMENT '" + fd.note + "'";
+};
+
+/**
+ * @description 清空表数据
+ * @param {String} table 表名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
+ */
+DB.prototype.emptyTable = function(table, timeout = 15000) {
+	if (!table || typeof table !== 'string') {
+		$.log.error("[DB] [emptyTable] 表名无效");
+		return 0;
+	}
+	return this.exec("TRUNCATE TABLE \`" + table + "\`;", timeout);
+};
+
+/**
+ * @description 检查表是否存在
+ * @param {String} table 表名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Boolean} 是否存在
+ */
+DB.prototype.hasTable = async function(table, timeout = 15000) {
+	if (!table || typeof table !== 'string') {
+		return false;
 	}
-	if (fd.pk) {
-		sql += " PRIMARY KEY";
+	try {
+		const list = await this.run("SHOW TABLES LIKE '" + table + "'", [], timeout);
+		return list && list.length > 0;
+	} catch (err) {
+		$.log.error(`[DB] [hasTable] 检查表是否存在失败: ${err.message}`);
+		return false;
 	}
-	return sql;
 };
 
 /**
- * @description 修改字段名称
+ * @description 检查字段是否存在
+ * @param {String} table 表名
  * @param {String} field 字段名
- * @param {String} name 变更后名称
- * @return {Promise|Number} 修改成功返回1，失败返回0
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Boolean} 是否存在
  */
-DB.prototype.field_name = async function(field, name) {
-	var fields = await this.fields(this.table, field);
-	if (!fields) {
-		return -1;
+DB.prototype.hasField = async function(table, field, timeout = 15000) {
+	if (!table || !field) {
+		return false;
 	}
-	if (fields.length === 0) {
+	try {
+		const list = await this.fields(table, field, timeout);
+		return list && list.length > 0;
+	} catch (err) {
+		$.log.error(`[DB] [hasField] 检查字段是否存在失败: ${err.message}`);
+		return false;
+	}
+};
+
+/**
+ * @description 备份表
+ * @param {String} table 表名
+ * @param {String} backup 备份表名
+ * @param {Number} timeout 超时时间(毫秒)
+ * @return {Promise|Number} 操作结果
+ */
+DB.prototype.backupTable = function(table, backup, timeout = 60000) {
+	if (!table || !backup) {
+		$.log.error("[DB] [backupTable] 参数不完整");
 		return 0;
 	}
-	var sql_sub = this.field_sql(fields[0]);
-	sql_sub = sql_sub.replace("`" + field + "`", "`" + name + "`").replace(' PRIMARY KEY', '');
-	var sql = "alter table `{0}` change `{1}` " + sql_sub;
-	sql = sql.replace('{0}', this.table).replace('{1}', field);
-	return await this.exec(sql);
+	return this.exec("CREATE TABLE \`" + backup + "\` LIKE \`" + table + "\`; INSERT INTO \`" + backup + "\` SELECT * FROM \`" + table + "\`;", timeout);
 };
 
-exports.DB = DB;
+module.exports = {
+	DB
+};