/
db.go
61 lines (52 loc) · 1.14 KB
/
db.go
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
54
55
56
57
58
59
60
61
package simple
import (
"github.com/dgraph-io/ristretto"
"github.com/dollarkillerx/go-proxy-manager/internal/app/backend/pkg/models"
"github.com/dollarkillerx/go-proxy-manager/internal/utils"
"github.com/glebarez/sqlite"
"github.com/google/uuid"
"gorm.io/gorm"
"log"
)
type Simple struct {
cache *ristretto.Cache
db *gorm.DB
}
func NewSimple() (*Simple, error) {
cache, err := utils.NewRCache()
if err != nil {
return nil, err
}
db, err := gorm.Open(sqlite.Open("gorm.db"), &gorm.Config{})
if err != nil {
return nil, err
}
db.AutoMigrate(
&models.UserCenter{},
&models.Tasks{},
)
var userCount int64
err = db.Model(&models.UserCenter{}).Count(&userCount).Error
if err != nil {
log.Println(err)
}
if err == nil && userCount == 0 {
sale := uuid.New().String()
err = db.Model(&models.UserCenter{}).Create(&models.UserCenter{
BasicModel: models.BasicModel{
ID: uuid.New().String(),
},
Account: "manager@manager.com",
Password: utils.GenPassword("admin", sale),
Sale: sale,
}).Error
if err != nil {
log.Fatalln(err)
}
}
sim := Simple{
cache: cache,
db: db,
}
return &sim, nil
}