Skip to content

Commit 403889a

Browse files
committed
refactor: replace Unwrap with Must in config merge calls
1 parent dcc97b1 commit 403889a

File tree

5 files changed

+50
-45
lines changed

5 files changed

+50
-45
lines changed

component/etcdv3/client.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ import (
1212
)
1313

1414
func New(conf *Config) *Client {
15-
conf = config.MergeR(DefaultCfg(), *conf).Unwrap()
15+
conf = config.MergeR(DefaultCfg(), *conf).Must()
1616
cfg := merge.Struct(new(client3.Config), conf).Unwrap()
1717
cfg.DialOptions = append(
1818
cfg.DialOptions,

component/gormclient/builder.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ func NewClients(conf map[string]*Config, logs log.Logger) map[string]*Client {
2323

2424
func New(conf *Config, logs log.Logger) *Client {
2525
logs = logs.WithName(Name)
26-
conf = config.MergeR(generic.Ptr(DefaultCfg()), conf).Unwrap()
26+
conf = config.MergeR(generic.Ptr(DefaultCfg()), conf).Must()
2727

2828
ormCfg := merge.Copy(new(gorm.Config), conf).Unwrap()
2929
ormCfg.NowFunc = func() time.Time { return time.Now().UTC() }

env/env.go

Lines changed: 39 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -28,25 +28,26 @@ func Set(key, value string) result.Error {
2828

2929
func MustSet(key, value string) { Set(key, value).Must() }
3030

31-
func GetDefault(name string, defaultVal string) string {
32-
val := Get(name)
33-
return lo.If(val != "", val).Else(defaultVal)
34-
}
35-
3631
func Get(names ...string) string {
3732
var val string
38-
GetVal(&val, names...)
33+
getVal(&val, names...)
3934
return val
4035
}
4136

4237
func MustGet(names ...string) string {
43-
var val string
44-
GetVal(&val, names...)
38+
val := Get(names...)
4539
assert.If(val == "", "env not found, names=%q", names)
4640
return val
4741
}
4842

49-
func GetVal(val *string, names ...string) {
43+
func GetOr(name string, defaultVal string) string {
44+
val := Get(name)
45+
return lo.If(val != "", val).Else(defaultVal)
46+
}
47+
48+
func GetWith(val *string, names ...string) { getVal(val, names...) }
49+
50+
func getVal(val *string, names ...string) {
5051
for _, name := range names {
5152
env, ok := Lookup(name)
5253
env = trim(env)
@@ -57,49 +58,52 @@ func GetVal(val *string, names ...string) {
5758
}
5859
}
5960

60-
func GetBoolVal(val *bool, names ...string) {
61-
dt := Get(names...)
62-
if dt == "" {
63-
return
61+
func GetBool(names ...string) bool {
62+
var val string
63+
getVal(&val, names...)
64+
if val == "" {
65+
return false
6466
}
6567

66-
v, err := strconv.ParseBool(dt)
68+
v, err := strconv.ParseBool(val)
6769
if err != nil {
68-
slog.Error(fmt.Sprintf("env: failed to parse string to bool, keys=%q value=%s err=%v", names, dt, err))
69-
return
70+
slog.Error(fmt.Sprintf("env: failed to parse string to bool, keys=%q value=%s err=%v", names, val, err))
71+
return false
7072
}
7173

72-
*val = v
74+
return v
7375
}
7476

75-
func GetIntVal(val *int, names ...string) {
76-
dt := Get(names...)
77-
if dt == "" {
78-
return
77+
func GetInt(names ...string) int {
78+
var val string
79+
getVal(&val, names...)
80+
if val == "" {
81+
return -1
7982
}
8083

81-
v, err := strconv.Atoi(dt)
84+
v, err := strconv.Atoi(val)
8285
if err != nil {
83-
slog.Error(fmt.Sprintf("env: failed to parse string to int, keys=%q value=%s err=%v", names, dt, err))
84-
return
86+
slog.Error(fmt.Sprintf("env: failed to parse string to int, keys=%q value=%s err=%v", names, val, err))
87+
return -1
8588
}
8689

87-
*val = v
90+
return v
8891
}
8992

90-
func GetFloatVal(val *float64, names ...string) {
91-
dt := Get(names...)
92-
if dt == "" {
93-
return
93+
func GetFloat(names ...string) float64 {
94+
var val string
95+
getVal(&val, names...)
96+
if val == "" {
97+
return -1
9498
}
9599

96-
v, err := strconv.ParseFloat(dt, 64)
100+
v, err := strconv.ParseFloat(val, 64)
97101
if err != nil {
98-
slog.Error(fmt.Sprintf("env: failed to parse string to float, keys=%q value=%s err=%v", names, dt, err))
99-
return
102+
slog.Error(fmt.Sprintf("env: failed to parse string to float, keys=%q value=%s err=%v", names, val, err))
103+
return -1
100104
}
101105

102-
*val = v
106+
return v
103107
}
104108

105109
func Lookup(key string) (string, bool) { return os.LookupEnv(keyHandler(key)) }
@@ -111,6 +115,8 @@ func Delete(key string) result.Error {
111115
})
112116
}
113117

118+
func MustDelete(key string) { Delete(key).Must() }
119+
114120
func Expand(value string) result.Result[string] {
115121
return result.Wrap(envsubst.String(value)).Log(func(e *zerolog.Event) {
116122
e.Str("value", value)

migrates/entmigrates/migrate.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ type GoMigrate struct {
5858

5959
// New returns a new GoMigrate.
6060
func New(db *entsql.Driver, cfg *Config, migrations []*Migration, tables []*schema.Table) *GoMigrate {
61-
cfg = config.MergeR(generic.Ptr(DefaultConfig), cfg).Unwrap()
61+
cfg = config.MergeR(generic.Ptr(DefaultConfig), cfg).Must()
6262

6363
orm := assert.Must1(gorm.Open(dp.New(dp.Config{
6464
DriverName: db.Dialect(),

running/runtime.go

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,6 @@ var (
2424

2525
Env = "debug"
2626

27-
// IsDebug
28-
// Deprecated: use Debug
29-
IsDebug = true
30-
3127
// InstanceID service id
3228
InstanceID = xid.New().String()
3329

@@ -66,13 +62,16 @@ var (
6662
enableDebug = monster.Bool("debug", false, "enable debug")
6763
)
6864

69-
func Debug() bool {
70-
return IsDebug || enableDebug.Get()
71-
}
65+
func Debug() bool { return enableDebug.Get() }
7266

7367
func init() {
74-
env.GetBoolVal(&IsDebug, "enable_debug", "debug", "dev_mode")
75-
env.GetVal(&Env, "env", "run_mode", "run_env")
68+
if env.GetBool("enable_debug", "debug", "dev_mode") {
69+
assert.Exit(enableDebug.Set(true))
70+
}
71+
72+
if e := env.Get("env", "run_mode", "run_env"); e != "" {
73+
Env = e
74+
}
7675

7776
id, err := machineid.ID()
7877
if err == nil {

0 commit comments

Comments
 (0)