使用 pgx 迁移 goose 的指南

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《使用 pgx 迁移 goose 的指南》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

我有一个使用 postgresql 的小型 go web 应用程序。

我的数据库驱动程序是pgx。现在我想在应用程序启动时自动运行迁移。我为此找到了鹅。

但是，我很难将 goose 与 pgx 驱动程序一起使用。

当我运行我的代码时，我会得到：

opening db
2023/08/09 09:39:28 sql: unknown driver "pgx" (forgotten import?)

这是缩短的 main.go：

package main

import (
"context"
"flag"
"fmt"
"log"
"os"

    _ "github.com/jackc/pgx/v5"
    "github.com/pressly/goose/v3"

)

func main() {

    // DSN is the PostgreSQL Data Source Name, Database Source Name.
    // Often the same as connection string.
    dsn := flag.String("dsn", "postgres://postgres:password@localhost:5555/plazet", "PostgreSQL data source name")
    flag.Parse()
    
    db, err := openDbConnectionPool(*dsn)
    if err != nil {
        fmt.Printf("Unable to connect to database: %v\n", err)
        os.Exit(1)
    }
    
    // Ping the database to check if the connection is working
    connection, err := db.Acquire(context.Background())
    if err != nil {
        fmt.Printf("Unable to acquire connection: %v\n", err)
        os.Exit(1)
    }
    
    println("Opening DB")
    
    sql, err := goose.OpenDBWithDriver("pgx", *dsn)
    if err != nil {
        log.Fatalf(err.Error())
    }
    
    println("Migrating")
    
    err = goose.Up(sql, "./migrations")
    if err != nil {
        log.Fatalf(err.Error())
    }
    
    defer connection.Release()
    
    defer db.Close()
    
    // Code for router and stuff...

}

我检查了 opendbwithdriver 并且 pgx 是预期值。在这种情况下我错过了什么？

正确答案

驱动程序pgx由github.com/jackc/pgx/v5/stdlib包（source code）注册：

package stdlib

// ...

func init() {
    pgxdriver = &driver{
        configs: make(map[string]*pgx.connconfig),
    }

    // if pgx driver was already registered by different pgx major version then we
    // skip registration under the default name.
    if !contains(sql.drivers(), "pgx") {
        sql.register("pgx", pgxdriver)
    }
    sql.register("pgx/v5", pgxdriver)
    // ...

尝试导入该包：

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

对于任何在这里搜索如何将 pgxpool 与 goose 一起使用的人 - 您现在可以使用包 pgx/v5/stdlib 中的 opendbfrompool 将 *pgxpool.pool 转换为 *sql.db。该功能在v5.5.0版本中引入。

import (
    "github.com/jackc/pgx/v5/pgxpool"
    "github.com/jackc/pgx/v5/stdlib"
    "github.com/pressly/goose/v3"
)

// ...

func main() {
    var pool *pgxpool.Pool
    // setup pool

    goose.SetBaseFS(embedMigrations)

    if err := goose.SetDialect("postgres"); err != nil {
       panic(err)
    }

    db := stdlib.OpenDBFromPool(pool)
    if err := goose.Up(db, "migrations"); err != nil {
        panic(err)
    }
    if err := db.Close(); err != nil {
        panic(err)
    }

    // run app
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~