123456789101112131415161718192021222324252627282930313233343536 |
- package utils
- import (
- "fmt"
- "os"
- "strings"
- )
- type Debug func(format string, v ...interface{})
- var hookGetEnv = func() string {
- return os.Getenv("DEBUG")
- }
- var hookPrint = func(input string) {
- fmt.Println(input)
- }
- func Init(flag string) Debug {
- enable := false
- env := hookGetEnv()
- parts := strings.Split(env, ",")
- for _, part := range parts {
- if part == flag {
- enable = true
- break
- }
- }
- return func(format string, v ...interface{}) {
- if enable {
- hookPrint(fmt.Sprintf(format, v...))
- }
- }
- }
|