-
-
Notifications
You must be signed in to change notification settings - Fork 10
/
tool.go
91 lines (75 loc) · 2.15 KB
/
tool.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
package mongo
import (
"fmt"
"github.com/urfave/cli"
"github.com/tidepool-org/platform/application"
"github.com/tidepool-org/platform/errors"
"github.com/tidepool-org/platform/pointer"
storeStructuredMongo "github.com/tidepool-org/platform/store/structured/mongo"
"github.com/tidepool-org/platform/tool"
)
const (
AddressesFlag = "addresses"
TLSFlag = "tls"
)
type Tool struct {
*tool.Tool
mongoConfig *storeStructuredMongo.Config
}
func NewTool() *Tool {
return &Tool{
Tool: tool.New(),
mongoConfig: storeStructuredMongo.NewConfig(),
}
}
func (t *Tool) Initialize(provider application.Provider) error {
if err := t.Tool.Initialize(provider); err != nil {
return err
}
if err := t.mongoConfig.Load(t.ConfigReporter().WithScopes("store")); err != nil {
return errors.Wrap(err, "unable to load store config")
}
t.CLI().Flags = append(t.CLI().Flags,
cli.StringFlag{
Name: fmt.Sprintf("%s,%s", AddressesFlag, "a"),
Usage: "comma-delimited list of address(es) to mongo database (host:port)",
},
cli.BoolFlag{
Name: fmt.Sprintf("%s,%s", TLSFlag, "t"),
Usage: "use TLS to connect to mongo database",
},
)
return nil
}
func (t *Tool) Terminate() {
t.mongoConfig = nil
t.Tool.Terminate()
}
func (t *Tool) ParseContext(ctx *cli.Context) bool {
if parsed := t.Tool.ParseContext(ctx); !parsed {
return parsed
}
if ctx.IsSet(AddressesFlag) {
t.mongoConfig.Addresses = storeStructuredMongo.SplitAddresses(ctx.String(AddressesFlag))
}
if ctx.IsSet(TLSFlag) {
t.mongoConfig.TLS = ctx.Bool(TLSFlag)
}
return true
}
func (t *Tool) NewMongoConfig() *storeStructuredMongo.Config {
mongoConfig := storeStructuredMongo.NewConfig()
if t.mongoConfig.Addresses != nil {
mongoConfig.Addresses = append([]string{}, t.mongoConfig.Addresses...)
}
mongoConfig.TLS = t.mongoConfig.TLS
mongoConfig.Database = t.mongoConfig.Database
mongoConfig.CollectionPrefix = t.mongoConfig.CollectionPrefix
if t.mongoConfig.Username != nil {
mongoConfig.Username = pointer.FromString(*t.mongoConfig.Username)
}
if t.mongoConfig.Password != nil {
mongoConfig.Password = pointer.FromString(*t.mongoConfig.Password)
}
return mongoConfig
}