`
zengshaotao
  • 浏览: 746952 次
  • 性别: Icon_minigender_1
  • 来自: 上海
社区版块
存档分类
最新评论

基于Tomcat的WebSocket

    博客分类:
  • java
阅读更多

原文地址:http://redstarofsleep.iteye.com/blog/1488639

 

2014年2月更新: 此API为Tomcat私有,当时Java没有标准API,现在Java有标准API,JSR536,此API不建议使用,新的用法请参照http://redstarofsleep.iteye.com/blog/1974620

 

之前大概的看过WebSocket,当时Tomcat还不支持WebSocket,所以当时写了一篇基于Jetty的WebSocket实现,地址如下:

http://redstarofsleep.iteye.com/blog/1307608

 

现在Tomcat7.0.27发布了,从这个版本开始Tomcat就支持WebSocket了。

Tomcat的WebSocket和Jetty的大致上差不多,大同小异,这里就简单的贴两个类吧(此例子未考虑多线程的情况):

 

第一个类,这个和Jetty一样,需要一个Servlet来处理WebSocket请求:

Java代码  收藏代码
  1. package lhc.websocket;  
  2.   
  3. import org.apache.catalina.websocket.StreamInbound;  
  4. import org.apache.catalina.websocket.WebSocketServlet;  
  5.   
  6. public class MyWebSocketServlet extends WebSocketServlet {  
  7.   
  8.     private static final long serialVersionUID = -7178893327801338294L;  
  9.   
  10.     @Override  
  11.     protected StreamInbound createWebSocketInbound(String arg0) {  
  12. System.out.println("##########");  
  13.         return new MyMessageInbound();  
  14.     }  
  15.   
  16. }  

这个Servlet继承自WebSocketServlet,实现createWebSocketInbound方法。该方法返回第二个类的实例。

 

第二个类,处理每一次具体的WebSocket任务:

Java代码  收藏代码
  1. package lhc.websocket;  
  2.   
  3. import java.io.IOException;  
  4. import java.nio.ByteBuffer;  
  5. import java.nio.CharBuffer;  
  6.   
  7. import lhc.init.InitServlet;  
  8.   
  9. import org.apache.catalina.websocket.MessageInbound;  
  10. import org.apache.catalina.websocket.WsOutbound;  
  11.   
  12. public class MyMessageInbound extends MessageInbound {  
  13.   
  14.     @Override  
  15.     protected void onBinaryMessage(ByteBuffer arg0) throws IOException {  
  16.         // TODO Auto-generated method stub  
  17.           
  18.     }  
  19.   
  20.     @Override  
  21.     protected void onTextMessage(CharBuffer msg) throws IOException {  
  22.         for (MessageInbound messageInbound : InitServlet.getSocketList()) {  
  23.             CharBuffer buffer = CharBuffer.wrap(msg);  
  24.             WsOutbound outbound = messageInbound.getWsOutbound();  
  25.             outbound.writeTextMessage(buffer);  
  26.             outbound.flush();  
  27.         }  
  28.           
  29.     }  
  30.   
  31.     @Override  
  32.     protected void onClose(int status) {  
  33.         InitServlet.getSocketList().remove(this);  
  34.         super.onClose(status);  
  35.     }  
  36.   
  37.     @Override  
  38.     protected void onOpen(WsOutbound outbound) {  
  39.         super.onOpen(outbound);  
  40.         InitServlet.getSocketList().add(this);  
  41.     }  
  42.       
  43.       
  44.   
  45. }  

 

 这个类继承自MessageInbound类,必须实现onBinaryMessage和onTextMessage方法。Jetty中只有一个onMessage方法,而Tomcat细化成了2个方法。

 

还要一个初始化的Servlet

Java代码  收藏代码
  1. package lhc.init;  
  2.   
  3. import java.util.ArrayList;  
  4. import java.util.List;  
  5.   
  6. import javax.servlet.ServletConfig;  
  7. import javax.servlet.ServletException;  
  8. import javax.servlet.http.HttpServlet;  
  9.   
  10. import org.apache.catalina.websocket.MessageInbound;  
  11.   
  12. public class InitServlet extends HttpServlet {  
  13.   
  14.     private static final long serialVersionUID = -3163557381361759907L;  
  15.       
  16.     private static List<MessageInbound> socketList;    
  17.       
  18.     public void init(ServletConfig config) throws ServletException {    
  19.         InitServlet.socketList = new ArrayList<MessageInbound>();    
  20.         super.init(config);    
  21.         System.out.println("Server start============");    
  22.     }    
  23.         
  24.     public static List<MessageInbound> getSocketList() {    
  25.         return InitServlet.socketList;    
  26.     }    
  27. }  

 

 

最后,web.xml中进行一下Servlet的配置。

Xml代码  收藏代码
  1. <?xml version="1.0" encoding="UTF-8"?>  
  2. <web-app xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://java.sun.com/xml/ns/javaee" xmlns:web="http://java.sun.com/xml/ns/javaee/web-app_2_5.xsd" xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_3_0.xsd" id="WebApp_ID" version="3.0">  
  3.   <display-name>wsoc</display-name>  
  4.   <welcome-file-list>  
  5.     <welcome-file>index.jsp</welcome-file>  
  6.   </welcome-file-list>  
  7.     
  8.   <servlet>  
  9.     <servlet-name>mywebsocket</servlet-name>  
  10.     <servlet-class>lhc.websocket.MyWebSocketServlet</servlet-class>  
  11.   </servlet>  
  12.   <servlet-mapping>  
  13.     <servlet-name>mywebsocket</servlet-name>  
  14.     <url-pattern>*.do</url-pattern>  
  15.   </servlet-mapping>  
  16.     
  17.   <servlet>  
  18.     <servlet-name>initServlet</servlet-name>  
  19.     <servlet-class>lhc.init.InitServlet</servlet-class>  
  20.     <load-on-startup>1</load-on-startup>  
  21.   </servlet>  
  22. </web-app>  

 

 

页面的话,就是标准的HTML5的websocket,这个和服务器是否是Tomcat或者Jetty是无关的:

Html代码  收藏代码
  1. <%@ page language="java" contentType="text/html; charset=UTF-8"  
  2.     pageEncoding="UTF-8"%>  
  3. <!DOCTYPE html>  
  4. <html>  
  5. <head>  
  6. <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">  
  7. <title>Index</title>  
  8. <script type="text/javascript">  
  9. var ws = null;  
  10. function startWebSocket() {  
  11.     if ('WebSocket' in window)  
  12.         ws = new WebSocket("ws://localhost:8080/wsoc/mywebsocket.do");  
  13.     else if ('MozWebSocket' in window)  
  14.         ws = new MozWebSocket("ws://localhost:8080/wsoc/mywebsocket.do");  
  15.     else  
  16.         alert("not support");  
  17.       
  18.       
  19.     ws.onmessage = function(evt) {  
  20.         alert(evt.data);  
  21.     };  
  22.       
  23.     ws.onclose = function(evt) {  
  24.         alert("close");  
  25.     };  
  26.       
  27.     ws.onopen = function(evt) {  
  28.         alert("open");  
  29.     };  
  30. }  
  31.   
  32. function sendMsg() {  
  33.     ws.send(document.getElementById('writeMsg').value);  
  34. }  
  35. </script>  
  36. </head>  
  37. <body onload="startWebSocket();">  
  38. <input type="text" id="writeMsg"></input>  
  39. <input type="button" value="send" onclick="sendMsg()"></input>  
  40. </body>  
  41. </html>  
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics