-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.go
109 lines (100 loc) · 2.02 KB
/
config.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
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
package config
import (
"flag"
"fmt"
"strings"
"github.com/utkarsh-pro/use/pkg/utils"
)
var Version = "0.1.0"
var Transport = "http"
var Address = ":8080"
var Storage = "stupid"
var StoragePath = ""
var WorkerID = 4095
var LogLevel = "info"
var DBSyncType = "none"
var DBReadOnly = false
func Setup() {
setupFlags()
}
func setupFlags() {
flag.StringVar(
&Transport,
"transport",
utils.GetEnvOrDefault(convertToEnvName("USE", "transport"), Transport),
"transport to use",
)
flag.StringVar(
&Address,
"address",
utils.GetEnvOrDefault(convertToEnvName("USE", "address"), Address),
"address to listen on",
)
flag.StringVar(
&Storage,
"storage",
utils.GetEnvOrDefault(convertToEnvName("USE", "storage"), Storage),
"storage to use",
)
flag.StringVar(
&StoragePath,
"storage-path",
utils.GetEnvOrDefault(convertToEnvName("USE", "storage-path"), StoragePath),
"path to the storage",
)
flag.IntVar(
&WorkerID,
"worker-id",
utils.StringToInt(utils.GetEnvOrDefault(convertToEnvName("USE", "worker-id"),
utils.IntToString(WorkerID))),
"worker id",
)
flag.StringVar(
&LogLevel,
"log-level",
utils.GetEnvOrDefault(convertToEnvName("USE", "log-level"), LogLevel),
"log level",
)
flag.StringVar(
&DBSyncType,
"db-sync-type",
utils.GetEnvOrDefault(convertToEnvName("USE", "db-sync-type"), DBSyncType),
"db sync type",
)
flag.BoolVar(
&DBReadOnly,
"db-read-only",
utils.StringToBool(
utils.GetEnvOrDefault(convertToEnvName("USE", "db-read-only"), utils.BoolToString(DBReadOnly)),
),
"db read only",
)
flag.Parse()
}
func convertToEnvName(prefix, name string) string {
name = strings.ReplaceAll(strings.ToUpper(name), "-", "_")
return prefix + "_" + name
}
func String() string {
return fmt.Sprintf(
`
transport: %s,
address: %s,
storage: %s,
version: %s,
storage-path: %s,
worker-id: %d,
log-level: %s
db-sync-type: %s
db-read-only: %t`,
Transport,
Address,
Storage,
Version,
StoragePath,
WorkerID,
LogLevel,
DBSyncType,
DBReadOnly,
)
}