欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

使用动态SQL处理table_name作为输入参数的存储过程(MySQL)

程序员文章站 2023-01-22 14:12:22
关于mysql如何创建和使用存储过程,参考笔记《MySQL存储过程和函数创建》以及官网:https://dev.mysql.com/doc/refman/5.7/en/create-procedure.html 本篇主要示例使用了输入参数的存储过程,并解决使用表名作为输入参数的问题,因为之前遇到过需 ......

关于mysql如何创建和使用存储过程,参考笔记《mysql存储过程和函数创建》以及官网:

本篇主要示例使用了输入参数的存储过程,并解决使用表名作为输入参数的问题,因为之前遇到过需要使用表名作为参数的存储过程,很难处理。

问题描述:

假设我们有test1-test12共12个相同结构的车辆里程表,我们想要对这12个表进行去重,那么逻辑上比较简单的办法是写12个存储过程处理或者写一个存储过程每执行一次改一次表名并重新编译,但是这样都太麻烦了。

接下来很容易的就会想到是否可以使用表名作为输入参数,这样每次执行给定表名即可。

因此初始的存储过程代码如下:

delimiter //
drop procedure if exists del_dupilicate;
create definer=`root`@`localhost` procedure `del_dupilicate`(in table_name varchar(64))
begin
declare v_min_id,v_group_count int;
declare v_get_on_time,v_get_off_time datetime;
declare v_car_no varchar(255);
declare done int default false;
declare my_cur cursor for select get_on_time,get_off_time,car_no,min(id),count(1) as count from table_name group by get_on_time,get_off_time,car_no having count>1;
declare continue handler for not found set done = true;
open my_cur;
myloop: loop
fetch my_cur into v_get_on_time,v_get_off_time,v_car_no,v_min_id,v_group_count;
if done then
leave myloop;
end if;
delete from table_name where get_on_time=v_get_on_time and get_off_time=v_get_off_time and car_no=v_car_no and id>v_min_id;
commit;
end loop;
close my_cur;
end;
//
delimiter ;

上述存储过程可以正常编译,但是执行却一定会报table not exist的错误,因为mysql会错误的把输入变量table_name当做真正的数据库表名,这显然是错误的。

那么如何在sql中引用变量呢?一个可行的办法是使用动态sql,把变量拼入sql语句中然后执行动态sql。

所以根据官网()提供的语法,对于上述procedure中的delete语句可以改写成如下格式:

set @del_sql=concat('delete from ',table_name,' where get_on_time=',v_get_on_time,' and get_off_time=',v_get_off_time,' and car_no=',v_car_no,' and id>',v_min_id)
prepare stmt from @del_sql;
execute stmt;
deallocate prepare stmt;
//注意prepare目前只能在存储过程中使用,函数和触发器都不适用。

ps:需要注意的是官网在示例prepare的语法时使用了?作为占位符,但是经试验?不能作为表名的占位符(实际上官网只示例了?可以作为整数字面量的占位符,我猜测凡是数据库对象用?作为占位符都会报错),想要将表名变量整合入sql中只能使用concat函数,concat的函数的输入支持local variables、user defined variables和input variables。

好,delete语句处理完毕,但是对于cursor中的select语句呢?官网明确说明游标中不能使用动态sql,也就是不能使用prepare语句,那只能换一种思路了。

游标的作用是什么呢?是获取一个结果集以便进行遍历,那么可否使用临时表代替游标来存储结果集,这样可以使用动态sql创建临时表(mysql的临时表是session级别的,不同会话可以使用相同名称的临时表,会话释放时临时表自动删除):

set @tmp_table_name=concat(table_name,'_tmp');
set @cur_sql=concat('create temporary table ',@tmp_table_name,' as select get_on_time,get_off_time,car_no,min(id) as min_id,count(1) as count from ',table_name,' group by get_on_time,get_off_time,car_no having count>1');
prepare stmt from @cur_sql;
execute stmt;
deallocate prepare stmt;
然后整个存储过程的逻辑就可以更改了,因为我们把中间结果集存入了临时表,那就无需遍历cursor了,同时连declare的local variables也省了(因为这些本地变量是用于遍历游标时存储列值的),只需要delete ... join即可,因此最终的存储过程修改为:
create definer=`root`@`localhost` procedure `del_dupilicate`(in table_name varchar(64))
begin

set @tmp_table_name=concat(table_name,'_tmp');

set @cur_sql=concat('create temporary table ',@tmp_table_name,' as select get_on_time,get_off_time,car_no,min(id) as min_id,count(1) as count from ',table_name,' group by get_on_time,get_off_time,car_no having count>1');
prepare stmt from @cur_sql;
execute stmt;
deallocate prepare stmt;

set @del_sql=concat('delete a from ',table_name,' a join ',@tmp_table_name,' b on a.get_on_time=b.get_on_time and a.get_off_time=b.get_off_time and a.car_no=b.car_no and a.id != b.min_id');
prepare stmt from @del_sql;
execute stmt;
deallocate prepare stmt;

set @drop_tmp_sql=concat('drop temporary table ',@tmp_table_name);
prepare stmt from @drop_tmp_sql;
execute stmt;
deallocate prepare stmt;

end
 调用:
call del_dupilicate('test1');
 上述存储过程经过了实测,可以正常的删除重复数据。