Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refine executor description in WebSocketMessageBrokerStats

Browse Source 
Closes gh-33104
This commit is contained in:
rstoyanchev
2024-07-12 17:35:21 +01:00
parent bd31e8dacc
commit 007a347ade
2 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
spring-websocket/src/main/java/org/springframework/web/socket/config/WebSocketMessageBrokerStats.java
View File

@@ -30,6 +30,7 @@ import org.springframework.beans.factory.SmartInitializingSingleton;
import org.springframework.core.task.TaskExecutor;
import org.springframework.lang.Nullable;
import org.springframework.messaging.simp.stomp.StompBrokerRelayMessageHandler;
import org.springframework.scheduling.SchedulingTaskExecutor;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
@@ -239,9 +240,15 @@ public class WebSocketMessageBrokerStats implements SmartInitializingSingleton {
if (this.sockJsTaskScheduler == null) {
return "null";
}
if (this.sockJsTaskScheduler instanceof ThreadPoolTaskScheduler threadPoolTaskScheduler) {
return getExecutorStatsInfo(threadPoolTaskScheduler.getScheduledThreadPoolExecutor());
if (!(this.sockJsTaskScheduler instanceof SchedulingTaskExecutor)) {
return "thread-per-task";
}
if (this.sockJsTaskScheduler instanceof ThreadPoolTaskScheduler tpts) {
return getExecutorStatsInfo(tpts.getScheduledThreadPoolExecutor());
}
return "unknown";
}
@@ -250,8 +257,12 @@ public class WebSocketMessageBrokerStats implements SmartInitializingSingleton {
return "null";
}
if (executor instanceof ThreadPoolTaskExecutor threadPoolTaskScheduler) {
executor = threadPoolTaskScheduler.getThreadPoolExecutor();
if (!(executor instanceof SchedulingTaskExecutor) && (executor instanceof TaskExecutor)) {
return "thread-per-task";
}
if (executor instanceof ThreadPoolTaskExecutor tpte) {
executor = tpte.getThreadPoolExecutor();
}
if (executor instanceof ThreadPoolExecutor) {