package models
|
|
import (
|
"vamicro/config"
|
|
"basic.com/valib/logger.git"
|
"github.com/jinzhu/gorm"
|
_ "github.com/jinzhu/gorm/dialects/sqlite"
|
)
|
|
var db *gorm.DB
|
var db2 *gorm.DB
|
var err error
|
|
// Init creates a connection to mysql database and
|
// migrates any new models
|
func Init() {
|
db, err = gorm.Open(config.DBconf.Name, "../config/chanmanage-service.db")
|
if err != nil {
|
logger.Debug("db open error ", err)
|
}
|
db.LogMode(true)
|
//db.SetLogger(&DbLogger{})
|
db.AutoMigrate(&PollConfig{})
|
|
db2, err = gorm.Open(config.DBconf.Name, "../config/camera-service.db")
|
if err != nil {
|
logger.Debug("db open error ", err)
|
}
|
db2.LogMode(true)
|
db2.AutoMigrate(&PollConfig{})
|
|
}
|
|
// GetDB ...
|
func GetDB() *gorm.DB {
|
return db
|
}
|
|
func CloseDB() {
|
db.Close()
|
}
|
