gaore-common-sdk-go/sdk/utils/debug.go

package utils

import (
	"fmt"
	"os"
	"strings"
)

type Debug func(format string, v ...interface{})

var hookPrint = func(input string) {
	fmt.Println(input)
}

var hookGetEnv = func() string {
	return os.Getenv("DEBUG")
}

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...))
		}
	}
}