package models

import (
	"github.com/golang/glog"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
	"webserver/extend/config"
)

var db *gorm.DB
var err error

// Init creates a connection to mysql database and
// migrates any new models
func Init() {
	// dbname, file path
	db, err = gorm.Open(config.DBconf.Name, config.DBconf.FilePath)
	db.LogMode(true)
	db.Exec("PRAGMA foreign_keys = ON")
	if err != nil {
		glog.Error("db open error ", err)
	}
	db.AutoMigrate(&Area{}, &Camera{}, &LocalConfig{}, &VSSLocalSettingTbl{}, &Task{}, &CameraTask{}, &TaskSdk{}, &Sdk{})
}

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

func CloseDB() {
	db.Close()
}