enable log level from env

Author: chanmaoganda

cmd/cmd.go

@@ -4,7 +4,6 @@ import (
 	"github.com/chanmaoganda/fileshare/cmd/cache"
 	"github.com/chanmaoganda/fileshare/cmd/fileshare/client"
 	"github.com/chanmaoganda/fileshare/cmd/fileshare/server"
-	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -17,11 +16,6 @@ var RootCmd = &cobra.Command{
 }
 
 func init() {
-	logrus.SetLevel(logrus.DebugLevel)
-	logrus.SetFormatter(&logrus.TextFormatter{
-		ForceColors: true,
-	})
-
 	// server commands
 	RootCmd.AddCommand(server.ServerCmd)
 

internal/core/setup/setup.go

@@ -4,6 +4,7 @@ import (
 	"os"
 
 	"github.com/chanmaoganda/fileshare/internal/config"
+	"github.com/chanmaoganda/fileshare/internal/pkg/logger"
 	"github.com/chanmaoganda/fileshare/internal/pkg/util"
 	"github.com/chanmaoganda/fileshare/internal/service"
 	"github.com/sirupsen/logrus"
@@ -16,6 +17,7 @@ func SetupClient(cmd *cobra.Command, args []string) error {
 		logrus.Error(err)
 		return err
 	}
+	logger.SetupLogger()
 	service.InitClientService()
 	// if directory cannot be set correctly, following actions will panic
 	return setupDirectory()
@@ -27,6 +29,7 @@ func SetupServer(cmd *cobra.Command, args []string) error {
 		logrus.Error(err)
 		return err
 	}
+	logger.SetupLogger()
 	service.InitServerService()
 	// if directory cannot be set correctly, following actions will panic
 	return setupDirectory()

internal/pkg/logger/logger.go

@@ -0,0 +1,21 @@
+package logger
+
+import (
+	"os"
+
+	"github.com/sirupsen/logrus"
+)
+
+func SetupLogger() {
+	levelString := os.Getenv("LOG")
+
+	level, err := logrus.ParseLevel(levelString)
+	if err != nil {
+		logrus.SetLevel(logrus.InfoLevel)
+	}
+
+	logrus.SetLevel(level)
+	logrus.SetFormatter(&logrus.TextFormatter{
+		ForceColors: true,
+	})
+}