mysql 存储批量插入

原创

mob649e81637cea 2024-09-01 03:21:06 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e81637cea的原创作品，请联系作者获取转载授权，否则将追究法律责任

MySQL 批量插入的实现方法

批量插入数据是数据库操作中常见的需求，尤其是在需要导入大量数据时。对于刚入行的小白开发者，了解如何在 MySQL 中实现批量插入是一项基本技能。本文将详细介绍这一过程的步骤，并提供相应的代码示例。

一、整体流程

在进行批量插入之前，我们需要明确一些步骤。以下表格展示了实现 MySQL 存储批量插入的基本流程：

步骤	描述
1	创建数据库和数据表
2	准备需要插入的数据
3	使用 MySQL 执行批量插入操作
4	验证数据插入是否成功

二、步骤详解

步骤 1: 创建数据库和数据表

首先，我们需要在 MySQL 中创建一个数据库和数据表。假设我们要创建一个简单的用户表，用于存储用户信息。

-- 创建一个数据库
CREATE DATABASE userDB;

-- 使用数据库
USE userDB;

-- 创建一个用户表
CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(50) NOT NULL,
    age INT NOT NULL
);

代码解释：

CREATE DATABASE userDB;：创建一个名为 userDB 的数据库。
USE userDB;：选择使用 userDB 数据库。
CREATE TABLE users (...);：创建一个名为 users 的表，包含自增的 id、用户 name 和 age 字段。

步骤 2: 准备需要插入的数据

在实际开发中，数据往往来自文件、API 或其它数据源。为了便于说明，我们这里手动准备一组示例数据：

# 准备要插入的数据
data_to_insert = [
    ('Alice', 25),
    ('Bob', 30),
    ('Charlie', 35)
]

代码解释：

这里使用 Python 列表，准备了一组包含用户名称和年龄的元组，后续将批量插入到 users 表中。

步骤 3: 使用 MySQL 执行批量插入操作

批量插入操作可以通过单条 SQL 语句来实现。我们可以使用 Python 的 mysql-connector-python 库来执行插入操作。首先确保安装了这个库：

pip install mysql-connector-python

然后，可以使用以下代码实现批量插入：

import mysql.connector

# 连接到 MySQL 数据库
connection = mysql.connector.connect(
    host='localhost',
    user='your_username',     # 替换为你的 MySQL 用户名
    password='your_password',  # 替换为你的 MySQL 密码
    database='userDB'         # 使用刚创建的数据库
)

cursor = connection.cursor()

# 准备 SQL 插入语句
insert_query = "INSERT INTO users (name, age) VALUES (%s, %s)"
# 执行批量插入
cursor.executemany(insert_query, data_to_insert)

# 提交事务
connection.commit()

# 关闭连接
cursor.close()
connection.close()

代码解释：

import mysql.connector：导入 MySQL 连接库。
使用 mysql.connector.connect(...) 连接到 MySQL 数据库，传入主机、用户名、密码和数据库名。
cursor = connection.cursor()：获取游标操作数据库。
insert_query = "INSERT INTO users (name, age) VALUES (%s, %s)"：准备插入 SQL 查询语句，将数据占位符 %s 用于之后的替换。
cursor.executemany(insert_query, data_to_insert)：执行批量插入，通过 executemany 方法将数据列表一次性插入到数据库中。
connection.commit()：提交事务，将更改保存到数据库。
cursor.close() 和 connection.close()：关闭游标和连接，确保资源释放。

步骤 4: 验证数据插入是否成功

最后，我们可以通过查询表格来验证数据是否成功插入：

# 重新连接到数据库
connection = mysql.connector.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='userDB'
)

cursor = connection.cursor()

# 查询所有用户数据
cursor.execute("SELECT * FROM users")

# 获取并打印所有结果
for row in cursor.fetchall():
    print(row)

# 关闭连接
cursor.close()
connection.close()

代码解释：