package models

import (
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
	"vamicro/config"
	"fmt"
)

var db *gorm.DB
var oauth2DB *gorm.DB


// Init creates a connection to mysql database and
// migrates any new models
func Init() {
	var err error
	db, err = gorm.Open(config.DBconf.Name, "../config/op-log-sets.db")
	if err != nil {
		fmt.Println("Db Init failed", err)
		return
	} else {
		fmt.Println("Db Init ok", err)
	}


	db.LogMode(true)

	oauth2DB, err = gorm.Open(config.DBconf.Name, "../config/oauth2.db")
	if err != nil {
		fmt.Println("oauth2DB Init failed", err)
		return
	} else {
		fmt.Println("oauth2DB Init ok", err)
	}
	oauth2DB.LogMode(true)
	oauth2DB.AutoMigrate(&Oauth2Client{})

	//db.SetLogger(&DbLogger{})
	//db.AutoMigrate(&Operations{},&Modules{})
}

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

func CloseDB() {
	db.Close()
}