package models

import (
	commonModels "vamicro/camera-common/models"
	"vamicro/config"

	"basic.com/valib/logger.git"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
)

var db *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/camera-service.db")

	if err != nil {
		logger.Debug("db open error ", err)
	}
	db.LogMode(true)
	//db.SetLogger(&DbLogger{})
	// err1 := db.Debug().AutoMigrate(&commonModels.Area{}, &commonModels.Camera{}, &commonModels.CameraArea{},
	// 	&commonModels.Sensor{}, &commonModels.CameraSensor{}, &commonModels.TrainTags{})
	err1 := db.Debug().AutoMigrate(&commonModels.TrainTags{})
	if err1 != nil {
		logger.Debug("自动迁移失败 ", err1)
	}

	commonModels.SetDB(db)
}

// GetDB ...
func GetDB() *gorm.DB {
	return db
}

func CloseDB() {
	db.Close()
}