diff options
Diffstat (limited to 'golang/cmd')
-rw-r--r-- | golang/cmd/fatcatd/reinit.go | 42 | ||||
-rw-r--r-- | golang/cmd/fatcatd/root.go | 2 |
2 files changed, 43 insertions, 1 deletions
diff --git a/golang/cmd/fatcatd/reinit.go b/golang/cmd/fatcatd/reinit.go new file mode 100644 index 00000000..b7cfa468 --- /dev/null +++ b/golang/cmd/fatcatd/reinit.go @@ -0,0 +1,42 @@ + +package main + +import ( + "github.com/spf13/viper" + log "github.com/sirupsen/logrus" + "github.com/go-pg/pg" + "github.com/spf13/cobra" + "github.com/gobuffalo/packr" +) + +var reinitCmd = &cobra.Command{ + Use: "reinit [options]", + Short: "Reset database schema", + Run: func(cmd *cobra.Command, args[] string) { + main_reinit() + }, +} + +func main_reinit() { + + box := packr.NewBox("../../sql") + sql_schema, err := box.MustString("fatcat-schema.sql") + if err != nil { + log.Panicf("finding SQL file: {}", err) + } + + db_options, err := pg.ParseURL(viper.GetString("db_url")) + if err != nil { + log.Panicf("parsing DB string: {}", err) + } + db := pg.Connect(db_options) + defer db.Close() + + log.Info("Starting load...") + _, err = db.Exec(sql_schema) + if err != nil { + log.Fatalf("Error loading SQL: {}", err) + } + log.Info("Success!") + +} diff --git a/golang/cmd/fatcatd/root.go b/golang/cmd/fatcatd/root.go index b5263b06..12be35b3 100644 --- a/golang/cmd/fatcatd/root.go +++ b/golang/cmd/fatcatd/root.go @@ -29,12 +29,12 @@ func init() { serveCmd.Flags().String("db-url", "postgres://", "database connecion string") rootCmd.AddCommand(serveCmd) + rootCmd.AddCommand(reinitCmd) } func initConfig() { viper.SetDefault("port", 9411) - viper.SetDefault("db_url", "postgres://bnewbold@localhost/fatcat") viper.SetEnvPrefix("FATCAT") viper.AutomaticEnv() |