国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 開發(fā) > Java > 正文

SpringMVC整合websocket實現(xiàn)消息推送及觸發(fā)功能

2024-07-14 08:39:53
字體:
供稿:網(wǎng)友

本文為大家分享了SpringMVC整合websocket實現(xiàn)消息推送,供大家參考,具體內(nèi)容如下

1.創(chuàng)建websocket握手協(xié)議的后臺

(1)HandShake的實現(xiàn)類

/**  *Project Name: price  *File Name:  HandShake.java  *Package Name: com.yun.websocket  *Date:     2016年9月3日 下午4:44:27  *Copyright (c) 2016,578888218@qq.com All Rights Reserved. */  package com.yun.websocket;  import java.util.Map;  import org.springframework.http.server.ServerHttpRequest; import org.springframework.http.server.ServerHttpResponse; import org.springframework.http.server.ServletServerHttpRequest; import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.server.HandshakeInterceptor;  /**  *Title:   HandShake<br/>  *Description:  *@Company:  青島勵圖高科<br/>  *@author:  劉云生  *@version:  v1.0  *@since:   JDK 1.7.0_80  *@Date:   2016年9月3日 下午4:44:27 <br/> */ public class HandShake implements HandshakeInterceptor{    @Override   public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,       Map<String, Object> attributes) throws Exception {     // TODO Auto-generated method stub     String jspCode = ((ServletServerHttpRequest) request).getServletRequest().getParameter("jspCode");     // 標記用戶     //String userId = (String) session.getAttribute("userId");     if(jspCode!=null){       attributes.put("jspCode", jspCode);     }else{       return false;     }     return true;   }    @Override   public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,       Exception exception) {     // TODO Auto-generated method stub        }  } 

(2)MyWebSocketConfig的實現(xiàn)類

/**  *Project Name: price  *File Name:  MyWebSocketConfig.java  *Package Name: com.yun.websocket  *Date:     2016年9月3日 下午4:52:29  *Copyright (c) 2016,578888218@qq.com All Rights Reserved. */  package com.yun.websocket;  import javax.annotation.Resource;  import org.springframework.stereotype.Component; import org.springframework.web.servlet.config.annotation.EnableWebMvc; import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter; import org.springframework.web.socket.config.annotation.EnableWebSocket; import org.springframework.web.socket.config.annotation.WebSocketConfigurer; import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;  /**  *Title:   MyWebSocketConfig<br/>  *Description:  *@Company:  青島勵圖高科<br/>  *@author:  劉云生  *@version:  v1.0  *@since:   JDK 1.7.0_80  *@Date:   2016年9月3日 下午4:52:29 <br/> */ @Component @EnableWebMvc @EnableWebSocket public class MyWebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer{   @Resource   MyWebSocketHandler handler;      @Override   public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {     // TODO Auto-generated method stub     registry.addHandler(handler, "/wsMy").addInterceptors(new HandShake());     registry.addHandler(handler, "/wsMy/sockjs").addInterceptors(new HandShake()).withSockJS();   }  } 

(3)MyWebSocketHandler的實現(xiàn)類

/**  *Project Name: price  *File Name:  MyWebSocketHandler.java  *Package Name: com.yun.websocket  *Date:     2016年9月3日 下午4:55:12  *Copyright (c) 2016,578888218@qq.com All Rights Reserved. */  package com.yun.websocket;  import java.io.IOException; import java.util.HashMap; import java.util.Iterator; import java.util.Map; import java.util.Map.Entry;  import org.springframework.stereotype.Component; import org.springframework.web.socket.CloseStatus; import org.springframework.web.socket.TextMessage; import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.WebSocketMessage; import org.springframework.web.socket.WebSocketSession;  import com.google.gson.GsonBuilder;  /**  *Title:   MyWebSocketHandler<br/>  *Description:  *@Company:  青島勵圖高科<br/>  *@author:  劉云生  *@version:  v1.0  *@since:   JDK 1.7.0_80  *@Date:   2016年9月3日 下午4:55:12 <br/> */ @Component public class MyWebSocketHandler implements WebSocketHandler{    public static final Map<String, WebSocketSession> userSocketSessionMap;    static {     userSocketSessionMap = new HashMap<String, WebSocketSession>();   }         @Override   public void afterConnectionEstablished(WebSocketSession session) throws Exception {     // TODO Auto-generated method stub     String jspCode = (String) session.getHandshakeAttributes().get("jspCode");     if (userSocketSessionMap.get(jspCode) == null) {       userSocketSessionMap.put(jspCode, session);     }     for(int i=0;i<10;i++){       //broadcast(new TextMessage(new GsonBuilder().create().toJson("/"number/":/""+i+"/"")));       session.sendMessage(new TextMessage(new GsonBuilder().create().toJson("/"number/":/""+i+"/"")));     }   }    @Override   public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {     // TODO Auto-generated method stub     //Message msg=new Gson().fromJson(message.getPayload().toString(),Message.class);     //msg.setDate(new Date()); //   sendMessageToUser(msg.getTo(), new TextMessage(new GsonBuilder().setDateFormat("yyyy-MM-dd HH:mm:ss").create().toJson(msg)));          session.sendMessage(message);   }    @Override   public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {     // TODO Auto-generated method stub     if (session.isOpen()) {       session.close();     }     Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap         .entrySet().iterator();     // 移除Socket會話     while (it.hasNext()) {       Entry<String, WebSocketSession> entry = it.next();       if (entry.getValue().getId().equals(session.getId())) {         userSocketSessionMap.remove(entry.getKey());         System.out.println("Socket會話已經(jīng)移除:用戶ID" + entry.getKey());         break;       }     }   }    @Override   public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {     // TODO Auto-generated method stub     System.out.println("Websocket:" + session.getId() + "已經(jīng)關(guān)閉");     Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap         .entrySet().iterator();     // 移除Socket會話     while (it.hasNext()) {       Entry<String, WebSocketSession> entry = it.next();       if (entry.getValue().getId().equals(session.getId())) {         userSocketSessionMap.remove(entry.getKey());         System.out.println("Socket會話已經(jīng)移除:用戶ID" + entry.getKey());         break;       }     }   }    @Override   public boolean supportsPartialMessages() {     // TODO Auto-generated method stub     return false;   }   /**    * 群發(fā)    * @Title:    broadcast     * @Description: TODO     * @param:    @param message    * @param:    @throws IOException    * @return:   void    * @author:   劉云生    * @Date:    2016年9月10日 下午4:23:30     * @throws    */   public void broadcast(final TextMessage message) throws IOException {     Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap         .entrySet().iterator();      // 多線程群發(fā)     while (it.hasNext()) {        final Entry<String, WebSocketSession> entry = it.next();        if (entry.getValue().isOpen()) {         new Thread(new Runnable() {            public void run() {             try {               if (entry.getValue().isOpen()) {                 entry.getValue().sendMessage(message);               }             } catch (IOException e) {               e.printStackTrace();             }           }          }).start();       }      }   }      /**    * 給所有在線用戶的實時工程檢測頁面發(fā)送消息    *    * @param message    * @throws IOException    */   public void sendMessageToJsp(final TextMessage message,String type) throws IOException {     Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap         .entrySet().iterator();      // 多線程群發(fā)     while (it.hasNext()) {        final Entry<String, WebSocketSession> entry = it.next();       if (entry.getValue().isOpen() && entry.getKey().contains(type)) {         new Thread(new Runnable() {            public void run() {             try {               if (entry.getValue().isOpen()) {                 entry.getValue().sendMessage(message);               }             } catch (IOException e) {               e.printStackTrace();             }           }          }).start();       }      }   } } 

2.創(chuàng)建websocket握手處理的前臺

<script>   var path = '<%=basePath%>';   var userId = 'lys';   if(userId==-1){     window.location.href="<%=basePath2%>" rel="external nofollow" ;   }   var jspCode = userId+"_AAA";   var websocket;   if ('WebSocket' in window) {     websocket = new WebSocket("ws://" + path + "wsMy?jspCode=" + jspCode);   } else if ('MozWebSocket' in window) {     websocket = new MozWebSocket("ws://" + path + "wsMy?jspCode=" + jspCode);   } else {     websocket = new SockJS("http://" + path + "wsMy/sockjs?jspCode=" + jspCode);   }   websocket.onopen = function(event) {     console.log("WebSocket:已連接");     console.log(event);   };   websocket.onmessage = function(event) {     var data = JSON.parse(event.data);     console.log("WebSocket:收到一條消息-norm", data);     alert("WebSocket:收到一條消息");   };   websocket.onerror = function(event) {     console.log("WebSocket:發(fā)生錯誤 ");     console.log(event);   };   websocket.onclose = function(event) {     console.log("WebSocket:已關(guān)閉");     console.log(event);   } </script> 

3.通過Controller調(diào)用進行websocket的后臺推送

/**  *Project Name: price  *File Name:  GarlicPriceController.java  *Package Name: com.yun.price.garlic.controller  *Date:     2016年6月23日 下午3:23:46  *Copyright (c) 2016,578888218@qq.com All Rights Reserved. */  package com.yun.price.garlic.controller;  import java.io.IOException; import java.util.Date; import java.util.List;  import javax.annotation.Resource; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpSession;  import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; import org.springframework.web.servlet.ModelAndView; import org.springframework.web.socket.TextMessage;  import com.google.gson.GsonBuilder; import com.yun.common.entity.DataGrid; import com.yun.price.garlic.dao.entity.GarlicPrice; import com.yun.price.garlic.model.GarlicPriceModel; import com.yun.price.garlic.service.GarlicPriceService; import com.yun.websocket.MyWebSocketHandler;  /**  * Title: GarlicPriceController<br/>  * Description:  *  * @Company: 青島勵圖高科<br/>  * @author: 劉云生  * @version: v1.0  * @since: JDK 1.7.0_80  * @Date: 2016年6月23日 下午3:23:46 <br/>  */ @Controller public class GarlicPriceController {   @Resource   MyWebSocketHandler myWebSocketHandler;   @RequestMapping(value = "GarlicPriceController/testWebSocket", method ={RequestMethod.POST,RequestMethod.GET}, produces = "application/json; charset=utf-8")   @ResponseBody   public String testWebSocket() throws IOException{     myWebSocketHandler.sendMessageToJsp(new TextMessage(new GsonBuilder().create().toJson("/"number/":/""+"GarlicPriceController/testWebSocket"+"/"")), "AAA");     return "1";   }    } 

4.所用到的jar包

<dependency>     <groupId>org.springframework</groupId>     <artifactId>spring-websocket</artifactId>     <version>4.0.1.RELEASE</version> </dependency>  

5.運行的環(huán)境

至少tomcat8.0以上版本,否則可能報錯

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持VeVb武林網(wǎng)。


注:相關(guān)教程知識閱讀請移步到JAVA教程頻道。
發(fā)表評論 共有條評論
用戶名: 密碼:
驗證碼: 匿名發(fā)表
主站蜘蛛池模板: 泸西县| 延边| 宁河县| 唐河县| 汶川县| 太保市| 信阳市| 苍梧县| 弋阳县| 藁城市| 大理市| 梓潼县| 襄城县| 镇江市| 青州市| 南木林县| 嵩明县| 嘉义县| 申扎县| 周至县| 保德县| 江口县| 贺州市| 新野县| 博客| 凉城县| 繁峙县| 兖州市| 泾川县| 镇坪县| 禄劝| 舟山市| 乾安县| 谷城县| 青龙| 三原县| 盐亭县| 大余县| 梁平县| 辉南县| 惠来县|