在 Spring Boot 服务上运行有限数量的线程

Question

我目前正在使用 Spring Boot 开发一个 Web 应用程序，但我在服务层遇到了问题。

我的服务层有一个繁重的方法。如果多个用户调用相同的服务，应用程序会因内存不足而停止。所以我只想限制该方法的并行运行线程的数量。到目前为止，我已经在该方法上使用了 synchronized。但它会将其限制为单线程方法。

@Service
public class DocumentService{

    private synchronized void doReplacement(){
       //should have limited no of multi threads (eg. 3)
    }

    private void normalMethod(){
       //no restrictions
    }

}

我能做些什么来完成这项任务。任何帮助将不胜感激。

Answer 1

使用某种请求限制（即每秒的请求数）可能比使用可以同时执行方法的线程数更好。例如直接使用Guava's RateLimiter，或者事件adding declarative support for with Spring's AOP。

如果您仍想使用线程，我的建议是使用 ExecutorService：

@Service
public class DocumentService {

    private final ExecutorService executor;

    @Autowired
    public DocumentService(
        @Value("${some.config.property}") int maxConcurrentThreads) {
        // will allow only the given number of threads
        executor = Executors.newFixedThreadPool(maxConcurrentThreads);
    }

    private void doReplacementWithLimitedConcurrency(String s, int i){
        Future<?> future = executor.submit(() -> doReplacement(s, i));
        future.get(); // will block until a thread picks up the task
                      // and finishes executing doReplacement
    }

    private void doReplacement(String s, int i){
    }

    // other methods

    @PreDestroy
    public void performThreadPoolCleanup() throws Exception {
        executor.shutdown();
        executor.awaitTermination(10, TimeUnit.SECONDS); 
        executor.shutdownNow();
    }
}