Java中websocket消息推送的实现代码

更新时间：2017年02月13日 11:40:27 作者：tianzhongshan

这篇文章主要介绍了Java中websocket消息推送的实现代码,非常不错，具有参考借鉴价值，需要的朋友可以参考下

一.服务层

package com.demo.websocket;
import java.io.IOException;
import java.util.Iterator;
import java.util.concurrent.ConcurrentLinkedQueue;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
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.handler.TextWebSocketHandler;
@Configuration
@EnableWebSocket
public class websocketListener implements WebSocketConfigurer, ServletContextListener{
  private ConcurrentLinkedQueue<WebSocketSession> sessions = new ConcurrentLinkedQueue<WebSocketSession>();
  private WebSocketHandlerTest handler;
  @Override
  public void contextDestroyed(ServletContextEvent arg0) {
    // TODO Auto-generated method stub
  }
  @Override
  public void contextInitialized(ServletContextEvent arg0) {
    // TODO Auto-generated method stub
  }
  @Override
  public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
    handler = new WebSocketHandlerTest();
    registry.addHandler(handler, "/ws/notifymessage.ws");
    registry.addHandler(handler, "/ws/sockjs/notifymessage.ws").withSockJS();
    new Thread(handler).start();
  }
  class WebSocketHandlerTest extends TextWebSocketHandler implements Runnable{
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
      sessions.remove(session);
    }
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
      sessions.add(session);
    }
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
      super.handleTextMessage(session, message);
    }
    @Override
    public void run() {
      System.out.println("等待推送....");
      try {
        int i = 0;
        for (;;) {
          synchronized (this) {
            try {
              Thread.sleep(3000);
            } catch (InterruptedException e) {
              // TODO Auto-generated catch block
              e.printStackTrace();
            }
          }
          if(i%10==0){
            nofity("消息推送测试......");
            System.out.println("推送消息了....");
          }else{
            System.out.println("本次不推送消息了....");
          }
          i++;
        }
      } catch (IOException e) {
        e.printStackTrace();
        System.out.println("失败....");
      }
    }
    private void nofity(String message) throws IOException {
      Iterator<WebSocketSession> iterator = sessions.iterator();
      while (iterator.hasNext()) {
        WebSocketSession session = iterator.next();
        synchronized(session){
          if(session.isOpen()){
            session.sendMessage(new TextMessage(message));
          }else{
            iterator.remove();
          }
        }
      }
    }
  }
}

二.前台界面监听

<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title>Insert title here</title>
</head>
<body>
websocket测试界面
</body>
<script type="text/javascript">
var websocketPath = "localhost:8080/demo-web";
if ('WebSocket' in window) {
  websocket = new WebSocket("ws://"+websocketPath+"/ws/notifymessage.ws");
} else if ('MozWebSocket' in window) {
  websocket = new MozWebSocket("ws://"+websocketPath+"/ws/notifymessage.ws");
} else {
  websocket = new SockJS("ws://"+websocketPath+"/ws/notifymessage.ws");
}
websocket.onopen = function (evnt) {
};
websocket.onmessage = function (evnt) {
  console.log(evnt);
};
websocket.onerror = function (evnt) {
};
websocket.onclose = function (evnt) {
}
</script>
</html>

注意web.xml中配置DispatcherServlet控制器

spring-servlet.xml空文件

<servlet>
    <servlet-name>spring</servlet-name>
    <servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
    <init-param>
      <param-name>contextConfigLocation</param-name>
      <param-value>/WEB-INF/spring-servlet.xml</param-value>
    </init-param>
    <load-on-startup>1</load-on-startup>
  </servlet>
  <servlet-mapping>
    <servlet-name>spring</servlet-name>
    <url-pattern>*.ws</url-pattern>
  </servlet-mapping>

以上所述是小编给大家介绍的Java中websocket消息推送的实现代码，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对脚本之家网站的支持！

您可能感兴趣的文章:

Java8 实现stream将对象集合list中抽取属性集合转化为map或list
这篇文章主要介绍了Java8 实现stream将对象集合list中抽取属性集合转化为map或list的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-02-02
maven如何利用springboot的配置文件进行多个环境的打包
这篇文章主要介绍了maven如何利用springboot的配置文件进行多个环境的打包,在Spring Boot中多环境配置文件名需要满足application-{profiles.active}.properties的格式，其中{profiles.active}对应你的环境标识，本文给大家详细讲解，需要的朋友可以参考下
2023-02-02
jdk1.8的环境配置过程
Java平台由Java虚拟机和Java应用程序接口搭建，Java语言则是进入这个平台的通道，用Java语言编写并编译的程序可以运行在这个平台上，本文给大家讲解jdk1.8的环境配置过程，感兴趣的朋友跟随小编一起看看吧
2022-11-11
Java8 HashMap扩容算法实例解析
这篇文章主要介绍了Java8 HashMap扩容算法实例解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-12-12
springcloud使用feign调用服务时参数内容过大问题
这篇文章主要介绍了springcloud使用feign调用服务时参数内容过大问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-03-03
Spring中基于xml的AOP的详细步骤
这篇文章主要介绍了Spring中基于xml的AOP的详细步骤，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2020-06-06
Java实现无向图的示例详解
边没有方向的图称为无向图，直观来说，若一个图中每条边都是无方向的，则称为无向图。本文将通过示例详细讲解Java如何实现无向图，需要的可以参考一下
2022-04-04
java.lang.UnsupportedOperationException分析及解决办法
日常开发中我遇到java.lang.UnsupportedOperationException:异常两次了,下面这篇文章主要给对大家介绍了关于java.lang.UnsupportedOperationException分析及解决办法,需要的朋友可以参考下
2024-03-03
IDEA actiBPM插件安装图文教程
这篇文章主要介绍了IDEA actiBPM插件安装教程,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-09-09
java 中 MD5加密的实例
这篇文章主要介绍了java 中 MD5加密的实例的相关资料,通过本文希望能帮助到大家，需要的朋友可以参考下
2017-09-09