spring-websocket+Java客户端的实现_浮梦丶天拓_新浪博客

1.业务场景描述

这个案例主要是用来监控用的，就是把接口中的数据实时的展示到页面上，比如一个下订单的接口，在外界访问订单接口时，订单存入到数据库后需要将订单数据展示到页面上。因此就可以在接口中使用Java客户端将订单信息发送到websocket服务器，然后页面作为客户端把接口订单数据暂时到页面上。在接口中将Java客户端的连接传入接口中，直接通过连接发送订单信息到服务器。连接websocket的客户端代码通过静态代码块的方式加载，然后将建立连接后的websocket连接存到静态变量，在接口中通过静态变量，就可以获取websocket连接，进而达到在接口中发送订单信息的目的。

2.所需jar包

服务端：spring-websocket，spring-messaging

     org.springframework spring-websocket ${spring.version} org.springframework spring-messaging ${spring.version}

客户端：javax.websocket-api，tyrus-standalone-client

            javax.websocket javax.websocket-api 1.1 org.glassfish.tyrus.bundles tyrus-standalone-client 1.9

3.websocket配置文件

import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.web.servlet.config.annotation.EnableWebMvc;import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;import org.springframework.web.socket.WebSocketHandler;import org.springframework.web.socket.config.annotation.EnableWebSocket;import org.springframework.web.socket.config.annotation.WebSocketConfigurer;import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;@Configuration@EnableWebMvc@EnableWebSocketpublic class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer { //添加处理器和拦截器，处理器后面的地址就是websocket的访问路径 //setAllowedOrigins：指定的域名或IP，如果不限制使用"*"就可以了@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(firstWebSocketHandler(), "/websocket/demo") .addInterceptors(myInterceptor()).setAllowedOrigins("*");} //直接注入自己定义的websocket处理器@Beanpublic WebSocketHandler firstWebSocketHandler(){ return new FirstWebSocketHandler();} //直接注入自己定义的websocket拦截器@Beanpublic WebSocketHandshakeInterceptor myInterceptor(){ return new WebSocketHandshakeInterceptor();}}

4.websocket拦截器

import org.springframework.http.server.ServerHttpRequest;import org.springframework.http.server.ServerHttpResponse;import org.springframework.web.socket.WebSocketHandler;import org.springframework.web.socket.server.HandshakeInterceptor;import java.util.Map;public class WebSocketHandshakeInterceptor implements HandshakeInterceptor { @Overridepublic void afterHandshake(ServerHttpRequest arg0, ServerHttpResponse arg1, WebSocketHandler arg2, Exception arg3) { } @Overridepublic boolean beforeHandshake(ServerHttpRequest arg0, ServerHttpResponse arg1, WebSocketHandler arg2,Map, Object> arg3) throws Exception { return true;}}

5.websocket处理器

import org.springframework.web.socket.*;import java.io.IOException;import java.util.ArrayList;import java.util.Collections;import java.util.List;public class FirstWebSocketHandler implements WebSocketHandler { //定义一个全局的初始化值count=0private static int count = 0; private static String data = "0.25435"; private volatile static List sessions = Collections.synchronizedList(new ArrayList()); private WebSocketSession session;//连接关闭后的操作@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception { this.session = session;sessions.remove(this.session);count = count-1;System.out.println(count);} //建立连接后的操作@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception { new Thread(new Runnable() { @Overridepublic void run() { count++;System.out.println(count);System.out.println("服务端"+data); try { session.sendMessage(new TextMessage(data));} catch (IOException e) { e.printStackTrace();}// while(true){// //向前端推送数据，一秒一次推送，格式：1,2,3,4,5.....// try {// System.out.println("服务端"+data);// session.sendMessage(new TextMessage(data));// Thread.sleep(1000);// } catch (Exception e) {// e.printStackTrace();// break;// }//// }} }).start();} //消息处理，在客户端通过Websocket API发送的消息会经过这里，然后进行相应的处理@Overridepublic void handleMessage(WebSocketSession session, WebSocketMessage