go/go-clickhouse
1
0
Fork 0
mirror of https://github.com/uptrace/go-clickhouse.git synced 2025-06-08 23:26:11 +02:00
Code Issues Releases Activity

chore: support multiple env keys

Browse Source
This commit is contained in:
Vladimir Mihailenco 2022-08-28 16:53:07 +03:00
parent 2c9485da7f
commit 7f26fb6fae
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
chdebug/debug.go
View File

@ -43,14 +43,17 @@ func WithWriter(w io.Writer) Option {
// - CHDEBUG=0 - disables the hook. // - CHDEBUG=0 - disables the hook.
// - CHDEBUG=1 - enables the hook. // - CHDEBUG=1 - enables the hook.
// - CHDEBUG=2 - enables the hook and verbose mode. // - CHDEBUG=2 - enables the hook and verbose mode.
func FromEnv(key string) Option { func FromEnv(keys ...string) Option {
if key == "" { if len(keys) == 0 {
key = "CHDEBUG" keys = []string{"CHDEBUG"}
} }
return func(h *QueryHook) { return func(h *QueryHook) {
if env, ok := os.LookupEnv(key); ok { for _, key := range keys {
h.enabled = env != "" && env != "0" if env, ok := os.LookupEnv(key); ok {
h.verbose = env == "2" h.enabled = env != "" && env != "0"
h.verbose = env == "2"
break
}
} }
} }
} }