存储过程及双循环游标使用

遇到了一个问题，有一张表，需要增加一列idx，根据collectionId分组查询数据，然后为idx赋值。
分割出来其实就是两个循环。

1. 外层循环SELECT DISTINCT collectionId FROM collection_item。
2. 内层循环 SELECT id FROM collection_item WHERE collectionId = collection_id。
3. 然后更新 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();