无法将 txdb 与 Gormigrate 一起使用

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《无法将 txdb 与 Gormigrate 一起使用》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我正在尝试使用 txdb 在隔离事务内迁移数据库，以便测试与数据库交互的代码以及测试迁移本身。但每次我运行代码时都会返回错误，并且 gorm 会发出一些警告：

$ database_url="postgres://postgres@localhost:5432/postgres?sslmode=disable" ./txdb_ejemplo
`txdb_postgres` is not officially supported, running under compatibility mode.

(/home/jorge/go/pkg/mod/gopkg.in/[email protected]/gormigrate.go:381)
[2019-08-12 19:46:28]  pq: current transaction is aborted, commands ignored until end of transaction block
panic: migration failed: pq: current transaction is aborted, commands ignored until end of transaction block

goroutine 1 [running]:
main.main()
        /home/jorge/proyectos/kue/fero/txdb_ejemplo/main.go:76 +0x3a1

这是运行提供数据库的 docker 服务的输出：

db_1  | 2019-08-13 01:46:08.788 utc [1] log:  database system is ready to accept connections
db_1  | 2019-08-13 01:46:28.564 utc [48] error:  function database() does not exist at character 8
db_1  | 2019-08-13 01:46:28.564 utc [48] hint:  no function matches the given name and argument types. you might need to add explicit type casts.
db_1  | 2019-08-13 01:46:28.564 utc [48] statement:  select database()
db_1  | 2019-08-13 01:46:28.564 utc [48] error:  syntax error at or near "and" at character 71
db_1  | 2019-08-13 01:46:28.564 utc [48] statement:  select count(*) from information_schema.tables where table_schema = ? and table_name = ?
db_1  | 2019-08-13 01:46:28.564 utc [48] error:  current transaction is aborted, commands ignored until end of transaction block
db_1  | 2019-08-13 01:46:28.564 utc [48] statement:  create table migrations (id varchar(255) primary key)

我不明白我做错了什么，但我提供了一个最小的功能示例来重现我遇到的问题，这里是包含代码的存储库，并且有 go 代码的入门： p>

package main

import (
    "fmt"
    "os"
    "time"

    "github.com/DATA-DOG/go-txdb"
    "github.com/jinzhu/gorm"
    _ "github.com/lib/pq"
    gormigrate "gopkg.in/gormigrate.v1"
)

var (
    migration_1 = func(tx *gorm.DB) error {
        type Person struct {
            gorm.Model
            Name string
        }

        return tx.CreateTable(&Person{}).Error
    }
    migration_2 = func(tx *gorm.DB) error {
        type Person struct {
            Age int
        }
        return tx.AutoMigrate(&Person{}).Error
    }
    return_nil = func(tx *gorm.DB) error {
        return nil
    }
)

type Person struct {
    gorm.Model
    Name string
    Age  int
}

func MigrateAll(gdb *gorm.DB) error {
    m := gormigrate.New(gdb, gormigrate.DefaultOptions, []*gormigrate.Migration{
        {
            ID:       "first",
            Migrate:  migration_1,
            Rollback: return_nil,
        },
        {
            ID:       "second",
            Migrate:  migration_2,
            Rollback: return_nil,
        },
    })
    return m.Migrate()
}

func main() {
    url := os.Getenv("DATABASE_URL")
    txdb.Register("txdb_postgres", "postgres", url)
    var db *gorm.DB
    var err error
    for i := 0; i < 3; i++ {
        db, err = gorm.Open("txdb_postgres", "tx_1")
        if err == nil {
            break
        }
        fmt.Printf("connection failed, retrying in 10 seconds. Reason: %s\n", err)
        time.Sleep(10 * time.Second)
    }
    if err != nil {
        panic(fmt.Sprintf("connection failed: %s", err))
    }

    defer db.Close()
    err = MigrateAll(db)
    if err != nil {
        panic(fmt.Sprintf("migration failed: %s", err))
    }
}

解决方案

问题在于 gorm 使用第一个 gorm.open 参数作为方言名称（它有一组预定义的方言），而不是作为 database/sql 驱动程序名称。

因此，您需要使用 gorm 已知方言名称作为第一个参数，在本例中，您需要 "postgres"，因为您将使用 postgresql 作为底层数据库，因此您需要 gorm 使用相应的 sql 方言。 p>

然后告诉 gorm 使用 "txdb" 驱动程序和 "tx_1" 作为连接 url。

这似乎有效：

package main

import (
    "fmt"
    "os"
    "time"

    "github.com/data-dog/go-txdb"
    "github.com/jinzhu/gorm"
    _ "github.com/lib/pq"
    gormigrate "gopkg.in/gormigrate.v1"
)

var (
    migration_1 = func(tx *gorm.db) error {
        type person struct {
            gorm.model
            name string
        }

        return tx.createtable(&person{}).error
    }
    migration_2 = func(tx *gorm.db) error {
        type person struct {
            age int
        }
        return tx.automigrate(&person{}).error
    }
    return_nil = func(tx *gorm.db) error {
        return nil
    }
)

type person struct {
    gorm.model
    name string
    age  int
}

func migrateall(gdb *gorm.db) error {
    m := gormigrate.new(gdb, gormigrate.defaultoptions, []*gormigrate.migration{
        {
            id:       "first",
            migrate:  migration_1,
            rollback: return_nil,
        },
        {
            id:       "second",
            migrate:  migration_2,
            rollback: return_nil,
        },
    })
    return m.migrate()
}

func main() {
    url := os.getenv("database_url")
    txdb.register("txdb", "postgres", url)
    var db *gorm.db
    var err error
    for i := 0; i < 3; i++ {
        db, err = gorm.open("postgres", "txdb", "tx_1")
        if err == nil {
            break
        }
        fmt.printf("connection failed, retrying in 10 seconds. reason: %s\n", err)
        time.sleep(10 * time.second)
    }
    if err != nil {
        panic(fmt.sprintf("connection failed: %s", err))
    }

    defer db.close()
    err = migrateall(db)
    if err != nil {
        panic(fmt.sprintf("migration failed: %s", err))
    }
}

或者，您可以直接传递 gorm.open *sql.db，相关部分如下：

s, err := sql.Open("txdb", "tx_1")
        // handle err
        db, err := gorm.Open("postgres", s)
        // handle err

那也行。

今天关于《无法将 txdb 与 Gormigrate 一起使用》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！