/
db.go
75 lines (62 loc) · 1.71 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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package db
import (
"context"
"errors"
"fmt"
"path/filepath"
"time"
"go.uber.org/zap"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"gorm.io/gorm/logger"
"github.com/tranHieuDev23/cato/internal/configs"
"github.com/tranHieuDev23/cato/internal/utils"
)
type gormZapLogger struct {
logger *zap.Logger
}
func (la gormZapLogger) LogMode(_ logger.LogLevel) logger.Interface {
return la
}
func (la gormZapLogger) Info(ctx context.Context, s string, args ...interface{}) {
utils.LoggerWithContext(ctx, la.logger).Info(s, zap.Any("args", args))
}
func (la gormZapLogger) Warn(ctx context.Context, s string, args ...interface{}) {
utils.LoggerWithContext(ctx, la.logger).Warn(s, zap.Any("args", args))
}
func (la gormZapLogger) Error(ctx context.Context, s string, args ...interface{}) {
utils.LoggerWithContext(ctx, la.logger).Error(s, zap.Any("args", args))
}
func (la gormZapLogger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) {
sql, rows := fc()
utils.LoggerWithContext(ctx, la.logger).Debug("SQL",
zap.String("sql", sql),
zap.Duration("elapsed", time.Since(begin)),
zap.Int64("rows", rows),
zap.Error(err),
)
}
func InitializeDB(
logger *zap.Logger,
databaseConfig configs.Database,
) (*gorm.DB, error) {
databaseFilePath, err := filepath.Abs(databaseConfig.FilePath)
if err != nil {
return nil, errors.New("failed to get abs database file path")
}
db, err := gorm.Open(
sqlite.Open(fmt.Sprintf("file://%s", databaseFilePath)),
&gorm.Config{
Logger: gormZapLogger{logger: logger},
},
)
if err != nil {
return nil, err
}
migrator := NewMigrator(db, logger)
err = migrator.Migrate(context.Background())
if err != nil {
return nil, err
}
return db, nil
}