Question

以下是有效的代码：

        Connection c = ds.getConnection();
        c.setAutoCommit(false);
        PreparedStatement stmt = c.prepareStatement("INSERT INTO items (name, description) VALUES(?, ?)");
        while (!(items = bus.take()).isEmpty()) {
          for (Item item : items) {
            stmt.setString(1, item.name);
            stmt.setString(2, item.description);
            stmt.addBatch();
          }
          stmt.executeBatch();
          c.commit();
        }

但是现在我需要填充另一个表，其中id是外键。如果我将INSERT与RETURNING id一起使用，则executeBatch将失败并显示“当没有预期时返回结果”错误。

我看到了解决这个问题的几种方法

单独插入而不是批量插入。
使用客户端生成的guid替换序列ID。
使用某种存储过程执行批量插入并返回ID列表。

在我看到的三种方法中，最后一种方法似乎既保留了批量插入的效率又返回了id，但对我来说这也是最复杂的，因为我从未编写过存储过程。

有没有更好的批量插入和获取ID的方法？使用postgresql特定的API而不是jdbc我没有问题。

如果没有，是否可以草拟这样的存储过程？

以下是表架构：

CREATE UNLOGGED TABLE items
(
  id serial,
  name character varying(1000),
  description character varying(10000)
)
WITH (
  OIDS=FALSE
);

Answer 1

这样的事情应该有效：

// tell the driver you want the generated keys
stmt =  c.prepareStatement("INSERT ... ", Statement.RETURN_GENERATED_KEYS);

stmt.executeBatch();

// now retrieve the generated keys
ResultSet rs = stmt.getGeneratedKeys();
while (rs.next()) {
 int id = rs.getInt(1);
 .. save the id somewhere or update the items list 
}

我认为（我不肯定！）按键按生成顺序返回。因此，ResultSet中的第一行应映射到您正在处理的列表中的第一个“项”。但要确认一下！

修改

如果这不起作用，请尝试指定为其生成值的实际列：

stmt = c.prepareStatement("INSERT ... ", new String[] {"id"});

从postgresql中的批处理插入行中检索序列ID

1 个答案: