postgresql：如何获取通过批量 copy_from 插入的行的主键？

目标是这样的：我有一组值要放入表中A，以及一组要放入表中的值B。进入的值B参考值在A（通过外键），所以插入后A值我需要知道如何在插入时引用它们B价值观。我需要这个尽可能快。

我做了B通过批量复制插入值：

def bulk_insert_copyfrom(cursor, table_name, field_names, values):
    if not values: return

    print "bulk copy from prepare..."
    str_vals = "\n".join("\t".join(adapt(val).getquoted() for val in cur_vals) for cur_vals in values)
    strf = StringIO(str_vals)
    print "bulk copy from execute..."
    cursor.copy_from(strf, table_name, columns=tuple(field_names))

这比做一个要快得多INSERT VALUES ... RETURNING id询问。我想为A值，但我需要知道id插入的行数。

有没有办法以这种方式执行批量复制，但要获取id插入的行的字段（主键），以便我知道哪些id与哪个有关联value?

如果没有，实现我的目标的最佳方法是什么？

编辑：根据要求提供示例数据：

a_val1 = [1, 2, 3]
a_val2 = [4, 5, 6]
a_vals = [a_val1, a_val2]

b_val1 = [a_val2, 5, 6, 7]
b_val2 = [a_val1, 100, 200, 300]
b_val3 = [a_val2, 9, 14, 6]
b_vals = [b_val1, b_val2, b_val3]

我想插入a_vals，然后插入b_vals，使用外键而不是对列表对象的引用。

自己生成 ID。

1. 开始交易
2. 锁表a
3. 调用 nextval() - 这是你的第一个 ID
4. 生成包含 ID 的副本
5. 表b也一样
6. 使用您的最终 ID + 1 调用 setval()
7. 提交交易

在第 2 步，您可能也想锁定序列的关系。如果代码调用 nextval() 并将该 ID 存储在某处，则在使用它时它可能已经在使用中。

稍微偏离主题的事实：如果您有很多后端执行大量插入，则可以设置一个“缓存”设置。这会以块为单位增加计数器。

http://www.postgresql.org/docs/9.1/static/sql-createsequence.html http://www.postgresql.org/docs/9.1/static/sql-createsequence.html