-
Notifications
You must be signed in to change notification settings - Fork 0
/
loader.go
107 lines (88 loc) · 3.13 KB
/
loader.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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package lsqlite
import (
"context"
"github.com/pkg/errors"
"go.uber.org/zap"
"admini.dev/admini/app/lib/database"
"admini.dev/admini/app/lib/filter"
"admini.dev/admini/app/lib/schema"
"admini.dev/admini/app/lib/schema/model"
"admini.dev/admini/app/loader"
"admini.dev/admini/app/loader/ldb"
"admini.dev/admini/app/loader/lsqlite/sqlite"
"admini.dev/admini/app/result"
"admini.dev/admini/app/util"
)
type Loader struct {
key string
db *database.Service
logger util.Logger
}
func NewLoader(ctx context.Context, logger util.Logger) func(key string, cfg []byte) (loader.Loader, error) {
return func(key string, cfg []byte) (loader.Loader, error) {
log := logger.With(zap.String("service", "loader.sqlite"), zap.String("source", key))
db, err := openDatabase(ctx, key, cfg, log)
if err != nil {
return nil, errors.Wrap(err, "error opening database")
}
return &Loader{key: key, db: db, logger: log}, nil
}
}
var _ loader.Loader = (*Loader)(nil)
func (l *Loader) Schema(ctx context.Context) (*schema.Schema, error) {
return sqlite.LoadDatabaseSchema(ctx, l.db, l.logger)
}
func (l *Loader) Connection(_ context.Context) (any, error) {
return l.db, nil
}
func (l *Loader) List(ctx context.Context, m *model.Model, opts *filter.Options) (*result.Result, error) {
return ldb.List(ctx, l.db, m, opts, l.logger)
}
func (l *Loader) Count(ctx context.Context, m *model.Model) (int, error) {
return ldb.Count(ctx, l.db, m, l.logger)
}
func (l *Loader) Get(ctx context.Context, m *model.Model, ids []any) (*result.Result, error) {
return ldb.Get(ctx, l.db, m, ids, l.logger)
}
func (l *Loader) Query(ctx context.Context, enums model.Models, sql string) (*result.Result, error) {
return ldb.Query(ctx, l.db, sql, enums, l.logger)
}
func (l *Loader) Add(ctx context.Context, m *model.Model, changes util.ValueMap) ([]any, error) {
return ldb.Add(ctx, l.db, m, changes, l.logger)
}
func (l *Loader) Save(ctx context.Context, m *model.Model, ids []any, changes util.ValueMap) ([]any, error) {
return ldb.Save(ctx, l.db, m, ids, changes, l.logger)
}
func (l *Loader) Remove(ctx context.Context, m *model.Model, fields []string, values []any, expected int) (int, error) {
return ldb.Remove(ctx, l.db, m, fields, values, expected, l.logger)
}
func (l *Loader) Default(_ context.Context, m *model.Model) ([]any, error) {
ret := make([]any, 0, len(m.Fields))
for _, f := range m.Fields {
ret = append(ret, f.Default)
}
return ret, nil
}
func LoadConfig(cfg []byte) (*database.SQLiteParams, error) {
params := &database.SQLiteParams{}
err := util.FromJSON(cfg, params)
if err != nil {
return nil, errors.Wrap(err, "error parsing database config")
}
return params, nil
}
func openDatabase(ctx context.Context, key string, cfg []byte, logger util.Logger) (*database.Service, error) {
params, err := LoadConfig(cfg)
if err != nil {
return nil, err
}
db, err := database.OpenSQLiteDatabase(ctx, key, params, logger)
if err != nil {
return nil, errors.Wrap(err, "error opening database")
}
_, err = db.SingleInt(ctx, "select 1 as x", nil, logger)
if err != nil {
return nil, errors.Wrap(err, "error connecting to database")
}
return db, nil
}