add inputDataPath 🚀

Author: madneal

config.yaml

@@ -1,6 +1,10 @@
 !!setting.ConfigSetting
+# F: Full search, scan all drivers in your pc; L: Local search, scan the paths you defined in inputDataPath
+searchMethod: F
+# if searchMethod is L, you should config inputDataPath
+inputDataPath:
+  - C:\Users\neal1\project\lucene-test\test-data
 compressFileList: null
 excludeFilePathList: [node_modules, $RECYCLE.BIN, Program Files, Windows]
 fileList: [docx, doc, xls, xlsx, pdf, txt]
 hasCreateIndex: true
-searchMethod: F

search/src/main/java/constants/CommonConstants.java

@@ -1,14 +1,11 @@
 package constants;
 
-import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
 
 public class CommonConstants {
 
-    public static List<String> DOCFILES = new ArrayList<String>(Arrays.asList(
-            "doc", "docx", "xls", "xlsx", "pdf", "txt"
-    ));
+    public static List<String> DOCFILES;
+    public static List<String> INPUT_DATA_PATH_LIST;
     public static List<String> EXCLUDE_FILE_PATHS;
     public static boolean IS_OPEN_CONTEXT = true;
     public static String INDEX_FILE_PATH = "index";

search/src/main/java/index/IndexUtil.java

@@ -30,7 +30,9 @@ public static void executeIndex(String searchType) {
                     totalCount += runIndexByEachPath(driver);
                 }
             } else {
-                fileBeans = FileUtil.getFolderFiles(CommonConstants.INPUT_FILE_PATH);
+                for (String path: CommonConstants.INPUT_DATA_PATH_LIST) {
+                    totalCount += runIndexByEachPath(path);
+                }
             }
             CommonConstants.TOTAL_FILE_NUM = String.valueOf(totalCount);
             BaseIndex.runIndex(fileBeans);
@@ -45,7 +47,7 @@ private static int runIndexByEachPath(String path) throws Exception {
         int totalNum = 0;
         if (files != null) {
             for (String file: files) {
-                file = path + file;
+                file = path + "\\" + file;
                 List<FileBean> fileBeans = new ArrayList<>();
                 fileBeans.addAll(FileUtil.getFolderFiles(file));
                 totalNum += fileBeans.size();

ui/src/main/java/controller/Controller.java

@@ -84,6 +84,8 @@ protected Void call() throws Exception {
     private void executeIndex() {
         ConfigSetting configSetting = ConfigController.readConfig();
         CommonConstants.EXCLUDE_FILE_PATHS = configSetting.getExcludeFilePathList();
+        CommonConstants.DOCFILES = configSetting.getFileList();
+        CommonConstants.INPUT_DATA_PATH_LIST = configSetting.getInputDataPath();
         IndexUtil.executeIndex(configSetting.getSearchMethod());
         if (configSetting.getHasCreateIndex() == false) {
             configSetting.setHasCreateIndex(true);

ui/src/main/java/setting/ConfigSetting.java

@@ -14,6 +14,8 @@ public class ConfigSetting {
 
     private boolean hasCreateIndex;
 
+    private List<String> inputDataPath;
+
     public String getSearchMethod() {
         return searchMethod;
     }
@@ -53,4 +55,12 @@ public boolean getHasCreateIndex() {
     public void setHasCreateIndex(boolean hasCreateIndex) {
         this.hasCreateIndex = hasCreateIndex;
     }
+
+    public List<String> getInputDataPath() {
+        return inputDataPath;
+    }
+
+    public void setInputDataPath(List<String> inputDataPath) {
+        this.inputDataPath = inputDataPath;
+    }
 }