adjusted log.go

log.go

@@ -5,17 +5,20 @@ import (
 	"io/ioutil"
 	"os"
 	"strings"
+	"sync"
 )
 
-var log *logrus.Logger
+var (
+	log  *logrus.Logger
+	once sync.Once
+)
 
-func init() {
+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)
@@ -29,7 +32,19 @@ func init() {
 			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() {
+	InitializeLogger()
 }