store.go

package main

import (
	"time"

	"github.com/go-redis/redis"
	"github.com/jinzhu/gorm"
	"github.com/sirupsen/logrus"
)

func initDB() {
	var err error
	// mysql conn
	for {
		db, err = gorm.Open("mysql", config.DB.User+":"+config.DB.Password+
			"@tcp("+config.DB.Host+":"+config.DB.Port+")/"+config.DB.Name+
			"?charset=utf8mb4&parseTime=True&loc=Local&timeout=90s")
		if err != nil {
			logrus.Warnf("waiting to connect to db: %s", err.Error())
			time.Sleep(time.Second * 2)
			continue
		}
		logrus.Info("Mysql connect successful.")
		break
	}

	// gorm debug log
	if config.APP.Debug {
		db.LogMode(true)
	}
}

// createTable gorm auto migrate tables
func createTables() {
	db.AutoMigrate(&Note{})
}

func initRedis() {
	// redis conn
	rdb = redis.NewClient(&redis.Options{
		Addr:     config.Redis.Host + ":" + config.Redis.Port,
		Password: config.Redis.Password,
		DB:       config.Redis.DB,
	})
	logrus.Info("Redis connect successful.")
}