MySQL使用存储过程批量更新数据库所有表某个字段值

言曌 2019年09月09日13:33:54 6 5238 views

博主分享免费Java教学视频，B站账号：Java刘哥

最近响应群里朋友完整开源之前那个博客系统，准备重构一番项目的代码，对数据库中的表决定都添加 create_by、update_by、create_time、update_time、del_flag 等字段。当时添加表的时候没有设置默认值，现在要对二三十张表某个字段，如对 del_flag 设置默认值为0，怎么做呢？一张表一张表地设置比较蠢，如何实现批量操作呢？比如查出所有的表名，然后来一个循环操作。是的，没错，可以使用存储过程。下面是对 sens_blog 这个库的所有的表中的 del_flag 设置默认值的示例

-- 如果存储过程存在就删除
DROP PROCEDURE IF EXISTS updateColumn;
CREATE PROCEDURE updateColumn()
BEGIN
-- 定义循环条件
DECLARE flag INT DEFAULT 0;
-- 保存表名
DECLARE tname VARCHAR(50);
-- 查询数据库sens_blog中含有del_flag列的表,如果区分大小写使用binary COLUMN_NAME = 'del_flag'
DECLARE result CURSOR FOR SELECT TABLE_NAME FROM INFORMATION_SCHEMA.Columns WHERE TABLE_SCHEMA = 'sens_blog' AND COLUMN_NAME = 'del_flag';
-- 退出循环
DECLARE CONTINUE HANDLER FOR SQLSTATE '02000' SET flag = 1;
-- 打开游标
OPEN result;
WHILE flag <> 1 DO
-- 游标指向下一个位置,可以有多个数据,比如FETCH result INTO tname,ttype,...;
FETCH result INTO tname;
-- 拼接字符串表名sql,根据需要使用CONCAT函数连接
-- 批量设置所有表的为del_flag字段0
-- SET @execSql = CONCAT('UPDATE ', tname, ' SET del_flag = 0');
-- 批量设置所有表的为del_flag字段默认值为0
SET @execSql = CONCAT('ALTER TABLE ', tname, ' ALTER COLUMN del_flag SET DEFAULT 0');
PREPARE stmt FROM @execSql;
EXECUTE stmt;
END WHILE;
END;
-- 调用存储过程更新数据
CALL updateColumn();