Fixed a race condition

Fixes #12
pieterclaerhout · May 23, 2020 · badf63b · badf63b
1 parent 4b4cee9
commit badf63b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/logger_internal.go b/logger_internal.go
@@ -6,6 +6,7 @@ import (
 	"os"
 	"runtime"
 	"strings"
+	"sync"
 	"time"
 
 	"github.com/fatih/color"
@@ -19,6 +20,8 @@ var colors = map[string]*color.Color{
 	"FATAL": color.New(color.FgHiRed),
 }
 
+var mutex sync.Mutex
+
 func init() {
 
 	TimeZone, _ = time.LoadLocation("Europe/Brussels")
@@ -73,6 +76,8 @@ func printNonColoredMessage(level string, message string) {
 func printColoredMessage(level string, message string) {
 	w := writerForLevel(level)
 	if c, ok := colors[level]; ok {
+		mutex.Lock()
+		defer mutex.Unlock()
 		c.EnableColor()
 		c.Fprint(w, message)
 		w.Write([]byte("\n"))