我正在嘗試在 db 中批量插入密鑰,這里是代碼 Key Struct
type tempKey struct {
keyVal string
lastKey int
}
測驗鍵
data := []tempKey{
{keyVal: "abc", lastKey: 10},
{keyVal: "dns", lastKey: 11},
{keyVal: "qwe", lastKey: 12},
{keyVal: "dss", lastKey: 13},
{keyVal: "xcmk", lastKey: 14},
}
插入部分
dbUrl := "db url...."
conn, err := pgx.Connect(context.Background(), dbUrl)
if err != nil {
println("Errrorr...")
}
defer conn.Close(context.Background())
sqlStr := "INSERT INTO keys (keyval,lastval) VALUES "
dollars := ""
vals := []interface{}{}
count := 1
for _, row := range data {
dollars = fmt.Sprintf("%s($%d, $%d),", dollars, count, count 1)
vals = append(vals, row.keyVal, row.lastKey)
count = 2
}
sqlStr = dollars
sqlStr = sqlStr[0 : len(sqlStr)-1]
fmt.Printf("%s \n", sqlStr)
_, erro := conn.Exec(context.Background(), sqlStr, vals)
if erro != nil {
fmt.Fprint(os.Stderr, "Error : \n", erro)
}
運行時拋出錯誤:預期 10 個引數,得到 1
什么是批量插入的正確方法。
uj5u.com熱心網友回復:
使用批處理(https://github.com/jackc/pgx/blob/master/batch_test.go):
batch := &pgx.Batch{}
batch.Queue("insert into ledger(description, amount) values($1, $2)", "q1", 1)
batch.Queue("insert into ledger(description, amount) values($1, $2)", "q2", 2)
br := conn.SendBatch(context.Background(), batch)
uj5u.com熱心網友回復:
您正在手工制作 SQL 陳述句,這很好,但您沒有利用pgx可以幫助解決此問題的 SQL 陳述句(見下文)。
對于大型輸入,像這樣附加到 SQL 字串可能效率低下
dollars = fmt.Sprintf("%s($%d, $%d),", dollars, count, count 1)
但最終值也有一個尾隨,,您需要一個終止字符;來指示陳述句的結束。
順便說一句,這個字串截斷行是多余的:
sqlStr = sqlStr[0 : len(sqlStr)-1] // this is a NOOP
無論如何,在制作長字串時,最好使用像strings.Builder這樣性能更高的東西。
從pgx檔案中,使用pgx.Conn.CopyFrom:
func (c *Conn) CopyFrom(tableName Identifier, columnNames []string, rowSrc CopyFromSource) (int, error)
CopyFrom 使用 PostgreSQL 復制協議執行批量資料插入。它回傳復制的行數和錯誤。
Copy的示例用法:
rows := [][]interface{}{
{"John", "Smith", int32(36)},
{"Jane", "Doe", int32(29)},
}
copyCount, err := conn.CopyFrom(
pgx.Identifier{"people"},
[]string{"first_name", "last_name", "age"},
pgx.CopyFromRows(rows),
)
轉載請註明出處,本文鏈接:https://www.uj5u.com/qiye/419701.html
標籤:
上一篇:無法從函式回傳字串
