package models

import (
	"basic.com/valib/logger.git"
	"github.com/jinzhu/gorm"
)

var db *gorm.DB


func Init() {
	var err error
	db, err = gorm.Open("sqlite3", "./version.db")
	if err != nil {
		logger.Error("Db Init failed", err)
		return;
	} else {
		logger.Info("Db Init ok", err)
	}

	db.LogMode(true)
	db.AutoMigrate(&BeeNode{})
}

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

func CloseDB() {
	_ = db.Close()
}