zhangzengfei
2023-08-17 dc40143800edd914d0f721b06f90c8b4d307b979
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package kingdee
 
import (
    "fmt"
 
    "kingdee-dbapi/logger"
 
    //"gorm.io/driver/sqlserver"
    //"gorm.io/gorm"
 
    "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)
 
    logger.Debug("sqlServer地址:%s", sqlServer)
    // 打开数据库连接
    db, err = gorm.Open("mssql", sqlServer)
    // gorm 2.0
    //db, err = gorm.Open(sqlserver.Open(sqlServer), &gorm.Config{})
    if err != nil {
        return err
    }
 
    return nil
}
 
func GetDB() *gorm.DB {
    return db
}
 
// 2.0 CloseDB .
//func CloseDB() {
//    sqlDB, err := db.DB()
//    if err != nil {
//        return
//    }
//
//    sqlDB.Close()
//}
 
func CloseDB() {
    db.Close()
}