遇到了一个问题,有一张表,需要增加一列idx
,根据collectionId
分组查询数据,然后为idx
赋值。
分割出来其实就是两个循环。
- 外层循环
SELECT DISTINCT collectionId FROM collection_item
。 - 内层循环
SELECT id FROM collection_item WHERE collectionId = collection_id
。 - 然后更新
UPDATE collection_item SET idx = i WHERE id = item_id
。
存储过程实现
-- 修改分隔符
DELIMITER $$
CREATE PROCEDURE update_collection_item()
begin
-- 定义一会需要用到的各种属性
DECLARE collection_id BIGINT;
DECLARE item_id BIGINT;
DECLARE i INT;
DECLARE done INT DEFAULT TRUE;
-- 定义第一个游标
DECLARE cur1 CURSOR FOR SELECT DISTINCT collectionId FROM collection_item;
-- 定义第二个游标,条件上面定义的collection_id (collection_id 会在循环里被赋值)
DECLARE cur2 CURSOR FOR SELECT id FROM collection_item WHERE collectionId = collection_id ORDER BY createTime;
-- 定义数据结束的条件值
DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = FALSE;
-- 打开第一个游标
OPEN cur1;
-- 循环条件就是 done == true
WHILE done DO
-- 取出来游标里的值赋给 collection_id
-- 如果有表里有多个值 FETCH cur1 INTO (val1,val2);
FETCH cur1 INTO collection_id;
-- 是否还有数据的标识
IF done = TRUE THEN
SET i = 1;
-- 打开第二个游标
OPEN cur2;
-- 循环条件就是 done == true
WHILE done DO
-- 取出来游标里的值赋给 item_id
FETCH cur2 INTO item_id;
-- 是否还有数据的标识
IF DONE = TRUE THEN
-- 操作数据
UPDATE collection_item SET idx = i WHERE id = item_id;
SET i = i + 1;
-- 提交
COMMIT;
END IF;
END WHILE;
CLOSE cur2;
--游标没有数据之后done会变成FALSE,这里要手动改回来,外层循环才会继续
SET done = TRUE;
END IF;
END WHILE;
CLOSE cur1;
END;$$
DELIMITER ;
-- 调用存储过程
CALL update_collection_item();