package models

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

var db *gorm.DB
var err error

// Init .
func Init() error {
	// 打开数据库连接
	db, err = gorm.Open("sqlite3", "kingdee-api.db")
	if err != nil {
		return err
	}

	db.AutoMigrate(&Order{})

	return nil
}

// CloseDB .
func CloseDB() {
	db.Close()
}