4 migrate "github.com/go-pg/migrations"
6 "github.com/go-pg/pg/orm"
7 "github.com/go-redis/redis"
8 "github.com/jloup/utils"
11 var log = utils.StandardL().WithField("module", "db")
18 type DBConfig struct {
25 type RedisConfig struct {
26 Type string // tcp or unix
32 func Init(config DBConfig, redisConfig RedisConfig) {
39 log.Fatalf("cannot migratedb '%v'\n", err)
42 Redis = redis.NewClient(&redis.Options{
43 Network: redisConfig.Type,
44 Addr: redisConfig.Address,
45 Password: redisConfig.Password,
46 DB: redisConfig.Database,
49 _, err = Redis.Ping().Result()
52 log.Fatalf("redis init error %s", err)
57 func migratedb() error {
58 /* Remove after first MEP */
59 version, err := migrate.Version(DB)
65 return migrate.SetVersion(DB, 1)
69 mig := make([]migrate.Migration, 0)
71 for _, migration := range migrations {
72 mig = append(mig, migrate.Migration{
73 Version: migration.Version,
74 Up: func(db orm.DB) error {
75 for _, query := range migration.Up {
76 _, err := db.Exec(query)
84 Down: func(db orm.DB) error {
85 for _, query := range migration.Down {
86 _, err := db.Exec(query)
97 oldVersion, newVersion, err := migrate.RunMigrations(DB, mig, "up")
99 if oldVersion != newVersion {
100 log.Infof("Migrate DB: %v -> %v", oldVersion, newVersion)
102 log.Infof("DB up-to-date: version '%v'", newVersion)
107 func connect(config DBConfig) *pg.DB {
108 return pg.Connect(&pg.Options{
110 Password: config.Password,
111 Database: config.Database,
112 Addr: config.Address,