如何用一条SQL查询插入数据到多个表？

2024-07-13 00:35:31

如何使用一条 SQL 查询同时插入数据到多个表？

在数据库操作中，我们经常需要将数据插入到多个相关联的表中。例如，在电商系统中，创建新订单时，我们需要同时在订单表、订单详情表、库存表等多个表中插入数据。如果采用逐条执行 INSERT 语句的方式，不仅效率低下，还容易出现数据不一致的问题。那么，有没有更高效、更可靠的方法呢？

答案是肯定的！我们可以利用 SQL 中的事务和子查询特性，使用一条 SQL 查询就完成所有插入操作，大幅提升效率并确保数据一致性。

场景分析

假设我们正在开发一个博客系统，需要存储用户信息和文章信息。数据库中包含两个表：

users 表：存储用户信息，包含 id 和 username 两列。
articles 表：存储文章信息，包含 id、user_id 和 title 三列。

现在，我们需要将新用户 "Alice" 及其发布的文章 "Hello World" 的信息插入到数据库中。传统的方法是分别对 users 和 articles 表执行两条 INSERT 语句。但这种方法效率低下，并且如果其中一条语句执行失败，会导致数据不一致。

解决方案：事务与子查询

为了解决这个问题，我们可以采用以下步骤：

开启事务： 使用 BEGIN TRANSACTION 语句开启一个新的事务。事务可以将多个 SQL 语句视为一个原子操作，要么全部执行成功，要么全部回滚，确保数据的一致性。
插入数据到主表： 将数据插入到作为主表的目标表中，例如 users 表。
获取插入数据的 ID： 使用 LAST_INSERT_ID() 函数获取刚刚插入到主表中的记录的 ID。
插入数据到关联表： 使用子查询将数据插入到关联表中，例如 articles 表，并将步骤 3 中获取的 ID 作为关联表的外部键，建立表与表之间的关联关系。
提交事务： 使用 COMMIT 语句提交事务，使所有更改永久生效。

代码示例

以下是以 MySQL 数据库为例，实现上述步骤的代码示例：

BEGIN TRANSACTION;

INSERT INTO users (username) VALUES ('Alice');

SET @user_id = LAST_INSERT_ID();

INSERT INTO articles (user_id, title) VALUES (@user_id, 'Hello World');

COMMIT;

代码解析:

BEGIN TRANSACTION; 开启一个新的事务，保证操作的原子性。
INSERT INTO users (username) VALUES ('Alice'); 将用户名 "Alice" 插入到 users 表中。
SET @user_id = LAST_INSERT_ID(); 将 users 表中最后插入的记录的 ID（即 "Alice" 的用户 ID）存储到变量 @user_id 中，方便后续使用。
INSERT INTO articles (user_id, title) VALUES (@user_id, 'Hello World'); 将文章标题 "Hello World" 和对应的用户 ID (@user_id) 插入到 articles 表中，建立文章与用户的关联关系。
COMMIT; 提交事务，所有更改生效。