如何获得行值（s）后db插入？

Question

我使用Golang将数据插入到DB中，基本上我的查询如下所示

var cols = "(col1, col2, col3)"
var values = "($1, $2, $3)"
var query = fmt.Sprintf("INSERT INTO %s %s VALUES %s", myTable, cols, values)

res, err := db.Exec(query, thing.val1, thing.val2, thing.val3)

res中唯一可用的内容是lastInsertId和受影响的行的#。但我需要的是受影响的行。原因是我将数据插入到一个psql数据库中，该数据库有一个自动增量id列，所以我希望数据与此一起返回。

例如，使用Java hibernate，我可以做这个答案解释的事情。我不需要在数据库中重新查询ID。

编辑：我试图使用lastInsertId方法，但出现了这个错误

此驱动程序不支持LastInsertId

Answer 1

Postgres驱动程序不支持res. LastInsertId（）。但是，MySQL驱动程序支持它。

db。Exec（）不返回上次插入的id，而是返回db。QueryRow（）。

为了更好地理解，你可以参考这个链接。

在这里，我添加了一个可能对您有所帮助的示例。

var id int
err := db.QueryRow("INSERT INTO user (name) VALUES ('John') RETURNING id").Scan(&id)
if err != nil {
...
}

Answer 2

假设您只想要一个名为id的列中的自动递增值，这是一个带有pq驱动程序的插入

var cols = "(col1, col2, col3)"
var values = "($1, $2, $3)"
var query = fmt.Sprintf(
    "INSERT INTO %s %s VALUES %s RETURNING id",
    myTable, cols, values,
)
var id int
if err := db.QueryRow(
    query,
    thing.val1, thing.val2, thing.val3,
).Scan(&id); err != nil {
    panic(err)
}
fmt.Println("ID: ", id)

对于多个插入：

var cols = "(col1, col2, col3)"
var values = "($1, $2, $3),($4, $5, $6)"
var query = fmt.Sprintf(
    "INSERT INTO %s %s VALUES %s RETURNING id",
    myTable, cols, values,
)
var ids []int
rows, err := db.Query(
    query,
    thing.val1, thing.val2, thing.val3,
    thing.val4, thing.val5, thing.val6,
)
if err != nil {
    panic(err)
}
for rows.Next() {
    var id int
    if err := rows.Scan(&id); err != nil {
        panic(err)
    }
    ids = append(ids, id)
}
fmt.Println("IDs: ", ids)