1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
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: %v", err)
}
db_options, err := pg.ParseURL(viper.GetString("db_url"))
if err != nil {
log.Panicf("parsing DB string: %v", 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: %v", err)
}
log.Info("Loading dummy data...")
sql_dummy, err := box.MustString("dummy-data.sql")
if err != nil {
log.Panicf("finding SQL file: %v", err)
}
_, err = db.Exec(sql_dummy)
if err != nil {
log.Fatalf("Error loading SQL: %v", err)
}
log.Info("Success!")
}
|
