在使用 Go Buffalo 框架的开发环境中如何通过 TOML 文件配置数据库种子？

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《在使用 Go Buffalo 框架的开发环境中如何通过 TOML 文件配置数据库种子？》，聊聊，希望可以帮助到正在努力赚钱的你。

我正在尝试使用 TOML 固定装置在 Go Buffalo 框架中植入我的开发数据库。但是，我无法找到有关如何执行此操作的清晰示例或文档。

正确答案

要为数据库添加种子，您可以使用 grifts。创建新应用程序时，应该在 grifts/db.go 处生成一个虚拟占位符 grift，如下所示：

package grifts

import "github.com/gobuffalo/grift/grift"

var _ = grift.namespace("db", func() {

    grift.desc("seed", "seeds a database")
    grift.add("seed", func(c *grift.context) error {
        // add db seeding stuff here
        return nil
    })

})

我尝试在 grift 中使用 model.loadfixture("seed widgets") ，但有一些恐慌，因为它不在测试环境中。我认为不直接支持从 toml 装置进行数据库播种，但这将是一个有用的功能。然而，查看 loadfixture 中的代码，我们可以构建我们的自己的夹具加载：

假设您有一个名为 name = "seed widgets" 的场景的固定装置：

package grifts

import (
    "fmt"
    "os"
    "strings"

    "github.com/gobuffalo/grift/grift"
    "github.com/gobuffalo/suite/v4"
    "github.com/gobuffalo/suite/v4/fix"
)

var _ = grift.Namespace("db", func() {

    grift.Desc("seed", "Seeds a database")
    grift.Add("seed", func(c *grift.Context) error {
        // The DB connection will connect to the environment set in `GO_ENV` (defaults to `test`)
        // Set this environment variable in your `.env` file to `development`

        // NOTE: it may be better to put seed fixtures in a different directory
        //       to seperate concerns
        model, err := suite.NewModelWithFixtures(os.DirFS("./fixtures"))
        if err != nil {
            return err
        }

        sc, err := fix.Find("seed widgets")
        if err != nil {
            return err
        }

        for _, table := range sc.Tables {
            for _, row := range table.Row {
                q := "insert into " + table.Name
                keys := []string{}
                skeys := []string{}
                for k := range row {
                    keys = append(keys, k)
                    skeys = append(skeys, ":"+k)
                }

                q = q + fmt.Sprintf(" (%s) values (%s)", strings.Join(keys, ","), strings.Join(skeys, ","))
                if _, err = model.DB.Store.NamedExec(q, row); err != nil {
                    return err
                }
            }
        }

        return nil
    })

})

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