zhangzengfei
2024-10-20 b916f7c68a3e16413eac4be13f0404267561f90c
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
package models
 
import (
    "gat1400Exchange/pkg/logger"
 
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)
 
var db *gorm.DB
 
// Init creates a connection to mysql database and
// migrates any new models
func Init() error {
    var err error
 
    db, err = gorm.Open(sqlite.Open("./gat1400.db"), &gorm.Config{})
 
    if err != nil {
        logger.Debug("db open error ", err)
        return err
    }
    _ = db.AutoMigrate(
        &Ape{},
        &Cache{},
        &Cascade{},
        &Positions{},
        &SubPlatform{},
        &Subscribe{},
    )
 
    // 添加默认数据
    InitData()
 
    return nil
}
 
func GetDB() *gorm.DB {
    return db
}
 
func CloseDB() {
 
}
 
func InitData() {
 
}