插入多个条目后获取主键列表

Question

我正在使用python模块MySQLDb一次插入多个条目（有时最多300个）。我的sql看起来像......

INSERT INTO MyTable (row2,row3) values 
(row2_entry1,row3_entry1),(row2_entry2,row3_entry2),........(row2_entry300,row3_entry300)

此表的主键是自动递增的。我想要的是输入表格的条目的主键列表。如何使用批次

执行此操作

Answer 1

MySQL LAST_INSERT_ID()只返回一个值。在批量插入的情况下，它将返回第一个插入行的主键值。

您可以使用GUID标记来跟踪作为批量插入操作一部分的行。在表格中添加一个名为varchar(32)的额外batch_insert_id列。在进行批量插入时，请使用已知的GUID值。插入后，执行select查询以获取插入行的主键。

import uuid

batch_insert_id = uuid.uuid4().hex
# Execute your batch insert query, with batch_insert_id 
# as value of batch_insert_id for all rows
cursor.execute("SELECT id FROM mytable WHERE batch_insert_id = '%s'", (batch_insert_id,))
primary_keys = [row[0] for row in cursor.fetchall()]