/
flag.go
61 lines (47 loc) · 1.82 KB
/
flag.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
/*
Copyright 2019 The Vitess Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package topoproto
import (
"strings"
topodatapb "vitess.io/vitess/go/vt/proto/topodata"
)
// TabletTypeListFlag implements the pflag.Value interface, for parsing a command-line comma-separated
// list of values into a slice of TabletTypes.
type TabletTypeListFlag []topodatapb.TabletType
// String is part of the pflag.Value interface.
func (ttlv *TabletTypeListFlag) String() string {
return strings.Join(MakeStringTypeList(*ttlv), ",")
}
// Set is part of the pflag.Value interface.
func (ttlv *TabletTypeListFlag) Set(v string) (err error) {
*ttlv, err = ParseTabletTypes(v)
return err
}
// Type is part of the pflag.Value interface.
func (ttlv *TabletTypeListFlag) Type() string {
return "strings"
}
// TabletTypeFlag implements the pflag.Value interface, for parsing a command-line value into a TabletType.
type TabletTypeFlag topodatapb.TabletType
// String is part of the pflag.Value interfaces.
func (ttf *TabletTypeFlag) String() string {
return topodatapb.TabletType(*ttf).String()
}
// Set is part of the pflag.Value interfaces.
func (ttf *TabletTypeFlag) Set(v string) error {
t, err := ParseTabletType(v)
*ttf = TabletTypeFlag(t)
return err
}
// Type is part of the pflag.Value interface.
func (*TabletTypeFlag) Type() string { return "topodatapb.TabletType" }