MYSQL 批量生成触发器
祖仙教小凡仙 海鲨数据库架构师
上周实现银行一个需求,需求是这样的,有些项目的核心表只能通过JAVA程序进行修改,不允许其他程序修改。这个需求基于ORACLE的在去年也实现了。
大概是通过创建两个表,其中一个用来保存需要保护的核心表的名字,另外个叫黑名单的表保存禁止机器名。
然后写个过程 根据两个表生产触发器,有多少核心表,就生成多少个触发器。
上周基于ORACLE设计想法,在MYSQL上重现一次就能完成需求。
当然MYSQL的路很曲折
第一创建表结构
/*被保护的表列表*/
CREATE TABLE `PARAM_TABLE` (
`ID` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
`TABLENAME` varchar(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_german2_ci NOT NULL COMMENT '''注意表名的大小写''',
`MODIFY_TIME` datetime DEFAULT NULL,
PRIMARY KEY (`ID`)
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_german2_ci
/*黑名单表 注意用户名应该输入 ‘布拉格@192.168.2.13’ 形式*/
CREATE TABLE `black_user` (
`ID` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
`USER_NAME` varchar(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_german2_ci NOT NULL COMMENT '''username@ip''',
`MODIFY_TIME` datetime DEFAULT NULL,
PRIMARY KEY (`ID`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_german2_ci
第二 创建 存储过程
MYSQL的触发器好像不支持多个动作,本来想INSERT UPDATE DELETE合在一起,结果不行。只好分成3个来搞!
/*触发器核心程序*/
DELIMITER $$
USE `bookstore`$$
DROP PROCEDURE IF EXISTS `p_procedurecode`$$
CREATE DEFINER=`root`@`%` PROCEDURE `p_procedurecode`(IN v_schema VARCHAR(50))
BEGIN
DECLARE v_command_num INTEGER DEFAULT 1; ##触发器类型数量
DECLARE no_more_PARAM_TABLES INTEGER DEFAULT 0; ##定义游标退出变量
DECLARE v_table_name VARCHAR(100); ##定义存放表名的变量
DECLARE c_result CURSOR FOR SELECT DISTINCT TABLENAME FROM PARAM_TABLE; ##定义游标的SQL
DECLARE CONTINUE HANDLER FOR NOT FOUND SET no_more_PARAM_TABLES=1; ##关联游标NOT FOUD变量
##如果没有找到数据则设置该变量为1
OPEN c_result;
WHILE no_more_PARAM_TABLES <> 1 DO ##第一层 外层游标循环 这里要判断退出标志
FETCH c_result INTO v_table_name; ##提取游标结果
IF no_more_PARAM_TABLES <> 1 THEN ##第二次判断 游标是否可退出
SET v_command_num = 1; ##如果有数据则进入内循环
WHILE v_command_num < 4 DO ##内循环退出标志
SET @sql='DELIMITER $$ '; ##很烦这个 为啥不像ORACLE学习呢?
SET @sql=CONCAT(@sql,CHAR(10)); ##CONCAT是字符串拼接 不如ORACLE | 符合方便
SET @sql=CONCAT(@sql,'USE `');
SET @sql=CONCAT(@sql,v_schema);
SET @sql=CONCAT(@sql,'`$$');
SET @sql=CONCAT(@sql,CHAR(10)); ##CHAR 10是换行符号
IF v_command_num = 1 THEN ## 判断触发器类型生成不同的触发器名字
SET @sql=CONCAT(@sql,'DROP TRIGGER `tr_protect_u_');
END IF;
IF v_command_num = 2 THEN
SET @sql=CONCAT(@sql,'DROP TRIGGER `tr_protect_i_');
END IF;
IF v_command_num = 3 THEN
SET @sql=CONCAT(@sql,'DROP TRIGGER `tr_protect_d_');
END IF;
SET @sql=CONCAT(@sql,v_table_name);
SET @sql=CONCAT(@sql,'`$$');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,'CREATE TRIGGER ');
SET @sql=CONCAT(@sql, v_schema);
IF v_command_num = 1 THEN
SET @sql=CONCAT(@sql,'.`tr_protect_u_');
END IF;
IF v_command_num = 2 THEN
SET @sql=CONCAT(@sql,'.`tr_protect_i_');
END IF;
IF v_command_num = 3 THEN
SET @sql=CONCAT(@sql,'.`tr_protect_d_');
END IF;
SET @sql=CONCAT(@sql, v_table_name);
IF v_command_num = 1 THEN
SET @sql=CONCAT(@sql,'` BEFORE UPDATE ON ');
END IF;
IF v_command_num = 2 THEN
SET @sql=CONCAT(@sql,'` BEFORE INSERT ON ');
END IF;
IF v_command_num = 3 THEN
SET @sql=CONCAT(@sql,'` BEFORE DELETE ON ');
END IF;
##下面是拼接触发器核心内容,大意是判断当前用户是否在黑名单中
SET @sql=CONCAT(@sql, v_schema);
SET @sql=CONCAT(@sql,'.');
SET @sql=CONCAT(@sql,v_table_name);
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' FOR EACH ROW BEGIN');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' set @count=(SELECT COUNT(*) FROM ');
SET @sql=CONCAT(@sql, v_schema);
SET @sql=CONCAT(@sql,'.black_user WHERE user_name = USER());');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' IF @count>0 THEN ');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' SIGNAL SQLSTATE ''48000'' SET MESSAGE_TEXT = ''您没有权限操作参数表!'';' );
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' END IF;');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' END; ');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,' $$');
SET @sql=CONCAT(@sql,CHAR(10));
SET @sql=CONCAT(@sql,'DELIMITER ;');
SELECT @sql; ##由于MYSQL不支持动态执行CREATE TRRIGE的DDL语句 只能先打印出来
SET v_command_num = v_command_num +1;
END WHILE;
END IF;
END WHILE;
CLOSE C_RESULT;
END$$
DELIMITER ;
写这个存储过程时 感觉比ORACLE 繁琐。
1 不同动态执行SQL FOR DDL
2 DELIMITER 好烦
3 没有FOR CUR_XX () LOOP 语句
4 UNITE REPLACE 循环会多循环一次,不得不采用两个WHILE来循环游标
5 @变量 和自定义变量 感觉有很大的区别,实际没啥区别。
总得来说MSQL 开发对开发人员不太友好。也就是用户友好性不够好
第三执行 存储过程
不能动态执行,只好先把执行结果,就是打印DDL语句到LINUX 文件中
这里要用到MYSQL客户端 -E 参数 来执行SQL。然后执行导出来的DDL.SQL
#/bin/bash
#write by ZengFanKun doctor
#on 2021-02-28
USER=root
PWD=123456
#HOST=192.168.2.13
HOST=localhost
PROT=3306
DATABASE=bookstore
MYSQL_PATH=/app/mysql8/mysql8_soft
cd ${MYSQL_PATH}
./bin/mysql -uroot -p123456 -hlocalhost -P3306 -D${DATABASE} --vertical --skip-column-names -e"CALL p_procedurecode('${DATABASE}')" |grep -v row > create_trigger_ddl.sql
./bin/mysql -uroot -p123456 -hlocalhost -P3306 -D${DATABASE}
--force < create_trigger_ddl.sql
--vertical ##垂直显示结果 类似 \G
--skip-column-names ##不显示 字段名称 类似SQLPLUS 不显示标题
-e"CALL p_procedurecode('${DATABASE}')" ##执行存储过程
grep -v row > create_trigger_ddl.sql ##过滤掉 -----1row----
忽悠该类型的错误和警告
[root@centos7 mysql8]# sh make_tr_ddl.sh
mysql: [Warning] Using a password on the command line interface can be insecure.
mysql: [Warning] Using a password on the command line interface can be insecure.
ERROR 1360 (HY000) at line 3: Trigger does not exist
ERROR 1360 (HY000) at line 15: Trigger does not exist
ERROR 1360 (HY000) at line 27: Trigger does not exist
ERROR 1360 (HY000) at line 39: Trigger does not exist
ERROR 1360 (HY000) at line 51: Trigger does not exist
ERROR 1360 (HY000) at line 63: Trigger does not exist
本来是使用SHELL变量来登录的,不管如何折腾发现MYSQL报错
说 'ROOT'@'LOCALHOST' 拒绝登陆。只好让开发人员自己去填实参
执行脚本需要手工输入数据库的执行用户名,密码,端口,路径,SCHEMA
我用SQLyog 工具无法调试MYSQL的存储过程!