您能否使用golang-migrate与Go的嵌入功能一起使用？

Question

我想使用golang-migrate将一个sql文件迁移到我的postgresql数据库中。我可能做错了什么，但当我运行迁移命令时，它说找不到模式：

 $ go run ./cmd/  migrate
2022/04/05 16:20:29 no scheme
exit status 1

这是代码:

  // package dbschema contains the database schema, migrations, and seeding data.

    package dbschema
    
    import (
        "context"
        _ "embed" // Calls init function.
        "fmt"
        "log"
    
        "github.com/golang-migrate/migrate/v4"
        "github.com/golang-migrate/migrate/v4/database/postgres"
        "github.com/jmoiron/sqlx"
        "github.com/jonleopard/bootstrap/pkg/sys/database"
        _ "github.com/lib/pq"
    )
    
    var (
        //go:embed sql/000001_schema.up.sql
        schemaDoc string
    
        //go:embed sql/seed.sql
        seedDoc string
    )
    
    // Migrate attempts to bring the schema for db up to date with the migrations
    // defined in this package.
    func Migrate(ctx context.Context, db *sqlx.DB) error {
        if err := database.StatusCheck(ctx, db); err != nil {
            return fmt.Errorf("status check database: %w", err)
        }
    
        driver, err := postgres.WithInstance(db.DB, &postgres.Config{})
        if err != nil {
            return fmt.Errorf("Construct Migrate driver: %w", err)
        }
    
        m, err := migrate.NewWithDatabaseInstance(schemaDoc, "postgres", driver)
        if err != nil {
            log.Fatal(err)
        }
    
        return m.Up()
    }

- Jon Leopard

Answer 1

NewWithDatabaseInstance的定义为：

func NewWithDatabaseInstance(sourceURL string, databaseName string, databaseInstance database.Driver) (*Migrate, error)

所以第一个参数是一个URL，你正在传递脚本本身。 NewWithDatabaseInstance 调用 SchemeFromURL ,这就是生成你所看到的错误（因为你传递的URL不包含:）。 一个URL由“scheme”后面跟着:,再加上其他信息组成。

要使用golang-migrate配合embed，请查看文档中的示例：

//go:embed testdata/migrations/*.sql
var fs embed.FS

func main() {
    d, err := iofs.New(fs, "testdata/migrations")
    if err != nil {
        log.Fatal(err)
    }
    m, err := migrate.NewWithSourceInstance("iofs", d, "postgres://postgres@localhost/postgres?sslmode=disable")
    if err != nil {
        log.Fatal(err)
    }
    err = m.Up()
    if err != nil {
        // ...
    }
    // ...
}

您会注意到这里传递了一个文件夹（作为embed.FS），而不是一个单独的文件。这是因为golang-migrate旨在应用多个迁移，而不仅仅是对数据库运行单个脚本。您应该能够使用类似以下的内容：

//go:embed sql/*.sql
var schemaFs embed.FS
...
d, err := iofs.New(fs, "sql") // Get migrations from sql folder
if err != nil {
    log.Fatal(err)
}
m, err := migrate.NewWithInstance("iofs", d, "postgres", driver)

- Brits

---

这解决了我的问题！不知为何，我在文档中没有注意到那个例子。感谢您抽出时间写下这篇文章。 - Jon Leopard

最后一行应该是 m, err := migrate.NewWithInstance("iofs", d, "postgres", driver)。 - Jeremie Weldin