[logging] refs fibercrypto#298 - Add stderr and none options to SetOu…

…tput
simelo · Jan 20, 2020 · 8bb88cb · 8bb88cb
1 parent 4ded230
commit 8bb88cb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/util/logging/logging.go b/src/util/logging/logging.go
@@ -82,10 +82,17 @@ func Disable() {
 
 // SetOutput call correct function given a option to set logger's output
 func SetOutput(opt string) error {
-	if opt == "stdout" {
+	switch opt {
+	case "stdout":
 		SetOutputTo(os.Stderr)
 		return nil
-	} else {
+	case "stderr":
+		SetOutputTo(os.Stderr)
+		return nil
+	case "none":
+		SetOutputTo(NoWriter{})
+		return nil
+	default:
 		return SetOutputToFile(opt)
 	}
 }

diff --git a/src/util/util.go b/src/util/util.go
@@ -2,10 +2,11 @@ package util
 
 import (
 	"errors"
+	"strconv"
+
 	"github.com/fibercrypto/fibercryptowallet/src/core"
 	local "github.com/fibercrypto/fibercryptowallet/src/main"
 	"github.com/fibercrypto/fibercryptowallet/src/util/logging"
-	"strconv"
 )
 
 var logUtil = logging.MustGetLogger("FiberCrypto util")