add level config

README.md

@@ -11,6 +11,7 @@ Ela is a tool to migrator data between different elasticsearch with different ve
 ## Config
 
 ```yaml
+level: "info"         # log level which is used to log message, can be assigned to 'debug', 'info', 'warn', 'error'.
 elastics:             # elasticsearch clusters
   es5:                # cluster name which is unique
     addresses:        # elasticsearch addresses which has many masters

config/cfg.go

@@ -36,4 +36,5 @@ type ESConfig struct {
 type Config struct {
 	ESConfigs map[string]*ESConfig `mapstructure:"elastics"`
 	Tasks     []*TaskCfg           `mapstructure:"tasks"`
+	Level     string               `mapstructure:"level"`
 }

main.go

@@ -31,6 +31,8 @@ func main() {
 		return
 	}
 
+	utils.InitLogger(&cfg)
+
 	ctx := context.Background()
 	taskMgr, err := service.NewTaskMgr(&cfg)
 	if err != nil {

utils/logger.go

@@ -2,24 +2,37 @@ package utils
 
 import (
 	"context"
+	"github.com/CharellKing/ela/config"
 	"github.com/samber/lo"
 	log "github.com/sirupsen/logrus"
 	"os"
 )
 
 var logger *log.Logger
 
-func init() {
+func InitLogger(cfg *config.Config) {
+	levelMap := map[string]log.Level{
+		"debug": log.DebugLevel,
+		"info":  log.InfoLevel,
+		"warn":  log.WarnLevel,
+		"error": log.ErrorLevel,
+	}
+
+	level, ok := levelMap[cfg.Level]
+	if !ok {
+		level = log.InfoLevel
+	}
 	logger = &log.Logger{
 		Out:       os.Stdout,
 		Formatter: &log.JSONFormatter{},
 		Hooks:     make(log.LevelHooks),
-		Level:     log.InfoLevel,
+		Level:     level,
 	}
 	logger.SetReportCaller(true)
 }
 
 func GetLogger(ctx context.Context) *log.Entry {
+
 	entry := log.NewEntry(logger)
 
 	ctxKeyMap := map[CtxKey]func(ctx context.Context) string{