[ADD]first commit
This commit is contained in:
60
models/models.go
Normal file
60
models/models.go
Normal file
@@ -0,0 +1,60 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/go-pripro/shop/pkg/setting"
|
||||
"github.com/jinzhu/gorm"
|
||||
_ "github.com/jinzhu/gorm/dialects/mysql"
|
||||
"log"
|
||||
)
|
||||
|
||||
var db *gorm.DB
|
||||
|
||||
type Model struct {
|
||||
ID int `gorm:"primary_key" json:"id"`
|
||||
CreatedOn int `json:"created_on"`
|
||||
ModifiedOn int `json:"modified_on"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
var (
|
||||
err error
|
||||
dbType, dbName, user, password, host, tablePrefix string
|
||||
)
|
||||
|
||||
sec, err := setting.Cfg.GetSection("database")
|
||||
if err != nil {
|
||||
log.Fatal(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()
|
||||
tablePrefix = sec.Key("TABLE_PREFIX").String()
|
||||
|
||||
db, err = gorm.Open(dbType, fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=True&loc=Local",
|
||||
user,
|
||||
password,
|
||||
host,
|
||||
dbName))
|
||||
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
gorm.DefaultTableNameHandler = func(db *gorm.DB, defaultTableName string) string {
|
||||
return tablePrefix + defaultTableName
|
||||
}
|
||||
|
||||
//db.AutoMigrate(&Tag{}, &Article{}, &Auth{})
|
||||
//db.SingularTable(true)
|
||||
db.LogMode(true)
|
||||
db.DB().SetMaxIdleConns(10)
|
||||
db.DB().SetMaxOpenConns(100)
|
||||
}
|
||||
|
||||
func CloseDB() {
|
||||
defer db.Close()
|
||||
}
|
||||
Reference in New Issue
Block a user