package cc.mrbird.febs.websocket;
|
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
import org.springframework.scheduling.TaskScheduler;
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
|
import org.springframework.web.socket.config.annotation.EnableWebSocket;
|
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
|
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
|
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
|
|
import javax.annotation.Resource;
|
|
/**
|
*
|
* @author xxx
|
* @date 2020-09-01
|
**/
|
@Configuration
|
@EnableWebSocket
|
public class WebSocketConfig implements WebSocketConfigurer {
|
|
@Resource
|
private HttpAuthHandler httpAuthHandler;
|
@Resource
|
private WsAuthInterceptor wsAuthInterceptor;
|
|
@Override
|
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
|
registry.addHandler(httpAuthHandler, "wsxg")
|
.addInterceptors(wsAuthInterceptor)
|
.setAllowedOrigins("*");
|
}
|
|
@Bean
|
public TaskScheduler taskScheduler() {
|
/*
|
不显式配置TaskScheduler，Spring会使用默认的TaskScheduler实现类——ScheduledThreadPoolExecutor。
|
这个实现类使用一个线程池来执行任务，线程池的大小默认为1。
|
也就是说，如果不配置TaskScheduler，所有的定时任务都会在同一个线程中执行，可能会导致任务执行时间过长或者任务之间相互影响。
|
因此，为了更好地控制定时任务的执行，建议显式配置TaskScheduler，并根据具体需求设置线程池大小等属性。
|
这样可以确保定时任务的精确调度和高效执行。
|
*/
|
ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
|
taskScheduler.setPoolSize(10);
|
taskScheduler.initialize();
|
return taskScheduler;
|
}
|
/**
|
* 注入一个ServerEndpointExporter,该Bean会自动注册使用@ServerEndpoint注解申明的websocket endpoint
|
*/
|
@Bean
|
public ServerEndpointExporter serverEndpointExporter() {
|
return new ServerEndpointExporter();
|
}
|
}
|
