1、优化文件搜索线程处理

superptzl · Aug 5, 2019 · 64cf288 · 64cf288
1 parent 1911428
commit 64cf288
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/main/java/com/xwintop/xJavaFxTool/services/littleTools/FileSearchToolService.java b/src/main/java/com/xwintop/xJavaFxTool/services/littleTools/FileSearchToolService.java
@@ -1,6 +1,5 @@
 package com.xwintop.xJavaFxTool.services.littleTools;
 
-import cn.hutool.core.thread.ThreadUtil;
 import com.xwintop.xJavaFxTool.controller.littleTools.FileSearchToolController;
 import com.xwintop.xJavaFxTool.utils.ConfigureUtil;
 import com.xwintop.xcore.util.javafx.TooltipUtil;
@@ -16,6 +15,7 @@
 import org.apache.lucene.store.Directory;
 import org.apache.lucene.store.FSDirectory;
 import org.apache.lucene.store.NoLockFactory;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
 
 import java.io.File;
 import java.io.IOException;
@@ -24,6 +24,7 @@
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.*;
+import java.util.concurrent.ThreadPoolExecutor;
 
 /**
  * @ClassName: FileSearchToolService
@@ -38,6 +39,7 @@
 public class FileSearchToolService {
     private FileSearchToolController fileSearchToolController;
     private static Timer autoRefreshIndexTimer = null;
+    private static ThreadPoolTaskExecutor refreshIndexPoolTaskExecutor;
     private static final String searchIndexDir = ConfigureUtil.getConfigurePath("searchIndexDir/");
 
     private static Directory directory;
@@ -61,6 +63,13 @@ public class FileSearchToolService {
         } catch (Exception e) {
             log.error("创建文件失败！", e);
         }
+        refreshIndexPoolTaskExecutor = new ThreadPoolTaskExecutor();
+        refreshIndexPoolTaskExecutor.setCorePoolSize(5);
+        refreshIndexPoolTaskExecutor.setMaxPoolSize(20);
+        refreshIndexPoolTaskExecutor.setQueueCapacity(30);
+        // 设置拒绝策略
+        refreshIndexPoolTaskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
+        refreshIndexPoolTaskExecutor.initialize();
     }
 
     public static void getIndexWriter(Directory directory) throws IOException {
@@ -185,7 +194,7 @@ public void refreshIndexAction() throws Exception {
     }
 
     public void addSearchIndexFile(Path path) {
-        ThreadUtil.execute(() -> {
+        refreshIndexPoolTaskExecutor.execute(() -> {
             try {
                 DirectoryStream<Path> stream = Files.newDirectoryStream(path);
                 Iterator<Path> pathIterator = stream.iterator();