-
Notifications
You must be signed in to change notification settings - Fork 1
/
storage.go
77 lines (61 loc) · 1.49 KB
/
storage.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
package cockroachdb
import (
"context"
_ "embed"
"log/slog"
"github.com/jackc/pgx/v5/pgxpool"
"github.com/adlerhurst/eventstore/v2"
"github.com/adlerhurst/eventstore/v2/x"
)
type Config struct {
Pool *pgxpool.Pool
}
var (
_ eventstore.Eventstore = (*CockroachDB)(nil)
logger = slog.Default()
eventPool = x.NewPool[event]()
commandPool = x.NewPool[command]()
)
type CockroachDB struct {
client *pgxpool.Pool
pushAppName string
filterAppName string
}
func New(config *Config, opts ...storageOpt) *CockroachDB {
store := &CockroachDB{
client: config.Pool,
pushAppName: "es_push",
filterAppName: "es_filter",
}
for _, opt := range opts {
opt(store)
}
return store
}
type storageOpt func(*CockroachDB)
func WithLogger(l *slog.Logger) storageOpt {
return func(store *CockroachDB) {
logger = l
}
}
func WithPushAppName(name string) storageOpt {
return func(store *CockroachDB) {
store.pushAppName = name
}
}
func WithFilterAppName(name string) storageOpt {
return func(store *CockroachDB) {
store.filterAppName = name
}
}
//go:embed 0_setup.sql
var setupStmt string
func (store *CockroachDB) Setup(ctx context.Context) error {
_, err := store.client.Exec(ctx, setupStmt)
logger.ErrorContext(ctx, "setup failed", "cause", err)
return err
}
// Ready implements [eventstore.Eventstore]
func (store *CockroachDB) Ready(ctx context.Context) error {
return store.client.Ping(ctx)
}