MySql存储过程,循环遍历查询结果集,并执行相关逻辑

场景

需要从某个表中读取数据集合，然后循环取值，写sql执行逻辑，实现插入/更新操作。

创建存储过程

-- 创建存储过程之前需判断该存储过程是否已存在，若存在则删除
DROP PROCEDURE IF EXISTS batch_data;

-- 创建存储过程
CREATE PROCEDURE batch_data()
BEGIN
    -- 定义变量
    DECLARE i INT DEFAULT 0;
	DECLARE s int DEFAULT 0;
	DECLARE n VARCHAR(32);
		
		-- 定义游标，并将sql结果集赋值给游标
		DECLARE my_cursor CURSOR FOR select name from test.tree_table;
		
		-- 声明当游标遍历完后将标志变量置成1
		DECLARE CONTINUE HANDLER FOR NOT FOUND SET s=1;
		
		-- 打开游标
		OPEN my_cursor;
		
		  -- 将游标中的值赋值给变量，注意：变量名不要和返回列名同名，变量顺序要和sql结果顺序一致
			FETCH my_cursor into n;
			
			-- 当s != 1,一直循环
			while s<>1 do
			
			  -- 执行业务逻辑
				set i = i+1;
				INSERT INTO test.user VALUES(i,n,NOW());
				
				-- 读取下一条数据，读取完成置变量s=1
				FETCH my_cursor into n;
			end while;
		-- 关闭游标
		close my_cursor;
end

执行存储过程

-- 执行存储过程
CALL batch_data()

结果