blob: 4bdfc457669dd83fd894d37f3f6a9ac6f8052498 (
plain)
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
|
package omode
import (
"fmt"
"os"
"path"
)
// Mode used.
type Mode int
// Possible modes.
const (
Unknown Mode = iota
Server Mode = iota
TailClient Mode = iota
CatClient Mode = iota
GrepClient Mode = iota
MapClient Mode = iota
HealthClient Mode = iota
)
// New returns the mode based on the mode string.
func New(modeStr string) Mode {
switch modeStr {
case "dserver":
return Server
case "server":
return Server
case "dtail":
fallthrough
case "tail":
return TailClient
case "grep":
fallthrough
case "dgrep":
return GrepClient
case "cat":
fallthrough
case "dcat":
return CatClient
case "map":
fallthrough
case "dmap":
return MapClient
case "health":
return HealthClient
default:
panic(fmt.Sprintf("Unknown mode: '%s'", modeStr))
}
}
// Default mode.
func Default() Mode {
return New(path.Base(os.Args[0]))
}
func (m Mode) String() string {
switch m {
case Server:
return "server"
case TailClient:
return "tail"
case CatClient:
return "cat"
case GrepClient:
return "grep"
case MapClient:
return "map"
case HealthClient:
return "health"
default:
return "unknown"
}
}
|
