-
Notifications
You must be signed in to change notification settings - Fork 7
/
generated_services.go
147 lines (117 loc) · 3.21 KB
/
generated_services.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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
package types
import (
"fmt"
"log"
"github.com/zclconf/go-cty/cty"
"github.com/zclconf/go-cty/cty/gocty"
)
var generatedServicesCtyType cty.Type
type GeneratedServices struct {
Wallet *Wallet
Faucet *Faucet
NodeSets NodeSetMap `cty:"node_sets"`
PreGenerateJobs []NomadJob
}
func init() {
var gs GeneratedServices
t, err := gocty.ImpliedType(gs)
if err != nil {
log.Fatalf("failed to imply type of GeneratedServices struct: %s", err)
}
generatedServicesCtyType = t
}
func DefaultGeneratedServices() GeneratedServices {
return GeneratedServices{
NodeSets: NodeSetMap{},
}
}
func NewGeneratedServices(w *Wallet, f *Faucet, ns []NodeSet) *GeneratedServices {
nsm := NodeSetMap{}
preGenJobs := []NomadJob{}
for _, ns := range ns {
nsm[ns.Name] = ns
preGenJobs = append(preGenJobs, ns.PreGenerateJobs...)
}
return &GeneratedServices{
Wallet: w,
Faucet: f,
NodeSets: nsm,
PreGenerateJobs: preGenJobs,
}
}
func (gs GeneratedServices) ToCtyValue() (*cty.Value, error) {
val, err := gocty.ToCtyValue(gs, generatedServicesCtyType)
if err != nil {
return nil, fmt.Errorf("failed to convert GeneratedServices to cty.Value: %w", err)
}
return &val, nil
}
func (gs GeneratedServices) GetByName(name string) []GeneratedService {
if gs.Wallet != nil && gs.Wallet.Name == name {
return []GeneratedService{gs.Wallet.GeneratedService}
}
if gs.Faucet != nil && gs.Faucet.Name == name {
return []GeneratedService{gs.Faucet.GeneratedService}
}
if ns, ok := gs.NodeSets[name]; ok {
out := []GeneratedService{ns.Vega.GeneratedService}
if ns.DataNode != nil {
out = append(out, ns.DataNode.GeneratedService)
}
return out
}
return nil
}
func (gs GeneratedServices) GetNodeSet(name string) (*NodeSet, error) {
ns, ok := gs.NodeSets[name]
if !ok {
return nil, fmt.Errorf("node set with name %q not found", name)
}
return &ns, nil
}
// PreGenerateJobsIDs returns pre gen jobs ids across all node sets
func (gs GeneratedServices) PreGenerateJobsIDs() []string {
preGenJobsIDs := make([]string, 0, len(gs.PreGenerateJobs))
for _, preGenJob := range gs.PreGenerateJobs {
preGenJobsIDs = append(preGenJobsIDs, preGenJob.ID)
}
return preGenJobsIDs
}
func (gs GeneratedServices) GetNodeSetsByGroupName(groupName string) []NodeSet {
return FilterNodeSets(gs.NodeSets.ToSlice(), NodeSetFilterByGroupName(groupName))
}
func (gs GeneratedServices) GetValidators() []NodeSet {
var out []NodeSet
for _, ns := range gs.NodeSets {
if ns.IsValidator() {
out = append(out, ns)
}
}
return out
}
func (gs GeneratedServices) GetNonValidators() []NodeSet {
var out []NodeSet
for _, ns := range gs.NodeSets {
if !ns.IsValidator() {
out = append(out, ns)
}
}
return out
}
func (gs GeneratedServices) ListValidators() []VegaNodeOutput {
var validators []VegaNodeOutput
for _, nodeSet := range gs.NodeSets {
if nodeSet.Mode != NodeModeValidator {
continue
}
validators = append(validators, VegaNodeOutput{
VegaNode: nodeSet.Vega,
Tendermint: TendermintOutput{
NodeID: nodeSet.Tendermint.NodeID,
ValidatorPublicKey: nodeSet.Tendermint.ValidatorPublicKey,
},
NomadJobName: nodeSet.Name,
})
}
return validators
}