MySQL批量插入获取ID|极客教程

MySQL批量插入获取ID

在实际的数据库操作中，经常会遇到需要批量插入数据并且需要获取插入后的自增ID的情况。在这种情况下，我们需要一种方法来实现批量插入并且能够获取到每条数据的ID。

本文将介绍如何在MySQL数据库中进行批量插入并且获取到插入后的自增ID。我们将使用Python编写代码，并且通过PyMySQL库来实现数据库操作。

准备工作

在开始之前，我们需要安装PyMySQL库。我们可以使用pip来进行安装：

pip install pymysql

接下来，我们需要在MySQL数据库中创建一个示例表，用于存储我们的数据。这里我们使用一个示例表结构如下：

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255) NOT NULL
);

批量插入获取ID

首先，我们需要编写一个函数来实现批量插入数据并获取到插入后的自增ID。下面是一个示例的Python代码：

import pymysql

def batch_insert_get_id(data_list):
    conn = pymysql.connect(
        host='localhost',
        user='root',
        password='password',
        database='test'
    )
    cursor = conn.cursor()

    try:
        sql = "INSERT INTO users (name) VALUES (%s)"
        cursor.executemany(sql, data_list)
        conn.commit()

        ids = [cursor.lastrowid + i for i in range(len(data_list))]
        return ids

    except:
        conn.rollback()
        raise

    finally:
        cursor.close()
        conn.close()

data_list = [('Alice',), ('Bob',), ('Charlie',)]

ids = batch_insert_get_id(data_list)
print(ids)

在这段代码中，我们首先建立了与数据库的连接。然后使用executemany()方法执行批量插入操作。最后，我们通过cursor.lastrowid来获取最后插入的自增ID，并根据插入的数据量来推算出每条数据的ID。