SpringWebSocketConfig配置

package com.meeno.chemical.socket.task.config;import com.meeno.chemical.socket.task.handler.TaskProgressWebSocketHandler;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;/*** @description: SpringBoot WebSocket config* @author: Wzq* @create: 2020-07-16 10:50*/
@Configuration
@EnableWebSocket
public class SpringWebSocketConfig implements WebSocketConfigurer {@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {// 注册自定义消息处理，消息路径为`/ws/taskProgress`registry.addHandler(new TaskProgressWebSocketHandler(),"/ws/taskProgress").setAllowedOrigins("*");}
}

TaskProgressWebSocketHandler

package com.meeno.chemical.socket.task.handler;import org.springframework.http.HttpHeaders;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.adapter.standard.StandardWebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;import java.io.IOException;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;/*** @description: 任务进度WebSocketHandler* @author: Wzq* @create: 2020-07-16 10:51*/
public class TaskProgressWebSocketHandler extends TextWebSocketHandler {/***  concurrent包的线程安全Map，用来存放每个客户端对应的MyWebSocket对象。*/public final static ConcurrentHashMap<String, WebSocketSession> webSocketMap = new ConcurrentHashMap<String, WebSocketSession>();/*** 处理消息* @param session* @param message* @throws Exception*/@Overrideprotected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {//在这里自定义消息处理...TextMessage textMessage = new TextMessage("${message body}");session.sendMessage(textMessage);}/*** 连接建立后* @param session* @throws Exception*/@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {StandardWebSocketSession standardWebSocketSession = (StandardWebSocketSession) session;List<String> taskNameList = standardWebSocketSession.getNativeSession().getRequestParameterMap().get("taskName");String taskName = taskNameList.get(0);//保存所有会话webSocketMap.put(taskName,session);}/*** 连接关闭后* @param session* @param status* @throws Exception*/@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {if(!webSocketMap.isEmpty()){// remove session after closedWebSocketSession webSocketSession = webSocketMap.get(session.getId());if(webSocketSession != null){webSocketMap.remove(session.getId());}}}/*** 发送消息给所有人* @param content*/public static void sendMessageAll(String content){webSocketMap.forEach((taskNameKey,session) ->{TextMessage textMessage = new TextMessage(content);try {session.sendMessage(textMessage);} catch (IOException e) {e.printStackTrace();}});}/*** 发送消息给指定某个任务* @param taskName* @param content*/public static void sendMessage(String taskName,String content){webSocketMap.forEach((taskNameKey,session) ->{if(taskName.equals(taskNameKey)){TextMessage textMessage = new TextMessage(content);try {session.sendMessage(textMessage);} catch (IOException e) {e.printStackTrace();}}});}}

问题

启用是会和定时任务冲突
需要配置
ScheduledConfig.java

package com.meeno.chemical.common.scheduling.config;import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;/*** @description: 任务config* @author: Wzq* @create: 2020-07-16 11:05*/
@Configuration
public class ScheduledConfig {@Beanpublic TaskScheduler taskScheduler(){ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();taskScheduler.setPoolSize(10);taskScheduler.initialize();return taskScheduler;}}