adjusted log.go

log.go

@@ -5,31 +5,46 @@ import (
 	"io/ioutil"
 	"os"
 	"strings"
+	"sync"
 )
 
-var log *logrus.Logger
+var (
+	log  *logrus.Logger
+	once sync.Once
+)
+
+func InitializeLogger() {
+	once.Do(func() {
+		log = logrus.New()
+		// We do not want to log by default
+		log.SetOutput(ioutil.Discard)
+		log.SetLevel(logrus.PanicLevel)
+		// Check if DEBUG_I2P is set
+		if logLevel := os.Getenv("DEBUG_I2P"); logLevel != "" {
+			log.SetOutput(os.Stdout)
+			switch strings.ToLower(logLevel) {
+			case "debug":
+				log.SetLevel(logrus.DebugLevel)
+			case "warn":
+				log.SetLevel(logrus.WarnLevel)
+			case "error":
+				log.SetLevel(logrus.ErrorLevel)
+			default:
+				log.SetLevel(logrus.DebugLevel)
+			}
+			log.WithField("level", log.GetLevel()).Debug("Logging enabled.")
+		}
+	})
+}
+
+// GetLogger returns the initialized logger
+func GetLogger() *logrus.Logger {
+	if log == nil {
+		InitializeLogger()
+	}
+	return log
+}
 
 func init() {
-	log = logrus.New()
+	InitializeLogger()
-
-	// We do not want to log by default
-	log.SetOutput(ioutil.Discard)
-	log.SetLevel(logrus.PanicLevel)
-
-	// Check if DEBUG_I2P is set
-	if logLevel := os.Getenv("DEBUG_I2P"); logLevel != "" {
-		log.SetOutput(os.Stdout)
-		switch strings.ToLower(logLevel) {
-		case "debug":
-			log.SetLevel(logrus.DebugLevel)
-		case "warn":
-			log.SetLevel(logrus.WarnLevel)
-		case "error":
-			log.SetLevel(logrus.ErrorLevel)
-		default:
-			log.SetLevel(logrus.DebugLevel)
-		}
-
-		log.WithField("level", log.GetLevel()).Debug("Logging enabled.")
-	}
 }