yanghui
2021-06-28 b568d251606f82968b8103c485a0d70027c1d57e
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
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()
}