mysql之——存储过程 + 游标 + 事务

下面是自己曾经编写过的mysql数据库存储过程，留作存档，以后用到的时候拿来参考。

其中，涉及到了存储过程、游标（双层循环）、事务。

【说明】：代码中的注释只针对当时业务而言，无须理会。

代码如下：

DELIMITER $$
DROP PROCEDURE IF EXISTS `transferEmailTempData`$$

CREATE PROCEDURE transferEmailTempData(IN jobId VARCHAR())
BEGIN
    DECLARE idval VARCHAR() DEFAULT '';
    DECLARE taskIdval VARCHAR() DEFAULT '';
    DECLARE groupIdval VARCHAR() DEFAULT '';
    DECLARE emailval VARCHAR() DEFAULT '';

    /*标识正式表是否存在一条相同数据，即：groupId、email相同*/
    DECLARE infoId VARCHAR() DEFAULT '';

    /*标识事务错误*/
    DECLARE err INT DEFAULT ;

    /*达到一定数量就进行提交，计数器*/
    DECLARE counts INT DEFAULT ;

    /*标识是否回滚过*/
    DECLARE isrollback INT DEFAULT ;

    /*游标遍历时，作为判断是否遍历完全部记录的标记*/
    DECLARE done INTEGER DEFAULT ;

    /*获取临时表该任务的数据*/
    DECLARE cur CURSOR FOR SELECT id,taskId,groupId,email FROM `t_email_data_temp` WHERE taskId=jobId;

    /*根据群组id、email查询是否存在相同记录*/
    DECLARE cur2 CURSOR FOR SELECT id FROM `t_email_info` e WHERE e.`group_id` = groupIdval AND e.`email_address` = emailval; 

    /* 出现错误，设置为1，只要发生异常就回滚*/
    DECLARE CONTINUE HANDLER FOR SQLEXCEPTION SET err=;

    /*声明当游标遍历完全部记录后将标志变量置成某个值*/
    DECLARE CONTINUE HANDLER FOR NOT FOUND
    SET done=;

    /*开启事务*/
    START TRANSACTION;

    /*打开游标*/
    OPEN cur;

    /*使用LOOP循环遍历*/
    out_loop:LOOP

        /*将每一条结果对应的字段值赋值给变量*/
        FETCH cur INTO idval,taskIdval,groupIdval,emailval;
        IF done =  THEN
            LEAVE out_loop;
        END IF;

        /*打开第二个游标*/
        OPEN cur2;
            SET done = ;
            FETCH cur2 INTO infoId;

            /*如果正式表不存在相同groupId and email记录，添加到正式表*/
            IF done =  THEN

                /*插入正式表*/
                INSERT INTO `t_email_info` VALUES(idval,emailval,groupIdval,,'',NOW(),'admin',NOW(),'admin');

                /*删除临时数据*/
                DELETE FROM `t_email_data_temp` WHERE id = idval;

                /*计数器，每1000条才提交*/
                SET counts = counts + ;

                /*发生异常，回滚*/
                IF err= THEN
                    SET isrollback=;
                    ROLLBACK;
                ELSE
                    IF counts =  THEN
                        COMMIT;
                        /*达到1000条提交后，重置计数器*/
                        SET counts=;
                    END IF;
                END IF;
            ELSE
                /*已经存在相同记录，则删除该记录*/
                IF done= THEN
                    DELETE FROM `t_email_data_temp` WHERE id = idval;
                END IF;
            END IF;
            FETCH cur2 INTO infoId;
        CLOSE cur2;

        /*控制外部的循环，该步骤不能缺少，否则只循环一次就结束了*/
        SET done=;

    END LOOP out_loop;
    CLOSE cur;

    /*如果没有发生过回滚事件，则更新task状态*/
    /*如果回滚过，不更新task状态，下次执行任务的时候，会再次将剩余没有提交的数据进行添加到正式表*/
    IF isrollback= THEN
        UPDATE `t_email_task` t SET t.`if_finish` =  WHERE t.`id`=jobId;
    END IF;

    END$$

DELIMITER ;