From 6126b66f6a84768ded7b80f574d6b97560856da4 Mon Sep 17 00:00:00 2001
From: bryanqiu <bryanqiu@tencent.com>
Date: Fri, 21 Jul 2023 11:24:30 +0800
Subject: [PATCH] add SetLogLevelByName

---
 log.go | 56 ++++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 44 insertions(+), 12 deletions(-)

diff --git a/log.go b/log.go
index 62b8df2..7ca7b8a 100644
--- a/log.go
+++ b/log.go
@@ -16,16 +16,38 @@ import (
 )
 
 const (
-	PANIC       logLevel = 0
-	FATAL       logLevel = 1
-	ERROR       logLevel = 2
-	NOTICE      logLevel = 3
-	WARNING     logLevel = 4
-	INFO        logLevel = 5
-	DEBUG       logLevel = 9
+	PANIC       LogLevel = 0
+	FATAL       LogLevel = 1
+	ERROR       LogLevel = 2
+	NOTICE      LogLevel = 3
+	WARNING     LogLevel = 4
+	INFO        LogLevel = 5
+	DEBUG       LogLevel = 9
+	ErrLogLevel          = "~~~~~invalid loglevel~~~~~"
 	ErrLogPanic          = "~~~~panic~~~~~~"
 )
 
+var (
+	mapStrToLevel = map[string]LogLevel{
+		"PANIC":   PANIC,
+		"FATAL":   FATAL,
+		"ERROR":   ERROR,
+		"NOTICE":  NOTICE,
+		"WARNING": WARNING,
+		"INFO":    INFO,
+		"DEBUG":   DEBUG,
+	}
+	mapLevelToStr = map[LogLevel]string{
+		PANIC:   "PANIC",
+		FATAL:   "FATAL",
+		ERROR:   "ERROR",
+		NOTICE:  "NOTICE",
+		WARNING: "WARNING",
+		INFO:    "INFO",
+		DEBUG:   "DEBUG",
+	}
+)
+
 var (
 	golog        *log.Logger        = nil
 	mylog        *Logger            = nil
@@ -33,11 +55,11 @@ var (
 	modulelogs   map[string]*Logger = map[string]*Logger{}
 )
 
-type logLevel int
+type LogLevel int
 type Logger struct {
 	module   string
 	golog    *log.Logger
-	loglevel logLevel
+	loglevel LogLevel
 }
 
 type LogidCreator interface {
@@ -62,13 +84,23 @@ func New(module string) *Logger {
 	}
 }
 
-// ////////////////////////////////////////////////////////////////
-func (log *Logger) SetLogLevel(newlv logLevel) (oldlv logLevel) {
+// SetLogLevel set new log level
+func (log *Logger) SetLogLevel(newlv LogLevel) (oldlv LogLevel) {
 	oldlv = log.loglevel
 	log.loglevel = newlv
 	return oldlv
 }
 
+// SetLogLevelByName set new log level by level name
+func (log *Logger) SetLogLevelByName(newlv string) (oldlv string) {
+	if newlevel, ok := mapStrToLevel[newlv]; ok {
+		oldlv = log.loglevel
+		log.loglevel = newlevel
+		return mapLevelToStr[oldlv]
+	}
+	return ErrLogLevel
+}
+
 func (log *Logger) Print(prefix string, v interface{}) {
 	var str = ""
 	if pkg, err := json.Marshal(v); err != nil {
@@ -134,7 +166,7 @@ func (log *Logger) GetLogidStr(format string) string {
 	return ""
 }
 
-func (log *Logger) logwrite(typ logLevel, calldepth int, format string, v ...interface{}) {
+func (log *Logger) logwrite(typ LogLevel, calldepth int, format string, v ...interface{}) {
 	if typ > log.loglevel {
 		return
 	}