[Fix][core] when file type is local,can not get the format of file

1 files changed, 8 insertions, 4 deletions

diff --git a/groot-core/src/main/java/com/geedgenetworks/core/udf/knowlegdebase/handler/AbstractSingleKnowledgeBaseHandler.java b/groot-core/src/main/java/com/geedgenetworks/core/udf/knowlegdebase/handler/AbstractSingleKnowledgeBaseHandler.java
index 3869569..0089f5a 100644
--- a/groot-core/src/main/java/com/geedgenetworks/core/udf/knowlegdebase/handler/AbstractSingleKnowledgeBaseHandler.java
+++ b/groot-core/src/main/java/com/geedgenetworks/core/udf/knowlegdebase/handler/AbstractSingleKnowledgeBaseHandler.java
@@ -33,6 +33,13 @@ public abstract class AbstractSingleKnowledgeBaseHandler extends AbstractKnowled
         if ("http".equals(knowledgeBaseConfig.getFsType())) {
             this.knowledgeMetedataCache = getMetadata(knowledgeBaseConfig.getFsType(), knowledgeBaseConfig.getFsPath(), knowledgeBaseConfig.getFiles().get(0));
         }
+        if ("local".equals(knowledgeBaseConfig.getFsType())) {
+            knowledgeMetedataCache = new KnowLedgeBaseFileMeta();
+            knowledgeMetedataCache.setIsValid(1);
+            knowledgeMetedataCache.setFormat(knowledgeBaseConfig.getFiles().get(0).substring(knowledgeBaseConfig.getFiles().get(0).lastIndexOf(".") + 1));
+            knowledgeMetedataCache.setName(knowledgeBaseConfig.getFiles().get(0));
+            knowledgeMetedataCache.setPath(knowledgeBaseConfig.getFsPath() + knowledgeBaseConfig.getFiles().get(0));
+        }
         return buildKnowledgeBase();
     }
 
@@ -88,11 +95,8 @@ public abstract class AbstractSingleKnowledgeBaseHandler extends AbstractKnowled
                 case "aes":
                     result = AESUtil.decrypt(data, AES_KEY);
                     break;
-                case "csv":
-                    result = data;
-                    break;
                 default:
-                    logger.error("unknown format: " + knowledgeMetedataCache.getFormat());
+                    result = data;
             }
         } catch (Exception e) {
             logger.error("decrypt error", e);