From 3ea919e06cb2c98703e6f311df22444c33447e80 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= <lxh@cxh.cn>
Date: Wed, 18 May 2022 11:17:22 +0800
Subject: [PATCH] =?UTF-8?q?:art:=20=E9=85=8D=E7=BD=AE=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config.go | 22 +++++++++++-----------
 logger.go |  4 ++++
 2 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/config.go b/config.go
index 9c9768d..72be93f 100644
--- a/config.go
+++ b/config.go
@@ -2,22 +2,22 @@ package logger
 
 import "fmt"
 
-type mode int
+type mode string
 
-var (
-	Dev  mode = 0
-	Prod mode = 1
+const (
+	Dev  mode = "development"
+	Prod mode = "production"
 )
 
 // LogConfig 日志配置
 type LogConfig struct {
-	Mode       mode   `env:"LOG_MODE"`             // dev, prod
-	LokiEnable bool   `env:"LOG_LOKI_ENABLE"`      // 是否启用Loki
-	FileEnable bool   `env:"LOG_FILE_ENABLE"`      // 是否输出到文件
-	LokiHost   string `env:"LOG_LOKI_HOST"`        // Loki地址
-	LokiPort   int    `env:"LOG_LOKI_PORT"`        // Loki端口
-	LokiSource string `env:"LOG_LOKI_SOURCE_NAME"` // Loki的source名称
-	LokiJob    string `env:"LOG_LOKI_JOB_NAME"`    // Loki的job名称
+	Mode       mode   `env:"LOG_MODE" envDefault:"production"` // dev, prod
+	LokiEnable bool   `env:"LOG_LOKI_ENABLE"`                  // 是否启用Loki
+	FileEnable bool   `env:"LOG_FILE_ENABLE"`                  // 是否输出到文件
+	LokiHost   string `env:"LOG_LOKI_HOST"`                    // Loki地址
+	LokiPort   int    `env:"LOG_LOKI_PORT"`                    // Loki端口
+	LokiSource string `env:"LOG_LOKI_SOURCE_NAME"`             // Loki的source名称
+	LokiJob    string `env:"LOG_LOKI_JOB_NAME"`                // Loki的job名称
 }
 
 func (c LogConfig) getLokiPushURL() string {
diff --git a/logger.go b/logger.go
index 832a057..587f8e3 100644
--- a/logger.go
+++ b/logger.go
@@ -19,6 +19,10 @@ func init() {
 			fmt.Println("日志配置解析错误: " + err.Error())
 			c = LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false}
 		}
+		// 如果值错了，直接默认为Prod
+		if c.Mode != Dev && c.Mode != Prod {
+			c.Mode = Prod
+		}
 		InitLogger(c)
 	}
 }