springMvc中websokect的应用实例

来源：互联网发布：淘宝产品修图编辑：程序博客网时间：2024/05/01 11:55

1.需要用到的jar包

spring-websocket-4.0.5.RELEASE.jar

2.配置config文件：

spring-websocket.xml

内容：

3.配置web.xml

在web.xml中添加

<servlet>
<servlet-name>websocket</servlet-name>
<servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
<init-param>
<param-name>contextConfigLocation</param-name>
<param-value>classpath*:/config/spring-websocket.xml</param-value>
</init-param>
<async-supported>true</async-supported>
</servlet>
<servlet-mapping>
<servlet-name>websocket</servlet-name>
<url-pattern>/websocket</url-pattern>
<url-pattern>/socketjs/*</url-pattern>
</servlet-mapping>

另外在<filter></filter>中添加 <async-supported>true</async-supported>

4.创建握手处理的类,和消息处理的类

握手处理的类：

public class HandShakeInterceptor extends HttpSessionHandshakeInterceptor {
@Override
public boolean beforeHandshake(ServerHttpRequest request,
ServerHttpResponse response, WebSocketHandler wsHandler,
Map<String, Object> attributes) throws Exception {
//握手前
if (request instanceof ServletServerHttpRequest) {
ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
HttpSession session = servletRequest.getServletRequest().getSession(false);
// 标记用户
Integer id = (Integer) session.getAttribute("id");
if(id!=null){
attributes.put("id", id);
}else{
return false;
}
}
return true;
}

@Override
public void afterHandshake(ServerHttpRequest request,
ServerHttpResponse response, WebSocketHandler wsHandler,
Exception ex) {
//握手后
super.afterHandshake(request, response, wsHandler, ex);

}
}

消息处理的类

public class WebSocketHandler extends TextWebSocketHandler {
public static Map<Integer,WebSocketSession> DOCTORONLINE=new HashMap<Integer,WebSocketSession>();
@Override
protected void handleTextMessage(WebSocketSession session,TextMessage message) throws Exception {
super.handleTextMessage(session, message);
Integer userId=null;
if(message.getPayload()!=null&&message.getPayload()!=""){
userId=Integer.parseInt(message.getPayload());
}
if(DOCTORONLINE.get(userId)==null){
DOCTORONLINE.put(userId, session);
}
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
System.out.println("Websocket:" + session.getId() + "已经关闭");
Iterator<Entry<Integer, WebSocketSession>> it = DOCTORONLINE.entrySet().iterator();
// 移除Socket会话
while (it.hasNext()) {
Entry<Integer, WebSocketSession> entry = it.next();
if (entry.getValue().getId().equals(session.getId())) {
DOCTORONLINE.remove(entry.getKey());
System.out.println("Socket会话已经移除:用户ID" + entry.getKey());
break;
}
}
}
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
// TODO Auto-generated method stub
super.afterConnectionEstablished(session);
Integer id=(Integer)session.getAttributes().get("id");
//将消息推送给这个专家
if(DOCTORONLINE.get(id)==null){
DOCTORONLINE.put(id, session);
}

}
@Override
protected void handlePongMessage(WebSocketSession session, PongMessage message) throws Exception {
// TODO Auto-generated method stub
super.handlePongMessage(session, message);
}
@Override
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
// TODO Auto-generated method stub
super.handleTransportError(session, exception);
//信息传输时出错的处理
if (session.isOpen()) {
session.close();
}
Iterator<Entry<Integer, WebSocketSession>> it = DOCTORONLINE.entrySet().iterator();
// 移除Socket会话
while (it.hasNext()) {
Entry<Integer, WebSocketSession> entry = it.next();
if (entry.getValue().getId().equals(session.getId())) {
DOCTORONLINE.remove(entry.getKey());
System.out.println("Socket会话已经移除:用户ID" + entry.getKey());
break;
}
}
}
@Override
public boolean supportsPartialMessages() {
// TODO Auto-generated method stub
return false;
}
//发送信息给专家
public void sendMessageToDoctor(Integer doctorId, Integer id) throws Exception{
//消息编号必须存在
if(id!=null&&id>=0){
//找到指定的专家doctorId
WebSocketSession session=DOCTORONLINE.get(doctorId);
if(session!=null&&session.isOpen()){
TextMessage textMessage=new TextMessage(String.valueOf(id));
//将消息推送给这个专家
session.sendMessage(textMessage);
}
}
}
}

５.controller中

将id 保存在session中

在index.jsp页面 websocket的使用

var connectCount=0;
var socketFlg=0;

6.补充说明websokect的原理

http://blog.csdn.net/yinqingwang/article/details/52565133

http://blog.csdn.net/frank_good/article/details/50856585

阅读全文

0 0