From d9fdeabf85b7ee09842aaa332a9553a639524206 Mon Sep 17 00:00:00 2001
From: Yun <yun@yun.ink>
Date: Wed, 24 Jul 2024 19:36:32 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=9C=A8append=E7=9A=84?=
 =?UTF-8?q?=E6=97=B6=E5=80=99=E4=BC=9A=E6=9C=89panic=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 example/main.go | 24 ++++++++++++++++++++----
 langx.go        |  3 +++
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/example/main.go b/example/main.go
index 0325a85..baf5893 100644
--- a/example/main.go
+++ b/example/main.go
@@ -11,14 +11,29 @@ import (
 var assetsFs embed.FS
 
 func main() {
-	demoEmbed()
+	regByAppend()
 }
 
-func demoEmbed() {
+// 导入语言包 基于Append
+func regByAppend() {
+	langx.AppendCode(map[string]int{
+		"success": 200,
+	})
+	langx.AppendTrans("zh-CN", map[string]string{
+		"success": "成功",
+	})
+
+	code, msg := langx.GetTransFormat("zh-CN", "success", map[string]string{})
+	fmt.Println(code, msg)
+
+}
+
+// 导入语言包 基于Embed
+func regByEmbed() {
 	err := langx.RegisterEmbed(assetsFs)
 	fmt.Println(err)
 
-	code, msg := langx.GetTransFormat("zh", "success", map[string]string{}) 
+	code, msg := langx.GetTransFormat("zh", "success", map[string]string{})
 	fmt.Println(code, msg)
 	code, msg = langx.GetTransFormat("en", "error", map[string]string{
 		"msg": "这是失败的原因",
@@ -26,7 +41,8 @@ func demoEmbed() {
 	fmt.Println(code, msg)
 }
 
-func demo1() {
+// 导入语言包 基于文件
+func regByDir() {
 	langx.RegisterDir("./lang")
 
 	code, msg := langx.GetTransFormat("zh", "success", map[string]string{})
diff --git a/langx.go b/langx.go
index 29dcbfe..81f5cb3 100644
--- a/langx.go
+++ b/langx.go
@@ -66,6 +66,9 @@ func AppendTrans(langName string, trans map[string]string) {
 	l.mut.Lock()
 	defer l.mut.Unlock()
 	for k, v := range trans {
+		if l.transMap[langName] == nil {
+			l.transMap[langName] = map[string]string{}
+		}
 		l.transMap[langName][k] = v
 	}
 }