colorful

johnsonz · Jul 21, 2017 · 7ea2bb5 · 7ea2bb5
1 parent 46747f6
commit 7ea2bb5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/sni.go b/sni.go
@@ -16,6 +16,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/fatih/color"
 	"github.com/golang/glog"
 )
 
@@ -294,7 +295,11 @@ func createFile() {
 func checkErr(messge string, err error, level int) {
 	if err != nil {
 		switch level {
-		case Info, Warning, Debug:
+		case Info:
+			color.Set(color.FgGreen)
+			defer color.Unset()
+			glog.Infoln(messge, err)
+		case Warning, Debug:
 			glog.Infoln(messge, err)
 		case Error:
 			glog.Fatalln(messge, err)