initial commit

zhangzengfei

2023-09-04 e8e536d1cb52d2126c8c7ce2ba1c7a76f7208678

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
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()
}