WebSocket（后台代码）
WebSocket（后台代码）
⼀、WebSocket简单介绍
传统的HTTP协议是⽆状态的，每次请求（request）都要由客户端（如 浏览器）主动发起，服务端进⾏处理后返回response结果，⽽服务端很难主动向客户端发送数据；这种客户端是主动⽅，服务端是被动⽅的传统Web模式 对于信息变化不频繁的Web应⽤来说造成的⿇烦较⼩，⽽对于涉及实时信息的Web应⽤却带来了很⼤的不便，如带有即时通信、实时数据、订阅推送等功能的应 ⽤。在WebSocket规范提出之前，开发⼈员若要实现这些实时性较强的功能，经常会使⽤折衷的解决⽅法：轮询（polling）和Comet技术。其实后者本质上也是⼀种轮询，只不过有所改进。
  轮询是最原始的实现实时Web应⽤的解决⽅案。轮询技术要求客户端以设定的时间间隔周期性地向服务端发送请求，频繁地查询是否有新的数据改动。明显地，这种⽅法会导致过多不必要的请求，浪费流量和服务器资源。
  Comet技术⼜可以分为长轮询和流技术。长轮询改进了上述的轮询技术，减⼩了⽆⽤的请求。它会为某些数据设定过期时间，当数据过期后才会向服务端发送请求；这种机制适合数据的改动不是特别频繁的
情况。流技术通常是指客户端使⽤⼀个隐藏的窗⼝与服务端建⽴⼀个HTTP长连接，服务端会不断更新连接状态以保持HTTP长连接存活；这样的话，服务端就可以通过这条长连接主动将数据发送给客户端；流技术在⼤并发环境下，可能会考验到服务端的性能。
  这两种技术都是基于请求-应答模式，都不算是真正意义上的实时技术；它们的每⼀次请求、应答，都浪费了⼀定流量在相同的头部信息上，并且开发复杂度也较⼤。
  JavaEE 7中出了JSR-356:Java API for WebSocket规范。Tomcat,Nginx,Jetty等都⽀持WebSocket。Tomcat从7.0.27开始⽀持WebSocket，从7.0.47开始⽀持JSR-356。
⼆、代码
依赖
<!-- 这⾥我⽤的Spring Boot 1.5.9  ⾥⾯集成了WebSocket -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<!-- 如果版本太低没有集成就需要单独添加 WebSocket 的依赖 -->
<!-- <dependency>
<groupId>javax</groupId>
<artifactId>javaee-api</artifactId>
<version>7.0</version>
<scope>provided</scope>
</dependency> -->
<!-- GSON 依赖 List 转 String ⽤到了（不⽤可不添加） -->
<dependency>
<groupId&le.code.gson</groupId>
<artifactId>gson</artifactId>
<version>2.8.2</version>
</dependency>
代码
@Component
@ServerEndpoint(value ="/websocket")
public class WebSocket {
private static final Logger log = Logger(WebSocket.class);
//静态变量，⽤来记录当前在线连接数。应该把它设计成线程安全的。
private static int onlineCount =0;
//concurrent包的线程安全Set，⽤来存放每个客户端对应的MyWebSocket对象。
private static CopyOnWriteArraySet<WebSocket> webSocketSet =new CopyOnWriteArraySet<WebSocket>();
//与某个客户端的连接会话，需要通过它来给客户端发送数据
private Session session;
//全局变量的返回值（这个是我DAO的返回类型（⾃定义的 List ，可⾃⼰定义或者去掉））
private static List<TerminalStateDTO> terminalStateDTOS;
private static List<TerminalStateDTO> terminalStateDTOS;
//注解调⽤DAO 不起效需要⽤⼯具类调⽤DAO
/*@Autowired
private LifeboatEquipmentMapper lifeboatEquipmentMapper;*/
//WebSocket调⽤DAO（WebSocket只能通过⼯具类获取DAO）
private LifeboatEquipmentMapper lifeboatEquipmentMapper = ApplicationContext().getBean(LifeboatEquipmentMapper.c lass);
/**
* 连接建⽴成功调⽤的⽅法
*/
@OnOpen
public void onOpen(Session session){
log.info("有新连接加⼊！当前在线⼈数为"+getOnlineCount());
try{
//连接上之后如果是空的就查询⼀次
if(terminalStateDTOS == null){
terminalStateDTOS = lifeboatEquipmentMapper.lifeboatDNR();
}
this.session = session;
webSocketSet.add(this);//加⼊set中
addOnlineCount();//在线数加1
sendMessage(objToJson(terminalStateDTOS));//发⽅法
}catch(IOException e){
<("websocket⾥⾯onOpen异常", e);
}
}
// //连接打开时执⾏
// @OnOpen
/
/ public void onOpen(@PathParam("user") String user, Session session) {
//  currentUser = user;
//  System.out.println("Connected ... " + Id());
// }
/**
* 连接关闭调⽤的⽅法
*/
@OnClose
public void onClose(){
try{
subOnlineCount();//在线数减1
log.info("有⼀连接关闭！当前在线⼈数为"+getOnlineCount());
}catch(Exception e){
e.printStackTrace();
<("websocket⾥⾯onClose异常", e);
}
}
/**
* 收到客户端消息后调⽤的⽅法
*
* @param message 客户端发送过来的消息
*/
@OnMessage
public void onMessage(String message, Session session){
try{
log.info("来⾃客户端的消息:"+ message);
//返回值为空的时候查询⼀次
if(terminalStateDTOS == null){
//通过⼯具类调⽤DAO
terminalStateDTOS = lifeboatEquipmentMapper.lifeboatDNR();
}
//发消息
/
/发消息
for(WebSocket item : webSocketSet){
//发接收的是 String 需要将List转换成String（如果是字符串就不需要转换）                item.sendMessage(objToJson(terminalStateDTOS));
}
}catch(IOException e){
e.printStackTrace();
<("websocket⾥⾯onMessage异常", e);
}
}
/**
* @param session
* @param error
*/
@OnError
public void onError(Session session, Throwable error){
try{
<("发⽣错误");
error.printStackTrace();
}catch(Exception e){
e.printStackTrace();
<("websocket⾥⾯onError异常", e);
}
}
/**
* 收到客户端消息后调⽤的⽅法
* @param message 客户端发送过来的消息
*/
//    @param session 可选的参数
public void sendMessage(String message)throws IOException {
try{
BasicRemote().sendText(message);
}catch(IOException e){
e.printStackTrace();
<("websocket⾥⾯sendMessage异常", e);
}
}
/**
* 发⾃定义消息（暂时未⽤到）
*/
public static void sendInfo(String message)throws IOException {
try{
for(WebSocket item : webSocketSet){
try{
item.sendMessage(message);
}catch(IOException e){
continue;
}
}
}catch(Exception e){
e.printStackTrace();
<("websocket⾥⾯sendInfo异常", e);
}
}
public static synchronized int getOnlineCount(){
return onlineCount;
}
public static synchronized void addOnlineCount(){
}
public static synchronized void subOnlineCount(){
}
/**
* 对象转json
*
* @param obj 对象
* @return json字符串
*/
public static String objToJson(Object obj){
return new GsonBuilder().serializeNulls().create().toJson(obj);
}
}
配置（在本地运⾏的时候需要添加 打成WAR包时需要注释掉）
//在本地运⾏的时候需要添加配置（打成WAR包时需要注释掉）
@Configuration
public class WebSocketConfig {
/**
* ServerEndpointExporter 作⽤在线代码运行器
*
* 这个Bean会⾃动注册使⽤@ServerEndpoint注解声明的websocket endpoint
*
* @return
*/
@Bean
public ServerEndpointExporter serverEndpointExporter(){
return new ServerEndpointExporter();
}
}
WebSocket 调⽤DAO⼯具类代码
/**
* WebSocket 调⽤DAO的⼯具
*/
@Component
@Lazy(false)
public class ApplicationContextRegister implements ApplicationContextAware {
private static ApplicationContext APPLICATION_CONTEXT;
@Override
public void setApplicationContext(ApplicationContext applicationContext)throws BeansException {        APPLICATION_CONTEXT = applicationContext;
}
public static ApplicationContext getApplicationContext(){
return APPLICATION_CONTEXT;
}
}
测试

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。