From b03f12507c1c53b3599bece3ded0ad39753e65fd Mon Sep 17 00:00:00 2001 From: Alexander Kavon Date: Tue, 23 Jan 2024 06:33:41 -0500 Subject: seed package to seed database on the fly --- seed/main.go | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 seed/main.go (limited to 'seed/main.go') diff --git a/seed/main.go b/seed/main.go new file mode 100644 index 0000000..a2cc084 --- /dev/null +++ b/seed/main.go @@ -0,0 +1,35 @@ +package main + +import ( + "log" + "os" + "path/filepath" + + "gitlab.com/alexkavon/newsstand/seed/seeder" + "gitlab.com/alexkavon/newsstand/src/conf" + "gitlab.com/alexkavon/newsstand/src/db" +) + +func main() { + err := os.Setenv("NEWSSTAND_CONFIG_PATH", "./.seedrc.toml") + if err != nil { + log.Fatal(err) + } + config := conf.NewConf() + database := db.NewDb(config) + defer database.Close() + // load seeder + s := seeder.NewSeeder(database) + // migrate + migrations := filepath.Join(config.GetCwd(), "/migrations") + err = seeder.MigrateDatabase(database, migrations) + if err != nil { + log.Fatal(err) + } + + err = s.SeedUsers(10) + if err != nil { + log.Fatal(err) + } + os.Exit(0) +} -- cgit v1.2.3