/
nodes_remove.go
75 lines (61 loc) · 1.94 KB
/
nodes_remove.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
package cmd
import (
"code.vegaprotocol.io/vegacapsule/generator"
"code.vegaprotocol.io/vegacapsule/nomad"
"code.vegaprotocol.io/vegacapsule/state"
"context"
"fmt"
"github.com/spf13/cobra"
)
var datanodeBackupDir string
var nodesRemoveCmd = &cobra.Command{
Use: "remove",
Short: "Remove existing node set",
RunE: func(cmd *cobra.Command, args []string) error {
networkState, err := state.LoadNetworkState(homePath)
if err != nil {
return fmt.Errorf("failed to load network state: %w", err)
}
if networkState.Empty() {
return networkNotBootstrappedErr("nodes remove")
}
updatedNetworkState, err := nodesStopNode(context.Background(), *networkState, nodeName, true)
if err != nil {
return fmt.Errorf("failed stop node: %w", err)
}
updatedNetworkState, err = nodesRemoveNode(*updatedNetworkState, nodeName)
if err != nil {
return fmt.Errorf("failed remove node: %w", err)
}
return updatedNetworkState.Persist()
},
}
func init() {
nodesRemoveCmd.PersistentFlags().StringVar(&nodeName,
"name",
"",
"Name of the node tha should be removed",
)
nodesRemoveCmd.MarkFlagRequired("name")
nodesRemoveCmd.PersistentFlags().StringVar(&datanodeBackupDir, "datanode-backup-dir", "", "Directory where data node home directories should be backed up before removal")
}
func nodesRemoveNode(state state.NetworkState, name string) (*state.NetworkState, error) {
gen, err := generator.New(state.Config, *state.GeneratedServices, nomad.NewVoidJobRunner(), state.VegaChainID)
if err != nil {
return nil, err
}
nodeSet, err := state.GeneratedServices.GetNodeSet(name)
if err != nil {
return nil, err
}
if len(datanodeBackupDir) > 0 {
if err := gen.BackupDataNode(*nodeSet, datanodeBackupDir); err != nil {
return nil, fmt.Errorf("failed to backup data node: %w", err)
}
}
if err := gen.RemoveNodeSet(*nodeSet); err != nil {
return nil, err
}
delete(state.GeneratedServices.NodeSets, nodeSet.Name)
return &state, nil
}