第一种webSocket的WebSocketConfig.java类
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.concurrent.ConcurrentTaskScheduler;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import java.util.concurrent.Executors;
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
@Autowired
FlowWebSocketHandlerWs flowWebSocketHandlerWs;
@Autowired
FlowWebSocketDoorStatusWs flowWebSocketDoorStatusWs;
@Autowired
FlowWebSocketAttachmentWs flowWebSocketAttachmentWs;
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(flowWebSocketHandlerWs, "/doorLog/ws")
.addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*");
registry.addHandler(flowWebSocketDoorStatusWs, "/doorStatus/ws")
.addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*");
registry.addHandler(flowWebSocketAttachmentWs, "/attatchment/ws")
.addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*");
}
@Bean
public TaskScheduler taskScheduler() {
return new ConcurrentTaskScheduler(Executors.newSingleThreadScheduledExecutor());
}
}
第二种sockJs的WebSocketConfig.java类
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.concurrent.ConcurrentTaskScheduler;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import java.util.concurrent.Executors;
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
@Autowired
FlowWebSocketHandlerWs flowWebSocketHandlerWs;
@Autowired
FlowWebSocketDoorStatusWs flowWebSocketDoorStatusWs;
@Autowired
FlowWebSocketAttachmentWs flowWebSocketAttachmentWs;
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(flowWebSocketHandlerWs, "/doorLog/ws")
.addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*").withSockJS();
registry.addHandler(flowWebSocketDoorStatusWs, "/doorStatus/ws")
.addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*").withSockJS();
registry.addHandler(flowWebSocketAttachmentWs, "/attatchment/ws")
.addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*").withSockJS();
}
@Bean
public TaskScheduler taskScheduler() {
return new ConcurrentTaskScheduler(Executors.newSingleThreadScheduledExecutor());
}
}
webSocked拦截器WebSocketHandshakeInterceptor类
@Slf4j
public class WebSocketHandshakeInterceptor extends HttpSessionHandshakeInterceptor {
@Override
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler handler, Map<String, Object> attr) throws Exception {
String ip = request.getRemoteAddress().getHostString();
if (ip.indexOf(".0.0.1") > -1 || ip.indexOf(":0:0:1") > -1) {
try {
ip = InetAddress.getLocalHost().getHostAddress();
} catch (UnknownHostException e) {
return false;
}
}
User user = UserService.ipAndUserHashTable.get(ip);
if (user != null) {
return true;
} else {
user = UserService.ALL_ALIVE_USER_FOR_PHONE.get(ip);
if (user != null) {
return true;
}
}
return false;
}
@Override
public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler handler, Exception e) {
log.info("webSocket======握手成功");
}
}
FlowWebSocketHandlerWs类
@Component
public class FlowWebSocketHandlerWs extends TextWebSocketHandler {
private List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
@Override
public void afterConnectionEstablished(WebSocketSession session) {
sessions.add(session);
}
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException {
}
}
FlowWebSocketDoorStatusWs类
@Component
public class FlowWebSocketDoorStatusWs extends TextWebSocketHandler {
private List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
@Override
public void afterConnectionEstablished(WebSocketSession session) {
sessions.add(session);
}
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException {
}
}
FlowWebSocketAttachmentWs类
@Component
public class FlowWebSocketAttachmentWs extends TextWebSocketHandler {
private List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
@Override
public void afterConnectionEstablished(WebSocketSession session) {
sessions.add(session);
}
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException {
}
}