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
|
package shell
import (
"context"
"flag"
"fmt"
"github.com/seaweedfs/seaweedfs/weed/pb/master_pb"
"github.com/seaweedfs/seaweedfs/weed/storage/needle"
"github.com/seaweedfs/seaweedfs/weed/storage/super_block"
"github.com/seaweedfs/seaweedfs/weed/storage/types"
"io"
)
func init() {
Commands = append(Commands, &commandGrow{})
}
type commandGrow struct {
}
func (c *commandGrow) Name() string {
return "volume.grow"
}
func (c *commandGrow) Help() string {
return `grow volumes
volume.grow [-collection=<collection name>] [-dataCenter=<data center name>]
`
}
func (c *commandGrow) IsResourceHeavy() bool {
return false
}
func (c *commandGrow) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
volumeVacuumCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
growCount := volumeVacuumCommand.Uint("count", 2, "")
collection := volumeVacuumCommand.String("collection", "", "grow this collection")
dataCenter := volumeVacuumCommand.String("dataCenter", "", "grow volumes only from the specified data center")
rack := volumeVacuumCommand.String("rack", "", "grow volumes only from the specified rack")
dataNode := volumeVacuumCommand.String("dataNode", "", "grow volumes only from the specified data node")
diskType := volumeVacuumCommand.String("diskType", "", "grow volumes only from the specified disk type")
if err = volumeVacuumCommand.Parse(args); err != nil {
return nil
}
if *collection == "" {
return fmt.Errorf("collection option is required")
}
t, _, err := collectTopologyInfo(commandEnv, 0)
if err != nil {
return err
}
volumeGrowRequest := &master_pb.VolumeGrowRequest{
Collection: *collection,
DataCenter: *dataCenter,
Rack: *rack,
DataNode: *dataNode,
WritableVolumeCount: uint32(*growCount),
}
collectionFound := false
dataCenterFound := *dataCenter == ""
rackFound := *rack == ""
dataNodeFound := *dataNode == ""
diskTypeFound := *diskType == ""
for _, dc := range t.DataCenterInfos {
if dc.Id == *dataCenter {
dataCenterFound = true
}
for _, r := range dc.RackInfos {
if r.Id == *rack {
rackFound = true
}
for _, dn := range r.DataNodeInfos {
if dn.Id == *dataNode {
dataNodeFound = true
}
for _, di := range dn.DiskInfos {
if !diskTypeFound && di.Type == types.ToDiskType(*diskType).String() {
diskTypeFound = true
}
for _, vi := range di.VolumeInfos {
if !collectionFound && vi.Collection == *collection {
replicaPlacement, _ := super_block.NewReplicaPlacementFromByte(byte(vi.ReplicaPlacement))
volumeGrowRequest.Ttl = needle.LoadTTLFromUint32(vi.Ttl).String()
volumeGrowRequest.DiskType = vi.DiskType
volumeGrowRequest.Replication = replicaPlacement.String()
collectionFound = true
}
if collectionFound && dataCenterFound && rackFound && dataNodeFound && diskTypeFound {
break
}
}
}
}
}
}
if !dataCenterFound {
return fmt.Errorf("data center not found")
}
if !rackFound {
return fmt.Errorf("rack not found")
}
if !dataNodeFound {
return fmt.Errorf("data node not found")
}
if !diskTypeFound {
return fmt.Errorf("disk type not found")
}
if !collectionFound {
return fmt.Errorf("collection not found")
}
if err = commandEnv.MasterClient.WithClient(false, func(client master_pb.SeaweedClient) error {
if _, err := client.VolumeGrow(context.Background(), volumeGrowRequest); err != nil {
return err
}
return nil
}); err != nil {
return
}
return nil
}
|
