70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"game_demo/pkg/logging"
|
|
"game_demo/pkg/setting"
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/postgres"
|
|
"time"
|
|
)
|
|
|
|
var db = InitDbConnect()
|
|
|
|
type Model struct {
|
|
ID int `gorm:"primary_key" json:"id"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
DeletedAt *time.Time `sql:"index" json:"deleted_at"`
|
|
}
|
|
|
|
func init() {
|
|
db.LogMode(true)
|
|
|
|
db.SingularTable(true)
|
|
db.DB().SetMaxIdleConns(10)
|
|
db.DB().SetMaxOpenConns(100)
|
|
|
|
DbMigrate()
|
|
//CloseDB()
|
|
}
|
|
|
|
func InitDbConnect() *gorm.DB {
|
|
var (
|
|
err error
|
|
dbType, dbName, user, password, host, port, tablePrefix string
|
|
)
|
|
|
|
sec, err := setting.Cfg.GetSection("database")
|
|
if err != nil {
|
|
logging.Info(2, "Fail to get section 'database': %v", err)
|
|
}
|
|
|
|
dbType = sec.Key("TYPE").String()
|
|
dbName = sec.Key("NAME").String()
|
|
user = sec.Key("USER").String()
|
|
password = sec.Key("PASSWORD").String()
|
|
host = sec.Key("HOST").String()
|
|
port = sec.Key("PORT").String()
|
|
tablePrefix = sec.Key("TABLE_PREFIX").String()
|
|
|
|
db, err := gorm.Open(dbType, fmt.Sprintf("host=%s user=%s dbname=%s port=%s sslmode=disable password=%s", host, user, dbName, port, password))
|
|
|
|
if err != nil {
|
|
logging.Info(err)
|
|
}
|
|
|
|
gorm.DefaultTableNameHandler = func(db *gorm.DB, defaultTableName string) string {
|
|
return tablePrefix + defaultTableName
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
func DbMigrate() {
|
|
db.AutoMigrate(&Game{}, &Auth{})
|
|
}
|
|
|
|
func CloseDB() {
|
|
defer db.Close()
|
|
} |