使热更新产生的缓存文件包含用户词性 fix hankcs#1028

TimerYue · Nov 24, 2018 · 9acb8c4 · 9acb8c4
1 parent de42c07
commit 9acb8c4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/com/hankcs/hanlp/dictionary/CustomDictionary.java b/src/main/java/com/hankcs/hanlp/dictionary/CustomDictionary.java
@@ -110,6 +110,13 @@ private static boolean loadMainDictionary(String mainPath)
             }
             DataOutputStream out = new DataOutputStream(IOUtil.newOutputStream(mainPath + Predefine.BIN_EXT));
             // 缓存用户词性
+            if (customNatureCollector.isEmpty()) // 热更新
+            {
+                for (int i = Nature.begin.ordinal() + 1; i < Nature.values().length; ++i)
+                {
+                    customNatureCollector.add(Nature.values()[i]);
+                }
+            }
             IOUtil.writeCustomNature(out, customNatureCollector);
             // 缓存正文
             out.writeInt(attributeList.size());