20 KiB
20 KiB
开发中使用的存储过程
统计一下，数据库中哪些表中有site_id字段或者store_id字段，而且表中是有数据的
-- 检查符合条件的表
DELIMITER $$
CREATE PROCEDURE CheckTablesWithHasData()
BEGIN
DECLARE done INT DEFAULT FALSE;
DECLARE tbl_name VARCHAR(255);
DECLARE cur CURSOR FOR
SELECT DISTINCT TABLE_NAME
FROM information_schema.COLUMNS
WHERE COLUMN_NAME IN ('site_id', 'store_id')
AND TABLE_SCHEMA = DATABASE(); -- 使用当前数据库
DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = TRUE;
CREATE TEMPORARY TABLE tmp_results (table_name VARCHAR(255));
OPEN cur;
read_loop: LOOP
FETCH cur INTO tbl_name;
IF done THEN
LEAVE read_loop;
END IF;
SET @sql = CONCAT('INSERT INTO tmp_results SELECT ''', tbl_name, ''' FROM ', tbl_name, ' HAVING COUNT(*) > 0');
PREPARE stmt FROM @sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;
END LOOP;
CLOSE cur;
SELECT table_name FROM tmp_results;
DROP TEMPORARY TABLE tmp_results;
END$$
DELIMITER ;
-- 执行
CALL CheckTablesWithHasData();
--- 结果
+-----------------+
| table_name |
+-----------------+
| t_order_info |
| t_order_item |
| t_order_payment |
| t_order_refund |
+-----------------+
统计一下，数据库中哪些表中是有数据的
-- 检查符合条件的表
DELIMITER $$
CREATE PROCEDURE CheckTablesIsNotEmpty()
BEGIN
DECLARE done INT DEFAULT FALSE;
DECLARE tbl_name VARCHAR(255);
DECLARE cur CURSOR FOR
SELECT DISTINCT TABLE_NAME
FROM information_schema.COLUMNS
WHERE TABLE_SCHEMA = DATABASE(); -- 使用当前数据库
DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = TRUE;
CREATE TEMPORARY TABLE tmp_results (table_name VARCHAR(255));
OPEN cur;
read_loop: LOOP
FETCH cur INTO tbl_name;
IF done THEN
LEAVE read_loop;
END IF;
SET @sql = CONCAT('INSERT INTO tmp_results SELECT ''', tbl_name, ''' FROM ', tbl_name, ' HAVING COUNT(*) > 0');
PREPARE stmt FROM @sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;
END LOOP;
CLOSE cur;
SELECT table_name FROM tmp_results;
DROP TEMPORARY TABLE tmp_results;
END$$
DELIMITER ;
-- 执行
CALL CheckTablesIsNotEmpty();
对于空表或者无数据表，重置 AUTO_INCREMENT
-- 定义存储过程
DELIMITER $$
CREATE PROCEDURE ResetAutoIncrementForEmptyTables()
BEGIN
DECLARE done INT DEFAULT FALSE;
DECLARE tbl_name VARCHAR(255);
DECLARE cnt BIGINT;
-- 声明游标：获取所有表名
DECLARE cur CURSOR FOR
SELECT DISTINCT c.TABLE_NAME
FROM information_schema.COLUMNS c
INNER JOIN information_schema.TABLES t
ON c.TABLE_SCHEMA = t.TABLE_SCHEMA
AND c.TABLE_NAME = t.TABLE_NAME
WHERE
c.TABLE_SCHEMA = DATABASE()
AND t.TABLE_TYPE = 'BASE TABLE'; -- 排除视图
DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = TRUE;
OPEN cur;
read_loop: LOOP
FETCH cur INTO tbl_name;
IF done THEN
LEAVE read_loop;
END IF;
-- 动态获取行数
SET @sql = CONCAT('SELECT COUNT(*) INTO @cnt FROM `', tbl_name, '`');
PREPARE stmt FROM @sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;
-- 如果为空，重置 AUTO_INCREMENT
IF @cnt = 0 THEN
SET @reset_sql = CONCAT('ALTER TABLE `', tbl_name, '` AUTO_INCREMENT = 1');
PREPARE reset_stmt FROM @reset_sql;
EXECUTE reset_stmt;
DEALLOCATE PREPARE reset_stmt;
SELECT CONCAT('Reset AUTO_INCREMENT for table: ', tbl_name) AS message;
END IF;
END LOOP;
CLOSE cur;
END$$
DELIMITER ;
-- 执行
CALL ResetAutoIncrementForEmptyTables();
数据重置到初始化状态
1. 删除数据，保留表结构，当表中有site_id字段时，只删除site_id不为0的数据
-- 定义存储过程
-- 增强版数据库重置脚本
-- 支持 MySQL/PostgreSQL/SQL Server
DELIMITER $$
DROP PROCEDURE IF EXISTS reset_tables_has_site_id_where;
CREATE PROCEDURE reset_tables_has_site_id_where(
IN p_preserve_site_zero BOOLEAN, -- 是否保留 site_id = 0 的数据
IN p_dry_run BOOLEAN, -- 是否试运行（不实际执行）
IN p_exclude_tables TEXT -- 排除的表列表，逗号分隔
)
BEGIN
DECLARE v_done INT DEFAULT FALSE;
DECLARE v_table_name VARCHAR(255);
DECLARE v_has_site_id INT DEFAULT 0;
DECLARE v_table_count INT DEFAULT 0;
DECLARE v_processed_count INT DEFAULT 0;
DECLARE v_skipped_count INT DEFAULT 0;
DECLARE v_error_count INT DEFAULT 0;
DECLARE v_no_site_id_tables TEXT DEFAULT '';
DECLARE v_excluded_tables TEXT DEFAULT '';
-- 游标声明
DECLARE table_cursor CURSOR FOR
SELECT TABLE_NAME
FROM INFORMATION_SCHEMA.TABLES
WHERE TABLE_SCHEMA = DATABASE()
AND TABLE_TYPE = 'BASE TABLE'
AND TABLE_NAME NOT IN (
'sys_operation_log', 'sys_login_log', 'system_parameters' -- 系统表默认排除
)
ORDER BY TABLE_NAME;
DECLARE CONTINUE HANDLER FOR NOT FOUND SET v_done = TRUE;
-- 创建详细的日志表
DROP TEMPORARY TABLE IF EXISTS reset_operation_log;
CREATE TEMPORARY TABLE reset_operation_log (
id INT AUTO_INCREMENT PRIMARY KEY,
log_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
table_name VARCHAR(255) NOT NULL,
action_type ENUM('CLEARED', 'SKIPPED', 'ERROR', 'EXCLUDED') NOT NULL,
records_affected INT DEFAULT 0,
sql_statement TEXT,
error_message TEXT,
execution_time_ms INT DEFAULT 0
);
-- 开始事务（确保原子性）
START TRANSACTION;
OPEN table_cursor;
process_tables: LOOP
FETCH table_cursor INTO v_table_name;
IF v_done THEN
LEAVE process_tables;
END IF;
SET v_table_count = v_table_count + 1;
-- 检查是否在排除列表中
IF FIND_IN_SET(v_table_name, p_exclude_tables) > 0 OR
FIND_IN_SET(v_table_name, v_excluded_tables) > 0 THEN
INSERT INTO reset_operation_log (table_name, action_type, records_affected, sql_statement)
VALUES (v_table_name, 'EXCLUDED', 0, '表在排除列表中，跳过处理');
SET v_skipped_count = v_skipped_count + 1;
ITERATE process_tables;
END IF;
-- 检查表是否有 site_id 字段
SELECT COUNT(*) INTO v_has_site_id
FROM INFORMATION_SCHEMA.COLUMNS
WHERE TABLE_SCHEMA = DATABASE()
AND TABLE_NAME = v_table_name
AND COLUMN_NAME = 'site_id';
SET @start_time = UNIX_TIMESTAMP(NOW(3)) * 1000;
IF v_has_site_id > 0 THEN
-- 构建动态SQL
IF p_preserve_site_zero THEN
SET @sql = CONCAT('DELETE FROM `', v_table_name, '` WHERE site_id != 0');
ELSE
SET @sql = CONCAT('TRUNCATE TABLE `', v_table_name, '`');
END IF;
BEGIN
DECLARE EXIT HANDLER FOR SQLEXCEPTION
BEGIN
GET DIAGNOSTICS CONDITION 1 @sqlstate = RETURNED_SQLSTATE, @errmsg = MESSAGE_TEXT;
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement, error_message)
VALUES (v_table_name, 'ERROR', 0, @sql, CONCAT(@sqlstate, ' - ', @errmsg));
SET v_error_count = v_error_count + 1;
END;
IF p_dry_run THEN
-- 试运行模式，只记录不执行
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement)
VALUES (v_table_name, 'SKIPPED', 0, CONCAT('试运行: ', @sql));
SET v_skipped_count = v_skipped_count + 1;
ELSE
-- 实际执行
PREPARE stmt FROM @sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;
SET @row_count = ROW_COUNT();
SET @end_time = UNIX_TIMESTAMP(NOW(3)) * 1000;
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement, execution_time_ms)
VALUES (v_table_name, 'CLEARED', @row_count, @sql, (@end_time - @start_time));
SET v_processed_count = v_processed_count + 1;
END IF;
END;
ELSE
-- 表没有 site_id 字段
SET v_no_site_id_tables = CONCAT_WS(', ', v_no_site_id_tables, v_table_name);
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement)
VALUES (v_table_name, 'SKIPPED', 0, '表没有 site_id 字段，跳过处理');
SET v_skipped_count = v_skipped_count + 1;
END IF;
SET v_has_site_id = 0;
END LOOP;
CLOSE table_cursor;
IF p_dry_run THEN
ROLLBACK; -- 试运行模式回滚
SELECT '=== 试运行模式 - 未实际执行任何操作 ===' as notice;
ELSE
COMMIT; -- 提交事务
SELECT '=== 数据库重置完成 ===' as summary;
END IF;
-- 生成总结报告
SELECT
CONCAT('数据库: ', DATABASE()) as database_info,
CONCAT('开始时间: ', DATE_FORMAT(MIN(log_time), '%Y-%m-%d %H:%i:%s')) as start_time,
CONCAT('结束时间: ', DATE_FORMAT(MAX(log_time), '%Y-%m-%d %H:%i:%s')) as end_time,
CONCAT('总表数: ', v_table_count) as total_tables,
CONCAT('已处理表: ', v_processed_count) as processed_tables,
CONCAT('跳过表: ', v_skipped_count) as skipped_tables,
CONCAT('错误数: ', v_error_count) as error_count,
CONCAT('总耗时: ', SUM(execution_time_ms), 'ms') as total_duration
FROM reset_operation_log;
-- 显示没有 site_id 字段的表
IF v_no_site_id_tables != '' THEN
SELECT
'以下表没有 site_id 字段，已被跳过:' as warning_title,
v_no_site_id_tables as skipped_tables;
END IF;
-- 显示错误详情
IF v_error_count > 0 THEN
SELECT '=== 错误详情 ===' as error_title;
SELECT table_name, error_message, sql_statement
FROM reset_operation_log
WHERE action_type = 'ERROR'
ORDER BY log_time;
END IF;
-- 显示处理详情
SELECT '=== 处理详情 ===' as details_title;
SELECT
table_name as '表名',
CASE action_type
WHEN 'CLEARED' THEN '已清空'
WHEN 'SKIPPED' THEN '已跳过'
WHEN 'ERROR' THEN '错误'
WHEN 'EXCLUDED' THEN '已排除'
END as '操作状态',
records_affected as '影响行数',
execution_time_ms as '耗时(ms)',
CASE
WHEN error_message IS NOT NULL THEN error_message
ELSE LEFT(sql_statement, 100)
END as '详情'
FROM reset_operation_log
ORDER BY action_type, table_name;
-- 性能统计
SELECT '=== 性能统计 ===' as performance_title;
SELECT
action_type as '操作类型',
COUNT(*) as '表数量',
SUM(records_affected) as '总影响行数',
AVG(execution_time_ms) as '平均耗时(ms)',
SUM(execution_time_ms) as '总耗时(ms)'
FROM reset_operation_log
GROUP BY action_type;
-- 清理
DROP TEMPORARY TABLE IF EXISTS reset_operation_log;
END
$$
DELIMITER ;
-- 使用示例
-- 试运行（不实际执行）
CALL reset_tables_has_site_id_where(TRUE, TRUE, 'user,config');
-- 实际执行（保留site_id=0的数据，排除user和config表）
CALL reset_tables_has_site_id_where(TRUE, FALSE, '');
-- 完全清空所有表（不保留任何数据）
CALL reset_tables_has_site_id_where(FALSE, FALSE, '');
删除数据，保留表结构，当表中有store_id字段时，只删除store_id不为0的数据
DELIMITER $$
DROP PROCEDURE IF EXISTS reset_tables_has_store_id_where;
CREATE PROCEDURE reset_tables_has_store_id_where(
IN p_preserve_site_zero BOOLEAN, -- 是否保留 store_id = 0 的数据
IN p_dry_run BOOLEAN, -- 是否试运行（不实际执行）
IN p_exclude_tables TEXT -- 排除的表列表，逗号分隔
)
BEGIN
DECLARE v_done INT DEFAULT FALSE;
DECLARE v_table_name VARCHAR(255);
DECLARE v_has_store_id INT DEFAULT 0;
DECLARE v_table_count INT DEFAULT 0;
DECLARE v_processed_count INT DEFAULT 0;
DECLARE v_skipped_count INT DEFAULT 0;
DECLARE v_error_count INT DEFAULT 0;
DECLARE v_no_store_id_tables TEXT DEFAULT '';
DECLARE v_excluded_tables TEXT DEFAULT '';
-- 游标声明
DECLARE table_cursor CURSOR FOR
SELECT TABLE_NAME
FROM INFORMATION_SCHEMA.TABLES
WHERE TABLE_SCHEMA = DATABASE()
AND TABLE_TYPE = 'BASE TABLE'
AND TABLE_NAME NOT IN (
'sys_operation_log', 'sys_login_log', 'system_parameters' -- 系统表默认排除
)
ORDER BY TABLE_NAME;
DECLARE CONTINUE HANDLER FOR NOT FOUND SET v_done = TRUE;
-- 创建详细的日志表
DROP TEMPORARY TABLE IF EXISTS reset_operation_log;
CREATE TEMPORARY TABLE reset_operation_log (
id INT AUTO_INCREMENT PRIMARY KEY,
log_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
table_name VARCHAR(255) NOT NULL,
action_type ENUM('CLEARED', 'SKIPPED', 'ERROR', 'EXCLUDED') NOT NULL,
records_affected INT DEFAULT 0,
sql_statement TEXT,
error_message TEXT,
execution_time_ms INT DEFAULT 0
);
-- 开始事务（确保原子性）
START TRANSACTION;
OPEN table_cursor;
process_tables: LOOP
FETCH table_cursor INTO v_table_name;
IF v_done THEN
LEAVE process_tables;
END IF;
SET v_table_count = v_table_count + 1;
-- 检查是否在排除列表中
IF FIND_IN_SET(v_table_name, p_exclude_tables) > 0 OR
FIND_IN_SET(v_table_name, v_excluded_tables) > 0 THEN
INSERT INTO reset_operation_log (table_name, action_type, records_affected, sql_statement)
VALUES (v_table_name, 'EXCLUDED', 0, '表在排除列表中，跳过处理');
SET v_skipped_count = v_skipped_count + 1;
ITERATE process_tables;
END IF;
-- 检查表是否有 store_id 字段
SELECT COUNT(*) INTO v_has_store_id
FROM INFORMATION_SCHEMA.COLUMNS
WHERE TABLE_SCHEMA = DATABASE()
AND TABLE_NAME = v_table_name
AND COLUMN_NAME = 'store_id';
SET @start_time = UNIX_TIMESTAMP(NOW(3)) * 1000;
IF v_has_store_id > 0 THEN
-- 构建动态SQL
IF p_preserve_site_zero THEN
SET @sql = CONCAT('DELETE FROM `', v_table_name, '` WHERE store_id != 0');
ELSE
SET @sql = CONCAT('TRUNCATE TABLE `', v_table_name, '`');
END IF;
BEGIN
DECLARE EXIT HANDLER FOR SQLEXCEPTION
BEGIN
GET DIAGNOSTICS CONDITION 1 @sqlstate = RETURNED_SQLSTATE, @errmsg = MESSAGE_TEXT;
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement, error_message)
VALUES (v_table_name, 'ERROR', 0, @sql, CONCAT(@sqlstate, ' - ', @errmsg));
SET v_error_count = v_error_count + 1;
END;
IF p_dry_run THEN
-- 试运行模式，只记录不执行
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement)
VALUES (v_table_name, 'SKIPPED', 0, CONCAT('试运行: ', @sql));
SET v_skipped_count = v_skipped_count + 1;
ELSE
-- 实际执行
PREPARE stmt FROM @sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;
SET @row_count = ROW_COUNT();
SET @end_time = UNIX_TIMESTAMP(NOW(3)) * 1000;
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement, execution_time_ms)
VALUES (v_table_name, 'CLEARED', @row_count, @sql, (@end_time - @start_time));
SET v_processed_count = v_processed_count + 1;
END IF;
END;
ELSE
-- 表没有 store_id 字段
SET v_no_store_id_tables = CONCAT_WS(', ', v_no_store_id_tables, v_table_name);
INSERT INTO reset_operation_log
(table_name, action_type, records_affected, sql_statement)
VALUES (v_table_name, 'SKIPPED', 0, '表没有 store_id 字段，跳过处理');
SET v_skipped_count = v_skipped_count + 1;
END IF;
SET v_has_store_id = 0;
END LOOP;
CLOSE table_cursor;
IF p_dry_run THEN
ROLLBACK; -- 试运行模式回滚
SELECT '=== 试运行模式 - 未实际执行任何操作 ===' as notice;
ELSE
COMMIT; -- 提交事务
SELECT '=== 数据库重置完成 ===' as summary;
END IF;
-- 生成总结报告
SELECT
CONCAT('数据库: ', DATABASE()) as database_info,
CONCAT('开始时间: ', DATE_FORMAT(MIN(log_time), '%Y-%m-%d %H:%i:%s')) as start_time,
CONCAT('结束时间: ', DATE_FORMAT(MAX(log_time), '%Y-%m-%d %H:%i:%s')) as end_time,
CONCAT('总表数: ', v_table_count) as total_tables,
CONCAT('已处理表: ', v_processed_count) as processed_tables,
CONCAT('跳过表: ', v_skipped_count) as skipped_tables,
CONCAT('错误数: ', v_error_count) as error_count,
CONCAT('总耗时: ', SUM(execution_time_ms), 'ms') as total_duration
FROM reset_operation_log;
-- 显示没有 store_id 字段的表
IF v_no_store_id_tables != '' THEN
SELECT
'以下表没有 store_id 字段，已被跳过:' as warning_title,
v_no_store_id_tables as skipped_tables;
END IF;
-- 显示错误详情
IF v_error_count > 0 THEN
SELECT '=== 错误详情 ===' as error_title;
SELECT table_name, error_message, sql_statement
FROM reset_operation_log
WHERE action_type = 'ERROR'
ORDER BY log_time;
END IF;
-- 显示处理详情
SELECT '=== 处理详情 ===' as details_title;
SELECT
table_name as '表名',
CASE action_type
WHEN 'CLEARED' THEN '已清空'
WHEN 'SKIPPED' THEN '已跳过'
WHEN 'ERROR' THEN '错误'
WHEN 'EXCLUDED' THEN '已排除'
END as '操作状态',
records_affected as '影响行数',
execution_time_ms as '耗时(ms)',
CASE
WHEN error_message IS NOT NULL THEN error_message
ELSE LEFT(sql_statement, 100)
END as '详情'
FROM reset_operation_log
ORDER BY action_type, table_name;
-- 性能统计
SELECT '=== 性能统计 ===' as performance_title;
SELECT
action_type as '操作类型',
COUNT(*) as '表数量',
SUM(records_affected) as '总影响行数',
AVG(execution_time_ms) as '平均耗时(ms)',
SUM(execution_time_ms) as '总耗时(ms)'
FROM reset_operation_log
GROUP BY action_type;
-- 清理
DROP TEMPORARY TABLE IF EXISTS reset_operation_log;
END
$$
DELIMITER ;
-- 实际执行（保留site_id=0的数据，排除user和config表）
CALL reset_tables_has_store_id_where(TRUE, FALSE, '');