如何一次插入多个数据

Question

我知道一次插入多个数据效率更高：

 INSERT INTO test(n1, n2, n3)
VALUES(v1, v2, v3),(v4, v5, v6),(v7, v8, v9);

如何在 golang 中做到这一点？

 data := []map[string]string{
   {"v1":"1", "v2":"1", "v3":"1"},
   {"v1":"2", "v2":"2", "v3":"2"},
   {"v1":"3", "v2":"3", "v3":"3"},
}
//I do not want to do it
for _, v := range data {
    sqlStr := "INSERT INTO test(n1, n2, n3) VALUES(?, ?, ?)"
    stmt, _ := db.Prepare(sqlStr)
    res, _ := stmt.Exec(v["v1"], v["v2"], v["v3"])
}

使用字符串拼接，但效果不佳。 db.Prepare 更安全吧？

 sqlStr := "INSERT INTO test(n1, n2, n3) VALUES"
for k, v := range data {
    if k == 0 {
        sqlStr += fmt.Sprintf("(%v, %v, %v)", v["v1"], v["v2"], v["v3"])
    } else {
        sqlStr += fmt.Sprintf(",(%v, %v, %v)", v["v1"], v["v2"], v["v3"])
    }
}
res, _ := db.Exec(sqlStr)

我需要一个更安全、更高效的函数，一次插入多个数据。

原文由 leiyonglin 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

为什么不这样呢？ （这里写的没有测试所以可能会有语法错误）：

 sqlStr := "INSERT INTO test(n1, n2, n3) VALUES "
vals := []interface{}{}

for _, row := range data {
    sqlStr += "(?, ?, ?),"
    vals = append(vals, row["v1"], row["v2"], row["v3"])
}
//trim the last ,
sqlStr = sqlStr[0:len(sqlStr)-1]
//prepare the statement
stmt, _ := db.Prepare(sqlStr)

//format all vals at once
res, _ := stmt.Exec(vals...)

原文由 Not_a_Golfer 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

对于 Postgres lib pq 支持批量插入： https ://godoc.org/github.com/lib/pq#hdr-Bulk_imports

但同样可以通过下面的代码实现，但真正有用的地方是当一个人试图执行批量条件更新（相应地更改查询）时。

要为 Postgres 执行类似的批量插入，您可以使用以下函数。

 // ReplaceSQL replaces the instance occurrence of any string pattern with an increasing $n based sequence
func ReplaceSQL(old, searchPattern string) string {
   tmpCount := strings.Count(old, searchPattern)
   for m := 1; m <= tmpCount; m++ {
      old = strings.Replace(old, searchPattern, "$"+strconv.Itoa(m), 1)
   }
   return old
}

所以上面的例子变成了

sqlStr := "INSERT INTO test(n1, n2, n3) VALUES "
vals := []interface{}{}

for _, row := range data {
   sqlStr += "(?, ?, ?),"
   vals = append(vals, row["v1"], row["v2"], row["v3"])
}

//trim the last ,
sqlStr = strings.TrimSuffix(sqlStr, ",")

//Replacing ? with $n for postgres
sqlStr = ReplaceSQL(sqlStr, "?")

//prepare the statement
stmt, _ := db.Prepare(sqlStr)

//format all vals at once
res, _ := stmt.Exec(vals...)

原文由 Debasish Mitra 发布，翻译遵循 CC BY-SA 4.0 许可协议