如何在pgx中使用goose迁移？

huangapple
go评论199阅读模式
英文:

How to use goose migrations with pgx?

问题

我有一个小的Go Web应用程序，它使用PostgreSQL。

我的数据库驱动程序是pgx。现在我想在应用程序启动时自动运行迁移。我找到了goose来做这件事。

但是，我在使用goose和pgx驱动程序时遇到了困难。

当我运行我的代码时，我会得到以下错误消息：

  1. 打开数据库
  2. 2023/08/09 09:39:28 sql: 未知的驱动程序 "pgx"（是否忘记导入？）

以下是缩短的main.go代码：

  1. package main
  3. import (
  4.     "context"
  5.     "flag"
  6.     "fmt"
  7.     "log"
  8.     "os"
  10.     _ "github.com/jackc/pgx/v5"
  11.     "github.com/pressly/goose/v3"
  12. )
  14. func main() {
  15.     // DSN是PostgreSQL数据源名称，通常与连接字符串相同。
  16.     dsn := flag.String("dsn", "postgres://postgres:password@localhost:5555/plazet", "PostgreSQL数据源名称")
  17.     flag.Parse()
  19.     db, err := openDbConnectionPool(*dsn)
  20.     if err != nil {
  21.         fmt.Printf("无法连接到数据库：%v\n", err)
  22.         os.Exit(1)
  23.     }
  25.     // Ping数据库以检查连接是否正常工作
  26.     connection, err := db.Acquire(context.Background())
  27.     if err != nil {
  28.         fmt.Printf("无法获取连接：%v\n", err)
  29.         os.Exit(1)
  30.     }
  32.     println("打开数据库")
  34.     sql, err := goose.OpenDBWithDriver("pgx", *dsn)
  35.     if err != nil {
  36.         log.Fatalf(err.Error())
  37.     }
  39.     println("迁移")
  41.     err = goose.Up(sql, "./migrations")
  42.     if err != nil {
  43.         log.Fatalf(err.Error())
  44.     }
  46.     defer connection.Release()
  48.     defer db.Close()
  50.     // 路由和其他代码...
  51. }

我检查了OpenDBWithDriver和pgx，它们都是预期的值。在这种情况下，我漏掉了什么？

英文:

I have a small go web application that is using PostgreSQL.

My database driver is pgx. Now I am at a point where I want to run migrations automatically at applications start. I found goose for that.

However, I am struggling to use goose with the pgx driver.

When I run my code I will get:

  1. Opening DB
  2. 2023/08/09 09:39:28 sql: unknown driver "pgx" (forgotten import?)

Here is the shortened main.go:

  1. package main
  3. import (
  4. "context"
  5. "flag"
  6. "fmt"
  7. "log"
  8. "os"
  10.     _ "github.com/jackc/pgx/v5"
  11.     "github.com/pressly/goose/v3"
  13. )
  15. func main() {
  17.     // DSN is the PostgreSQL Data Source Name, Database Source Name.
  18.     // Often the same as connection string.
  19.     dsn := flag.String("dsn", "postgres://postgres:password@localhost:5555/plazet", "PostgreSQL data source name")
  20.     flag.Parse()
  21.     
  22.     db, err := openDbConnectionPool(*dsn)
  23.     if err != nil {
  24.     	fmt.Printf("Unable to connect to database: %v\n", err)
  25.     	os.Exit(1)
  26.     }
  27.     
  28.     // Ping the database to check if the connection is working
  29.     connection, err := db.Acquire(context.Background())
  30.     if err != nil {
  31.     	fmt.Printf("Unable to acquire connection: %v\n", err)
  32.     	os.Exit(1)
  33.     }
  34.     
  35.     println("Opening DB")
  36.     
  37.     sql, err := goose.OpenDBWithDriver("pgx", *dsn)
  38.     if err != nil {
  39.     	log.Fatalf(err.Error())
  40.     }
  41.     
  42.     println("Migrating")
  43.     
  44.     err = goose.Up(sql, "./migrations")
  45.     if err != nil {
  46.     	log.Fatalf(err.Error())
  47.     }
  48.     
  49.     defer connection.Release()
  50.     
  51.     defer db.Close()
  52.     
  53.     // Code for router and stuff...
  55. }

I checked the OpenDBWithDriver and pgx is an expected value. What am I missing in this case?

答案1

得分: 1

驱动程序 pgx 是由 github.com/jackc/pgx/v5/stdlib 包注册的（源代码）:

  1. package stdlib
  3. // ...
  5. func init() {
  6. 	pgxDriver = &Driver{
  7. 		configs: make(map[string]*pgx.ConnConfig),
  8. 	}
  10. 	// 如果不同的 pgx 主要版本已经注册了 pgx 驱动程序，则我们跳过使用默认名称进行注册。
  11. 	if !contains(sql.Drivers(), "pgx") {
  12. 		sql.Register("pgx", pgxDriver)
  13. 	}
  14. 	sql.Register("pgx/v5", pgxDriver)
  15. 	// ...
  16. }

尝试导入该包：

  1. import (
  2.     _ "github.com/jackc/pgx/v5/stdlib"
  3.     "github.com/pressly/goose/v3"
  4. )
英文:

The driver pgx is registered by the github.com/jackc/pgx/v5/stdlib package (source code):

  1. package stdlib
  3. // ...
  5. func init() {
  6. 	pgxDriver = &Driver{
  7. 		configs: make(map[string]*pgx.ConnConfig),
  8. 	}
  10. 	// if pgx driver was already registered by different pgx major version then we
  11. 	// skip registration under the default name.
  12. 	if !contains(sql.Drivers(), "pgx") {
  13. 		sql.Register("pgx", pgxDriver)
  14. 	}
  15. 	sql.Register("pgx/v5", pgxDriver)
  16. 	// ...

Try importing that package instead:

  1. import (
  2.     _ "github.com/jackc/pgx/v5/stdlib"
  3.     "github.com/pressly/goose/v3"
  4. )

huangapple
  • 本文由 发表于 2023年8月9日 15:53:10
  • 转载请务必保留本文链接：https://go.coder-hub.com/76865674-2.html
  • go
  • pgx
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定