add remove phrase DictAdd support

go-ego · Oct 2, 2021 · 55f64b1 · 55f64b1
1 parent b7662aa
commit 55f64b1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/phrase/phrase_dict.go b/phrase/phrase_dict.go
@@ -1,12 +1,13 @@
 package phrase
 
-// phraseDict is data map
+// phraseDict is phrase data map
 //
 // Generate from:
 // https://github.com/hotoo/pinyin/blob/master/data/phrases-dict.js
 //
 // Warning: Auto-generated file, don't edit.
-// If you want add more words, use phrase_dict_addition.go
+//
+// If you want add more words, use phrase_dict_add.go
 var phraseDict = map[string]string{
 	"阿Ｑ":         "ā Ｑ",
 	"阿爸":         "ā bà",

diff --git a/phrase/phrase_dict_add.go b/phrase/phrase_dict_add.go
@@ -1,10 +1,10 @@
 package phrase
 
 /*
-	go pinyin addition version
+	go pinyin addition dictionary
 */
 
-// DictAdd phrase dict addition
+// DictAdd phrase dict addition map
 var DictAdd = map[string]string{
 	"宿舍": "sù shè",
 	"不薄": "bù báo",
@@ -18,3 +18,8 @@ var DictAdd = map[string]string{
 func AddDict(text, py string) {
 	DictAdd[text] = py
 }
+
+// Remove remove a token from phrase dictionary.
+func Remove(text string) {
+	delete(DictAdd, text)
+}