Skip to content

线程池配置优化 #15

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
WhiteMagic2014 merged 3 commits into from
Jun 20, 2025
Merged

线程池配置优化 #15

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
525f1ff
弃用 parallelStream 改为自定义线程池避免和业务系统存在共用 forkJoinPool 的可能
lifeopsgo Jun 20, 2025
21b1330
优化线程池配置，当没有任务需要执行时自动关闭线程
lifeopsgo Jun 20, 2025
aeaa1d6
Merge pull request #1 from WhiteMagic2014/master
lifeopsgo Jun 20, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 32 additions & 5 deletions src/main/java/io/github/whitemagic2014/tts/TTS.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import io.github.whitemagic2014.tts.bean.TransRecord;
import io.github.whitemagic2014.tts.bean.Voice;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.concurrent.BasicThreadFactory;

import java.io.File;
import java.net.URISyntaxException;
Expand All @@ -17,7 +18,13 @@
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Stream;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.ThreadPoolExecutor.CallerRunsPolicy;
import java.util.concurrent.TimeUnit;

public class TTS {

Expand Down Expand Up @@ -66,6 +73,9 @@ public class TTS {
*/
private final Map<Long, TTSWebsocket> websocketMap = new ConcurrentHashMap<>();


private ExecutorService executor;

public TTS(Voice voice) {
this(voice, null);
}
Expand Down Expand Up @@ -166,8 +176,22 @@ public String trans() {

public void batchTrans() {
init();
Stream<TransRecord> stream = parallelThreadSize > 1 ? recordList.parallelStream() : recordList.stream();
stream.forEach(record -> doTrans(record.getContent(), record.getFilename()));
if (parallelThreadSize > 1) {
List<Future<?>> futureList = new ArrayList<>();
for (TransRecord record : recordList) {
Future<String> future = executor.submit(() -> doTrans(record.getContent(), record.getFilename()));
futureList.add(future);
}
for (Future<?> future : futureList) {
try {
future.get();
} catch (InterruptedException | ExecutionException e) {
throw new IllegalStateException(e);
}
}
} else {
recordList.forEach(record -> doTrans(record.getContent(), record.getFilename()));
}
}

/**
Expand Down Expand Up @@ -208,8 +232,11 @@ private void init() {
if (!storageFolder.exists()) {
storageFolder.mkdirs();
}
if (parallelThreadSize > 1) {
System.setProperty("java.util.concurrent.ForkJoinPool.common.parallelism", String.valueOf(parallelThreadSize));
if (executor == null && parallelThreadSize > 1) {
executor = new ThreadPoolExecutor(0, parallelThreadSize, 1, TimeUnit.MINUTES,
new LinkedBlockingQueue<>(parallelThreadSize),
new BasicThreadFactory.Builder().daemon(true).namingPattern("trans-worker-%d").build(),
new CallerRunsPolicy());
}
}

Expand Down