fix user dict load.

twlkyao · Sep 2, 2013 · 9534a78 · 9534a78
1 parent 3a0de75
commit 9534a78
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/conf/jieba/sougou.dict → conf/sougou.dict b/conf/jieba/sougou.dict → conf/sougou.dict
diff --git a/conf/jieba/user.dict → conf/user.dict b/conf/jieba/user.dict → conf/user.dict
@@ -3,3 +3,4 @@
 显瘦 3
 又拍云 3
 iphone 3
+鲜芋仙 3
diff --git a/src/main/java/com/huaban/analysis/jieba/WordDictionary.java b/src/main/java/com/huaban/analysis/jieba/WordDictionary.java
@@ -16,8 +16,7 @@
 public class WordDictionary {
     private static WordDictionary singleInstance;
     private static final String MAIN_DICT = "/dict.txt";
-    private static String SOUGOU_DICT = "jieba/sougou.dict";
-    private static String USER_DICT = "jieba/user.dict";
+    private static String USER_DICT_SUFFIX = ".dict";
 
     public final TrieNode trie = new TrieNode();
     public final Map<String, Double> freqs = new HashMap<String, Double>();
@@ -42,10 +41,11 @@ public synchronized static WordDictionary getInstance() {
      */
     public synchronized void init(File configFile) {
         if (!isLoaded) {
-            File sougouDict = new File(configFile, SOUGOU_DICT);
-            File userDict = new File(configFile, USER_DICT);
-            singleInstance.loadUserDict(sougouDict);
-            singleInstance.loadUserDict(userDict);
+            for (File userDict : configFile.listFiles()) {
+                if (userDict.getPath().endsWith(USER_DICT_SUFFIX)) {
+                    singleInstance.loadUserDict(userDict);
+                }
+            }
             isLoaded = true;
         }
     }