在mysql的实际开发场景中,当我们需要插入成千上万条测试数据或者批量迁移数据时,手动逐条插入效率极低,此时可以通过创建存储过程,结合while循环实现批量插入,大幅提升操作效率。

存储过程与while循环的基本语法
首先我们需要了解mysql中存储过程和while循环的基础语法,避免编写时出现语法错误。
存储过程创建语法
创建存储过程的基本格式如下,我们可以在其中定义变量、编写循环逻辑:
-- 创建存储过程的基本语法
DELIMITER //
CREATE PROCEDURE 存储过程名称()
BEGIN
-- 存储过程的逻辑内容
END //
DELIMITER ;
while循环语法
while循环会在条件满足时持续执行循环体内容,直到条件不成立时退出,语法如下:
WHILE 循环条件 DO
-- 循环体执行的逻辑
END WHILE;
使用while批量插入数据的完整步骤
接下来我们通过一个实际案例,演示如何创建一个存储过程,使用while循环批量插入用户数据到user表中。
第一步:准备目标表
首先我们需要有一个用于接收插入数据的表,这里先创建一个简单的user表:
-- 创建测试用的user表
CREATE TABLE IF NOT EXISTS `user` (
`id` INT PRIMARY KEY AUTO_INCREMENT,
`username` VARCHAR(50) NOT NULL,
`age` INT NOT NULL,
`create_time` DATETIME DEFAULT CURRENT_TIMESTAMP
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
第二步:定义存储过程并编写while循环逻辑
我们需要定义两个变量,一个用来记录当前插入的次数,一个用来设置总共需要插入的数据量,然后通过while循环逐条插入数据:
-- 创建批量插入数据的存储过程
DELIMITER //
CREATE PROCEDURE batch_insert_user(IN insert_count INT)
BEGIN
-- 定义计数器变量,初始值为0
DECLARE i INT DEFAULT 0;
-- 关闭自动提交,提升插入效率
SET autocommit = 0;
-- while循环,当i小于插入数量时持续执行
WHILE i < insert_count DO
-- 插入单条数据,username拼接计数器,age随机生成18到60之间的数
INSERT INTO `user` (`username`, `age`) VALUES (CONCAT('user_', i), FLOOR(18 + RAND() * 43));
-- 计数器自增1
SET i = i + 1;
END WHILE;
-- 提交事务
COMMIT;
-- 恢复自动提交
SET autocommit = 1;
END //
DELIMITER ;
第三步:调用存储过程执行批量插入
存储过程创建完成后,我们只需要调用它并传入需要插入的数据总量即可:
-- 调用存储过程,批量插入1000条数据 CALL batch_insert_user(1000);
执行完成后,我们可以查询user表验证数据是否插入成功:
-- 查询插入的数据,查看前10条 SELECT * FROM `user` LIMIT 10;
注意事项与优化建议
- 插入大量数据时,建议关闭自动提交,在循环结束后统一提交事务,可以减少事务提交的开销,提升插入速度。
- 如果插入的数据量特别大,比如超过10万条,建议分批次插入,避免单次事务过大导致性能问题或者事务日志溢出。
- 循环中的变量名不要和表的字段名重名,否则可能会出现赋值或者判断错误的问题。
- 如果不需要保留存储过程,可以在使用完成后删除,删除语法为
DROP PROCEDURE IF EXISTS batch_insert_user;。
常见问题解答
问:while循环的条件写错了导致无限循环怎么办?
如果出现无限循环,可以新开一个mysql连接,执行SHOW PROCESSLIST;找到正在执行的存储过程对应的进程id,然后使用KILL 进程id;终止进程,之后检查循环条件重新创建存储过程。
问:插入的数据中文出现乱码怎么办?
检查表的字符集设置,确保表使用utf8mb4字符集,同时在插入数据时如果包含中文,保证连接mysql的客户端字符集也设置为utf8mb4即可。