如何在PostgreSQL中批量插入并返回主键ID_利用INSERT INTO RETURNING语句

在PostgreSQL数据库的实际开发场景中，经常需要一次性插入多条数据，并且要获取这些新插入数据对应的主键ID，用于后续的业务逻辑处理。INSERT INTO RETURNING语句就是PostgreSQL提供的专门用于解决这类需求的语法，它可以在执行插入操作的同时，返回插入行的指定列值，其中就包括自增主键。

INSERT INTO RETURNING基础用法

首先来看单条插入返回主键的基础用法，假设我们有一张用户表，结构如下：

-- 创建用户表，id为自增主键
CREATE TABLE user_info (
    id SERIAL PRIMARY KEY,
    user_name VARCHAR(50) NOT NULL,
    age INT,
    create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

单条插入并返回主键ID的SQL语句写法如下：

-- 插入单条用户数据，返回生成的id
INSERT INTO user_info (user_name, age) VALUES ('张三', 25) RETURNING id;

执行上面的语句后，PostgreSQL会在插入数据的同时，返回新生成的主键id值，不需要额外执行查询语句去获取。

批量插入返回主键ID的实现

批量插入的场景下，只需要在VALUES后面跟上多组数据，RETURNING子句会返回所有插入行的主键ID，顺序和插入的数据顺序一致。

纯SQL批量插入返回示例

批量插入多条用户数据并返回所有主键的SQL如下：

-- 批量插入3条用户数据，返回所有生成的主键id
INSERT INTO user_info (user_name, age) 
VALUES 
('李四', 22),
('王五', 28),
('赵六', 30)
RETURNING id;

执行后结果会返回三个id值，分别对应三条插入的记录，顺序是李四、王五、赵六的对应主键。

不同编程语言中的使用示例

在实际开发中，我们通常会通过编程语言操作数据库，下面给出几种常见语言的实现示例。

Python（psycopg2驱动）示例

import psycopg2

# 连接数据库
conn = psycopg2.connect(
    dbname="test_db",
    user="postgres",
    password="123456",
    host="127.0.0.1",
    port="5432"
)
cursor = conn.cursor()

# 批量插入的数据
user_list = [('李四', 22), ('王五', 28), ('赵六', 30)]
# 构造插入SQL，使用RETURNING返回id
insert_sql = "INSERT INTO user_info (user_name, age) VALUES %s RETURNING id"
# 执行批量插入
cursor.executemany(insert_sql, user_list)
# 获取所有返回的主键id
id_list = cursor.fetchall()
conn.commit()

print("批量插入的主键ID列表：", id_list)
cursor.close()
conn.close()

Java（JDBC）示例

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

public class PostgresBatchInsert {
    public static void main(String[] args) {
        String url = "jdbc:postgresql://127.0.0.1:5432/test_db";
        String user = "postgres";
        String password = "123456";
        List<Integer> idList = new ArrayList<>();
        
        try (Connection conn = DriverManager.getConnection(url, user, password)) {
            // 批量插入的SQL
            String sql = "INSERT INTO user_info (user_name, age) VALUES (?, ?) RETURNING id";
            PreparedStatement pstmt = conn.prepareStatement(sql);
            
            // 添加批量参数
            pstmt.setString(1, "李四");
            pstmt.setInt(2, 22);
            pstmt.addBatch();
            
            pstmt.setString(1, "王五");
            pstmt.setInt(2, 28);
            pstmt.addBatch();
            
            pstmt.setString(1, "赵六");
            pstmt.setInt(2, 30);
            pstmt.addBatch();
            
            // 执行批量操作
            ResultSet rs = pstmt.executeQuery();
            // 遍历结果集获取主键ID
            while (rs.next()) {
                idList.add(rs.getInt("id"));
            }
            System.out.println("批量插入的主键ID列表：" + idList);
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

注意事项

• RETURNING子句返回的主键顺序和VALUES中数据的顺序完全一致，不需要额外排序。
• 如果插入的表主键不是自增类型，而是手动赋值的，RETURNING也会返回你插入时指定的主键值。
• 批量插入的数据量过大时，要注意数据库的连接超时和内存占用问题，可以适当拆分批次执行。
• 该语法是PostgreSQL的特有语法，其他数据库如MySQL、SQL Server的类似功能语法会有差异，不要混用。

INSERT INTO RETURNING语句是PostgreSQL中非常实用的特性，相比先插入再查询主键的方式，减少了数据库交互次数，能显著提升批量操作的效率，建议在符合场景的情况下优先使用。

PostgreSQLINSERT_INTO_RETURNING批量插入主键ID返回修改时间：2026-07-21 02:54:33