forked from temporalio/temporal
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathstore.go
More file actions
95 lines (85 loc) · 2.34 KB
/
store.go
File metadata and controls
95 lines (85 loc) · 2.34 KB
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
package sql
import (
"errors"
"fmt"
"slices"
"go.temporal.io/server/common/config"
"go.temporal.io/server/common/log"
"go.temporal.io/server/common/metrics"
"go.temporal.io/server/common/persistence/sql/sqlplugin"
"go.temporal.io/server/common/resolver"
expmaps "golang.org/x/exp/maps"
)
var ErrPluginNotSupported = errors.New("plugin not supported")
var supportedPlugins = map[string]sqlplugin.Plugin{}
// RegisterPlugin will register a SQL plugin
func RegisterPlugin(pluginName string, plugin sqlplugin.Plugin) {
if _, ok := supportedPlugins[pluginName]; ok {
panic("plugin " + pluginName + " already registered")
}
supportedPlugins[pluginName] = plugin
}
// NewSQLDB creates a returns a reference to a logical connection to the
// underlying SQL database. The returned object is tied to a single
// SQL database and the object can be used to perform CRUD operations on
// the tables in the database.
func NewSQLDB(
dbKind sqlplugin.DbKind,
cfg *config.SQL,
r resolver.ServiceResolver,
logger log.Logger,
mh metrics.Handler,
) (sqlplugin.DB, error) {
return createDB[sqlplugin.DB](dbKind, cfg, r, logger, mh)
}
// NewSQLAdminDB returns a AdminDB.
func NewSQLAdminDB(
dbKind sqlplugin.DbKind,
cfg *config.SQL,
r resolver.ServiceResolver,
logger log.Logger,
mh metrics.Handler,
) (sqlplugin.AdminDB, error) {
return createDB[sqlplugin.AdminDB](dbKind, cfg, r, logger, mh)
}
func createDB[T any](
dbKind sqlplugin.DbKind,
cfg *config.SQL,
r resolver.ServiceResolver,
logger log.Logger,
mh metrics.Handler,
) (T, error) {
var res T
plugin, err := getPlugin(cfg.PluginName)
if err != nil {
return res, err
}
db, err := plugin.CreateDB(dbKind, cfg, r, logger, mh)
if err != nil {
return res, err
}
//revive:disable-next-line:unchecked-type-assertion
res = db.(T)
return res, err
}
func getPlugin(pluginName string) (sqlplugin.Plugin, error) {
plugin, ok := supportedPlugins[pluginName]
if !ok {
keys := expmaps.Keys(supportedPlugins)
slices.Sort(keys)
return nil, fmt.Errorf(
"%w: unknown plugin %q, supported plugins: %v",
ErrPluginNotSupported,
pluginName,
keys,
)
}
return plugin, nil
}
func GetPluginVisibilityQueryConverter(pluginName string) (sqlplugin.VisibilityQueryConverter, error) {
plugin, err := getPlugin(pluginName)
if err != nil {
return nil, err
}
return plugin.GetVisibilityQueryConverter(), nil
}