kloud/pkg/DB/init.go at main · 00LT00/kloud

Repository navigation

package DB

import (

"gorm.io/driver/mysql"

"gorm.io/gorm"

"kloud/model"

"kloud/pkg/conf"

"log"

)

var db = new(gorm.DB)

func init() {

var err error

db, err = gorm.Open(mysql.Open(conf.GetConf().DB.Dsn()))

if err != nil {

panic(err)

}

if conf.GetConf().Mode == conf.Debug {

db = db.Debug()

}

initTable()

}

func initTable() {

err := db.AutoMigrate(model.GetModels()...)

if err != nil {

panic(err)

}

func GetDB() *gorm.DB {

return db

}

func Ping() {

sql, _ := db.DB()

err := sql.Ping()

if err != nil {

log.Println(err)

} else {

log.Println("connect to DB")

}