package kingdee

import (
	"fmt"
	"github.com/jinzhu/gorm"

	_ "github.com/jinzhu/gorm/dialects/mssql"
)

var db *gorm.DB
var err error

// Init .
func Init(username, password, addr, dbName string) error {
	var err error

	sqlServer := fmt.Sprintf("sqlserver://%s:%s@%s:1433?database=%s;encrypt=disable;",
		username, password, addr, dbName)

	// 打开数据库连接
	db, err = gorm.Open("mssql", sqlServer)
	if err != nil {
		return err
	}

	return nil
}

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

// CloseDB .
func CloseDB() {
	if db != nil {
		db.Close()
	}
}