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
31 func Init(config DBConfig, redisConfig RedisConfig) {
38 log.Fatalf("cannot migratedb '%v'\n", err)
41 Redis = redis.NewClient(&redis.Options{
42 Network: redisConfig.Type,
43 Addr: redisConfig.Address,
44 Password: redisConfig.Password,
48 _, err = Redis.Ping().Result()
51 log.Fatalf("redis init error %s", err)
56 func migratedb() error {
57 /* Remove after first MEP */
58 version, err := migrate.Version(DB)
64 return migrate.SetVersion(DB, 1)
68 mig := make([]migrate.Migration, 0)
70 for _, migration := range migrations {
71 mig = append(mig, migrate.Migration{
72 Version: migration.Version,
73 Up: func(db orm.DB) error {
74 for _, query := range migration.Up {
75 _, err := db.Exec(query)
83 Down: func(db orm.DB) error {
84 for _, query := range migration.Down {
85 _, err := db.Exec(query)
96 oldVersion, newVersion, err := migrate.RunMigrations(DB, mig, "up")
98 if oldVersion != newVersion {
99 log.Infof("Migrate DB: %v -> %v", oldVersion, newVersion)
101 log.Infof("DB up-to-date: version '%v'", newVersion)
106 func connect(config DBConfig) *pg.DB {
107 return pg.Connect(&pg.Options{
109 Password: config.Password,
110 Database: config.Database,
111 Addr: config.Address,