mysql 存储过程中使用多游标

本文转自:http://www.netingcn.com/mysql-procedure-muti-cursor.html

mysql的存储过程可以很方便使用游标来实现一些功能，存储过程的写法大致如下：

先创建一张表，插入一些测试数据：

DROP TABLE IF EXISTS netingcn_proc_test;

CREATE TABLE `netingcn_proc_test` (
  `id` INTEGER(11) NOT NULL AUTO_INCREMENT,
  `name` VARCHAR(20),
  `password` VARCHAR(20),
  PRIMARY KEY (`id`)
)ENGINE=InnoDB;

insert into netingcn_proc_test(name, password) values
('procedure1', 'pass1'),
('procedure2', 'pass2'),
('procedure3', 'pass3'),
('procedure4', 'pass4');

下面就是一个简单存储过程的例子：

drop procedure IF EXISTS test_proc;
delimiter //
create procedure test_proc()
begin
	-- 声明一个标志done， 用来判断游标是否遍历完成
	DECLARE done INT DEFAULT 0;

	-- 声明一个变量，用来存放从游标中提取的数据
	-- 特别注意这里的名字不能与由游标中使用的列明相同，否则得到的数据都是NULL
	DECLARE tname varchar(50) DEFAULT NULL;
	DECLARE tpass varchar(50) DEFAULT NULL;

	-- 声明游标对应的 SQL 语句
	DECLARE cur CURSOR FOR
		select name, password from netingcn_proc_test;

	-- 在游标循环到最后会将 done 设置为 1
	DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;

	-- 执行查询
	open cur;
	-- 遍历游标每一行
	REPEAT
		-- 把一行的信息存放在对应的变量中
		FETCH cur INTO tname, tpass;
		if not done then
			-- 这里就可以使用 tname， tpass 对应的信息了
			select tname, tpass;
		end if;
 	UNTIL done END REPEAT;
	CLOSE cur;
end
//
delimiter ;

-- 执行存储过程
call test_proc();

需要注意的是变量的声明、游标的声明和HANDLER声明的顺序不能搞错，必须是先声明变量，再申明游标，最后声明HANDLER。上述存储过程的例子中只使用了一个游标，那么如果要使用两个或者更多游标怎么办，其实很简单，可以这么说，一个怎么用两个就是怎么用的。例子如下：

drop procedure IF EXISTS test_proc_1;
delimiter //
create procedure test_proc_1()
begin
	DECLARE done INT DEFAULT 0;
	DECLARE tid int(11) DEFAULT 0;
	DECLARE tname varchar(50) DEFAULT NULL;
	DECLARE tpass varchar(50) DEFAULT NULL;

	DECLARE cur_1 CURSOR FOR
		select name, password from netingcn_proc_test;

	DECLARE cur_2 CURSOR FOR
		select id, name from netingcn_proc_test;

	DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;

	open cur_1;
	REPEAT
		FETCH cur_1 INTO tname, tpass;
		if not done then
			select tname, tpass;
		end if;
 	UNTIL done END REPEAT;
	CLOSE cur_1;

	-- 注意这里，一定要重置done的值为 0
	set done = 0;

	open cur_2;
	REPEAT
		FETCH cur_2 INTO tid, tname;
		if not done then
			select tid, tname;
		end if;
 	UNTIL done END REPEAT;
	CLOSE cur_2;
end
//
delimiter ;

call test_proc_1();

上述代码和第一个例子中基本一样，就是多了一个游标声明和遍历游标。这里需要注意的是，在遍历第二个游标前使用了set done = 0，因为当第一个游标遍历玩后其值被handler设置为1了，如果不用set把它设置为 0 ，那么第二个游标就不会遍历了。当然好习惯是在每个打开游标的操作前都用该语句，确保游标能真正遍历。当然还可以使用begin语句块嵌套的方式来处理多个游标,例如：

drop procedure IF EXISTS test_proc_2;
delimiter //
create procedure test_proc_2()
begin
	DECLARE done INT DEFAULT 0;
	DECLARE tname varchar(50) DEFAULT NULL;
	DECLARE tpass varchar(50) DEFAULT NULL;

	DECLARE cur_1 CURSOR FOR
		select name, password from netingcn_proc_test;

	DECLARE cur_2 CURSOR FOR
		select id, name from netingcn_proc_test;

	DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;

	open cur_1;
	REPEAT
		FETCH cur_1 INTO tname, tpass;
		if not done then
			select tname, tpass;
		end if;
 	UNTIL done END REPEAT;
	CLOSE cur_1;

	begin
		DECLARE done INT DEFAULT 0;
		DECLARE tid int(11) DEFAULT 0;
		DECLARE tname varchar(50) DEFAULT NULL;

		DECLARE cur_2 CURSOR FOR
			select id, name from netingcn_proc_test;

		DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;

		open cur_2;
		REPEAT
			FETCH cur_2 INTO tid, tname;
			if not done then
				select tid, tname;
			end if;
	 	UNTIL done END REPEAT;
		CLOSE cur_2;
	end;
end
//
delimiter ;

call test_proc_2();

1. create procedure my_procedure() -- 创建存储过程  
2. begin -- 开始存储过程  
3. declare my_id varchar(32); -- 自定义变量1  
4. declare my_name varchar(50); -- 自定义变量2  
5. DECLARE done INT DEFAULT FALSE; -- 自定义控制游标循环变量,默认false  
6.   
7. DECLARE My_Cursor CURSOR FOR ( SELECT id, name FROM t_people ); -- 定义游标并输入结果集  
8. DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = TRUE; -- 绑定控制变量到游标,游标循环结束自动转true  
9.   
10. OPEN My_Cursor; -- 打开游标  
11.   myLoop: LOOP -- 开始循环体,myLoop为自定义循环名,结束循环时用到  
12.     FETCH My_Cursor into my_id, my_name; -- 将游标当前读取行的数据顺序赋予自定义变量12  
13.     IF done THEN -- 判断是否继续循环  
14.       LEAVE myLoop; -- 结束循环  
15.     END IF;  
16.     -- 自己要做的事情,在 sql 中直接使用自定义变量即可  
17.     UPDATE t_user SET c_name = my_name WHERE id = my_id and rtrim(ltrim(c_name)) = ''; -- 左右去空格  
18.   
19.     COMMIT; -- 提交事务  
20.   END LOOP myLoop; -- 结束自定义循环体  
21.   CLOSE My_Cursor; -- 关闭游标  
22. END; -- 结束存储过程