Add the Mute levels to to logger

haroldcampbell · Jun 16, 2021 · a32c65e · a32c65e
1 parent 926c778
commit a32c65e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/utils/logging.go b/utils/logging.go
@@ -85,6 +85,10 @@ func YellowText(v interface{}) string {
 }
 
 func Tag(format string, v ...interface{}) {
+	if muteLevel == FullMute {
+		return
+	}
+
 	message := fmt.Sprintf(format, v...)
 	log.Printf("%s %s", CyanText("==>"), message)
 }
@@ -114,6 +118,10 @@ func PrettyMongoString(v interface{}) string {
 }
 
 func Log(stem string, format string, v ...interface{}) {
+	if muteLevel == FullMute || muteLevel == TestMute {
+		return
+	}
+
 	t := time.Now()
 
 	var buf []byte
@@ -128,6 +136,10 @@ func Log(stem string, format string, v ...interface{}) {
 }
 
 func Error(stem string, format string, v ...interface{}) {
+	if muteLevel == FullMute {
+		return
+	}
+
 	t := time.Now()
 
 	var buf []byte

diff --git a/utils/routine_logger.go b/utils/routine_logger.go
@@ -17,14 +17,8 @@ const (
 	FullMute
 )
 
-// var isRunningInTest = false
 var muteLevel MuteLevel
 
-func init() {
-	// isRunningInTest = flag.Lookup("test.v") != nil
-	// fmt.Printf("test.v: %v\n", flag.Lookup("test.v") == nil)
-}
-
 func SetMuteLevel(l MuteLevel) {
 	muteLevel = l
 }