MySQL 批量插入并返回批量自增ID的方法

在MySQL数据库表中插入大量记录时,若逐条插入时,性能非常性,也比较耗,因此MySQL提供批量插入记录的方法,具体语法为:

Insert into myTable(Field1,Field2,...Fieldn) values
(V11,V12,…V1n),
(V21,V22,…V2n),
……,
(Vn1,Vn2,…Vnn)

上述语句,可以一次性插入大量的数据,具体记录的数量只受系统变量max_allowed_packet的限制,同时提高高数据插入的速度;但有一个问题,就是插入数据后,无法得知插入记录的自增ID,如果需要逐条去查询,那个效率会低得让人无法忍受!

这里就有一种解决办法,可以获取新插入记录的ID,具体思路是:

  1. 锁表,并获取当前表的自增ID;
  2. 执行插入操作;
  3. 获取当前表的自增ID;
  4. 表解锁

其中步骤2,3可以放在存储过程中执行(MySQL不支持在存储过程中锁表);这样的话,前后两次获得的自增ID区间内的值,就是新插入记录的自增ID,可以依据插入顺序为其赋值,

测试表如下:

CREATE TABLE `device`  (
  `ID` bigint(20) UNSIGNED NOT NULL AUTO_INCREMENT,
  `Name` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
  `IP` varchar(32) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL,
  `Port` int(8) NULL DEFAULT NULL,
  PRIMARY KEY (`ID`) USING BTREE
) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Compact;

对测试表作批量插入并取得自增ID的语句如下:

LOCK tables `device` WRITE; 	-- 锁表
	-- 取得插入前的自增ID
	SELECT `AUTO_INCREMENT` into @MaxID1 FROM  INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'MyDataBase' AND  TABLE_NAME  = 'device'; 
	INSERT INTO `device`(`ID`, `Name`, `IP`, `Port`) VALUES 
	(0, 'Dev1', '192.168.2.1', 80),
	(0, 'Dev2', '192.168.2.2', 80),
	(0, 'Dev3', '192.168.2.3', 80),
	(0, 'Dev4', '192.168.2.4', 80),
	(0, 'Dev5', '192.168.2.5', 80);
	-- 取得插入后的自增ID
	SELECT `AUTO_INCREMENT` into @MaxID2 FROM  INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'MyDataBase' AND  TABLE_NAME  = 'device';
	SELECT  @MaxID1,@MaxID2;		-- 输出自增ID
	UNLOCK table;		-- 解除所有表的锁定

在输入记录中,@MaxID1和@MaxID2区间的ID,就是新增记录的ID,而且是按顺序连续递增的,可以依次分配给插入前的数据;

注意:在实际应用过程中,以上语句可能会存在锁表的隐患,因为插入数据过程中可能会发生异常,导致无法正常执行Unlock talbe,因此建议把步骤2,3放到存储过程中,将在存储过程作异常捕捉,以防止异常影响到Unlock talbe的执行;

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在PostgreSQL(简称PGSQL)中,批量插入ID通常不直接支持像某些SQL方言(如MySQL的`INSERT INTO SELECT`)那样一次性插入多行并自动获取连续的自ID。因为PostgreSQL的序列(sequences)是用于生成单个值的,而不是为整个批处理分配ID。 然而,你可以通过以下步骤实现类似的功能: 1. **使用循环或临时表**: - 创建一个序列(`CREATE SEQUENCE your_sequence_name`)。 - 在循环中,每次插入一行数据后,使用`nextval('your_sequence_name')`获取下一个ID,然后将这个ID插入到数据库。 ```sql DO $$ BEGIN FOR i IN 1..10 LOOP INSERT INTO your_table (id, other_columns) VALUES (nextval('your_sequence_name'), ...); -- 如果有多个列需要插入ID,替换为相应的值 END LOOP; END $$; ``` 2. **利用存储过程**: - 创建一个存储过程,负责插入数据和获取ID。 ```sql CREATE OR REPLACE FUNCTION insert_data() RETURNS VOID AS $$ DECLARE id INTEGER DEFAULT nextval('your_sequence_name'); BEGIN FOR i IN 1..10 LOOP INSERT INTO your_table (id, other_columns) VALUES (id, ...); -- 递ID并继续插入 id := id + 1; END LOOP; END; $$ LANGUAGE plpgsql; -- 调用存储过程 PERFORM insert_data(); ``` 请注意,以上方法虽然可以达到类似的效果,但并不直接支持一次批量插入操作,且效率可能会比单行插入稍低。如果需要优化性能,可能需要考虑其他策略,比如预先计算所有行的ID再一次性插入
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值