-
Notifications
You must be signed in to change notification settings - Fork 0
/
data.go
86 lines (72 loc) · 1.88 KB
/
data.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
76
77
78
79
80
81
82
83
84
85
86
package infrastructure
import (
"context"
"github.com/go-kratos/kratos/v2/log"
"github.com/google/wire"
"github.com/richingm/knowledge/internal/biz"
"github.com/richingm/knowledge/internal/conf"
"github.com/richingm/knowledge/migrations"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"time"
)
type Data struct {
db *gorm.DB
}
// 用来承载事务的上下文
type contextTxKey struct{}
// NewTransaction .
func NewTransaction(d *Data) biz.Transaction {
return d
}
// ExecTx gorm Transaction
func (d *Data) ExecTx(ctx context.Context, fn func(ctx context.Context) error) error {
return d.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
ctx = context.WithValue(ctx, contextTxKey{}, tx)
return fn(ctx)
})
}
// DB 根据此方法来判断当前的 db 是不是使用 事务的 DB
func (d *Data) DB(ctx context.Context) *gorm.DB {
tx, ok := ctx.Value(contextTxKey{}).(*gorm.DB)
if ok {
return tx
}
return d.db
}
func NewGormDB(c *conf.Data, logger log.Logger) (*gorm.DB, error) {
dsn := c.Database.Source
// 创建GORM日志适配器
gormLogger := NewGormLogAdapter(logger)
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{Logger: gormLogger})
if err != nil {
log.NewHelper(logger).Info("closing the data resources")
return nil, err
}
sqlDB, err := db.DB()
if err != nil {
return nil, err
}
sqlDB.SetMaxIdleConns(50)
sqlDB.SetMaxOpenConns(150)
sqlDB.SetConnMaxLifetime(time.Second * 25)
return db, err
}
func NewData(c *conf.Data, logger log.Logger) (*Data, func(), error) {
cleanup := func() {
log.NewHelper(logger).Info("closing the data resources")
}
d := &Data{}
db, err := NewGormDB(c, logger)
if err != nil {
return d, cleanup, err
}
d.db = db
var ctx context.Context
err = migrations.InitMigrate(ctx, db)
if err != nil {
return d, cleanup, nil
}
return d, cleanup, nil
}
var ProviderSet = wire.NewSet(NewData, NewGormDB, NewKnowledgeRepo)