在go语言中使用PGX CopyFrom扩展包实现向Postgres数据库插入数据示例

Go  /  管理员 发布于 1年前   420

PGX CopyFrom是一个PostgreSQL扩展，用于将数据从一个文件复制到 PostgreSQL 数据库。

要使用 CopyFrom，首先需要安装 PGX 扩展，

然后创建一个表结构与源文件相匹配的表，最后使用 COPY 命令将数据从文件导入到表中。

下面是一个如何使用PGX 的 CopyFrom 函数向 Postgres 数据库插入数据的示例。

HScodes 表是我自己的参考，你需要用你自己的表和结构来代替它。

代码示例：

 type HScodes struct {
     Sid     int    `json:"sid" db:"sid"`
     HScode  string `json:"name" db:"hscode"`
     Keyword string `json:"value" db:"keyword"`
 }
 
 var dataToInsert []models.HScodes
 
 rowsToInsert := [][]interface{}{}
 
 for i := 0; i < len(dataToInsert); i++ {
     row := []interface{}{dataToInsert[i].HScode, dataToInsert[i].Keyword}
     rowsToInsert = append(rowsToInsert, row)
 }
 
 copyCount, err := database.WrapCopyFrom(ctx, pgx.Identifier{"hscodes"},
 []string{"hscode", "keyword"},
 pgx.CopyFromRows(rowsToInsert))

相关链接：

https://github.com/jackc/pgx/issues/684