C0lla commited on
Commit
e4e7048
1 Parent(s): d09ec67

Delete database

Browse files
Files changed (1) hide show
  1. database/database.go +0 -97
database/database.go DELETED
@@ -1,97 +0,0 @@
1
- package database
2
-
3
- import (
4
- "embed"
5
- "log"
6
- "os"
7
- "path/filepath"
8
- "time"
9
-
10
- "github.com/divyam234/teldrive/utils"
11
- "github.com/divyam234/teldrive/utils/kv"
12
- "github.com/pressly/goose/v3"
13
- "go.etcd.io/bbolt"
14
- "gorm.io/driver/postgres"
15
- "gorm.io/gorm"
16
- "gorm.io/gorm/logger"
17
- "gorm.io/gorm/schema"
18
- )
19
-
20
- //go:embed migrations/*.sql
21
- var embedMigrations embed.FS
22
- var DB *gorm.DB
23
- var BoltDB *bbolt.DB
24
- var KV kv.KV
25
-
26
- func InitDB() {
27
-
28
- var err error
29
-
30
- newLogger := logger.New(
31
- log.New(os.Stdout, "\r\n", log.LstdFlags),
32
- logger.Config{
33
- SlowThreshold: time.Second,
34
- LogLevel: logger.Silent,
35
- IgnoreRecordNotFoundError: true,
36
- ParameterizedQueries: true,
37
- Colorful: false,
38
- },
39
- )
40
-
41
- DB, err = gorm.Open(postgres.Open(utils.GetConfig().DatabaseUrl), &gorm.Config{
42
- NamingStrategy: schema.NamingStrategy{
43
- TablePrefix: "teldrive.",
44
- SingularTable: false,
45
- },
46
- PrepareStmt: false,
47
- NowFunc: func() time.Time {
48
- return time.Now().UTC()
49
- },
50
- Logger: newLogger,
51
- })
52
- if err != nil {
53
- panic(err)
54
- }
55
-
56
- sqlDB, err := DB.DB()
57
- if err != nil {
58
- panic(err)
59
- }
60
- sqlDB.SetMaxIdleConns(10)
61
- sqlDB.SetMaxOpenConns(100)
62
-
63
- sqlDB.SetConnMaxLifetime(time.Hour)
64
- go func() {
65
- DB.Exec(`select 1`)
66
- if utils.GetConfig().RunMigrations {
67
- migrate()
68
- }
69
- }()
70
-
71
- config := utils.GetConfig()
72
- BoltDB, err = bbolt.Open(filepath.Join(config.ExecDir, "teldrive.db"), 0666, &bbolt.Options{
73
- Timeout: time.Second,
74
- NoGrowSync: false,
75
- })
76
- if err != nil {
77
- panic(err)
78
- }
79
- KV, err = kv.New(kv.Options{Bucket: "teldrive", DB: BoltDB})
80
-
81
- if err != nil {
82
- panic(err)
83
- }
84
- }
85
-
86
- func migrate() {
87
-
88
- goose.SetBaseFS(embedMigrations)
89
-
90
- if err := goose.SetDialect("postgres"); err != nil {
91
- panic(err)
92
- }
93
- db, _ := DB.DB()
94
- if err := goose.Up(db, "migrations"); err != nil {
95
- panic(err)
96
- }
97
- }