一.服務層

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 sessions = new ConcurrentLinkedQueue();

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 iterator = sessions.iterator();

while (iterator.hasNext()) {

WebSocketSession session = iterator.next();

synchronized(session){

if(session.isOpen()){

session.sendMessage(new TextMessage(message));

}else{

iterator.remove();

}

}

}

}

}

}

二.前臺界面監聽

Insert title here

websocket測試界面

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) {

}

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

spring-servlet.xml空文件

spring

org.springframework.web.servlet.DispatcherServlet

contextConfigLocation

/WEB-INF/spring-servlet.xml

1

spring

*.ws

以上所述是小編給大家介紹的Java中websocket消息推送的實現代碼，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對聚米學院網站的支持！

總結

以上是生活随笔為你收集整理的java socket 推送机制_Java中websocket消息推送的实现代码的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。