forked from Netflix/rend
-
Notifications
You must be signed in to change notification settings - Fork 0
/
flags.go
69 lines (54 loc) · 2.5 KB
/
flags.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
// Copyright 2015 Netflix, Inc.
//
// 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 f
import "flag"
import "os"
// 2-character keys with 26 possibilities = 676 keys
// 3-character keys with 26 possibilities = 17,576 keys
// 4-character keys with 26 possibilities = 456,976 keys
// 5-character keys with 26 possibilities = 11,881,376 keys
var Binary bool
var Text bool
var KeyLength int
var NumOps int
var NumWorkers int
var Port int
var Pprof string
var Host string
// Flags
func init() {
flag.BoolVar(&Binary, "binary", false, "Use the binary protocol. Cannot be combined with --text or -t.")
flag.BoolVar(&Binary, "b", false, "Use the binary protocol. Cannot be combined with --text or -t. (shorthand)")
flag.BoolVar(&Text, "text", false, "Use the text protocol (default). Cannot be combined with --binary or -b.")
flag.BoolVar(&Text, "t", false, "Use the text protocol (default). Cannot be combined with --binary or -b. (shorthand)")
flag.IntVar(&KeyLength, "key-length", 4, "Length in bytes of each key. Smaller values mean more overlap.")
flag.IntVar(&KeyLength, "kl", 4, "Length in bytes of each key. Smaller values mean more overlap. (shorthand)")
flag.IntVar(&NumOps, "num-ops", 1000000, "Total number of operations to perform.")
flag.IntVar(&NumOps, "n", 1000000, "Total number of operations to perform. (shorthand)")
flag.IntVar(&NumWorkers, "workers", 10, "Number of communication goroutines to run.")
flag.IntVar(&NumWorkers, "w", 10, "Number of communication goroutines to run.")
flag.IntVar(&Port, "port", 11212, "Port to connect to.")
flag.IntVar(&Port, "p", 11212, "Port to connect to. (shorthand)")
flag.StringVar(&Pprof, "pprof", "", "File to be filled with pprof sampling data")
flag.StringVar(&Host, "h", "localhost", "Hostname / IP to connect to.")
flag.StringVar(&Host, "host", "localhost", "Hostname / IP to connect to. (shorthand)")
flag.Parse()
if (Binary && Text) || KeyLength <= 0 || NumOps <= 0 {
flag.Usage()
os.Exit(1)
}
if !Binary && !Text {
Text = true
}
}