JavaWeb SpringMVC整合WebSocket(注解实现)

Spring4.0以后引入了对websocket的支持 。相信你已经搭建了SpringMVC的开发框架。

所以这里只给出websocket的相关依赖:
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-websocket</artifactId>
<version>4.0.1.RELEASE</version>
</dependency>
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-messaging</artifactId>
<version>4.0.1.RELEASE</version>
</dependency>
Java代码如下:

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArraySet;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import org.apache.log4j.Logger;

/**
* @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
* 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
* @author
*/
@ServerEndpoint(value="/websocket")
public class WebSocketTest {
private static Logger logger = Logger.getLogger(WebSocketTest.class);
//线程安全的静态变量，表示在线连接数
private static volatile int  onlineCount = 0;

//用来存放每个客户端对应的WebSocketTest对象，适用于同时与多个客户端通信
public static CopyOnWriteArraySet<WebSocketTest> webSocketSet = new CopyOnWriteArraySet<WebSocketTest>();
//若要实现服务端与指定客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
public static ConcurrentHashMap<Session,Object> webSocketMap = new ConcurrentHashMap<Session,Object>();

//与某个客户端的连接会话，通过它实现定向推送(只推送给某个用户)
private Session session;

/**
* 连接建立成功调用的方法
*
* @param session 可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
*/
@OnOpen
public void onOpen(Session session){
this.session = session;
webSocketSet.add(this);     //加入set中
webSocketMap.put(session,this); //加入map中
addOnlineCount();    //在线数加1
System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());
}

/**
* 连接关闭调用的方法
*/
@OnClose
public void onClose(Session closeSession) {
webSocketSet.remove(this); //从set中删除
webSocketMap.remove(closeSession); //从map中删除
subOnlineCount();          //在线数减1
System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());
}

/**
* 收到客户端消息后调用的方法
*
* @param message 客户端发送过来的消息
* @param session 可选的参数
* @throws Exception
*/
@OnMessage
public void onMessage(String message,Session mySession) throws Exception {

logger.info("来自客户端的消息:" + message);
//--------------群发消息(多用于聊天室场景)
/* for (WebSocketTest item : webSocketSet) {
try {
item.sendAllMessage(message);
} catch (IOException e) {
e.printStackTrace();
continue;
}
}*/

//推送给单个客户端
for (Session session : webSocketMap.keySet()) {
if (session.equals(mySession)) {
WebSocketTest item = (WebSocketTest) webSocketMap.get(mySession);
try {
String msg="嗨，这是返回的信息";
item.sendMessage(mySession,msg);
} catch (IOException e) {
e.printStackTrace();
}
}

}

}

/**
* 发生错误时调用
*
* @param session
* @param error
*/
@OnError
public void onError(Session session, Throwable error) {
logger.info("发生错误");
// error.printStackTrace();
}

//给所有客户端发送信息
public void sendAllMessage(String message) throws IOException {
this.session.getBasicRemote().sendText(message);
}

//定向发送信息
public void sendMessage(Session mySession,String message) throws IOException {
synchronized(this) {try {
if(mySession.isOpen()){//该session如果已被删除，则不执行发送请求，防止报错
//this.session.getBasicRemote().sendText(message);
mySession.getBasicRemote().sendText(message);
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

}

}

public static synchronized int getOnlineCount() {
return onlineCount;
}

public static synchronized void addOnlineCount() {
WebSocketTest.onlineCount++;
}

public static synchronized void subOnlineCount() {
WebSocketTest.onlineCount--;
}

}

   值得注意的一点是，如果想要在如上的websocket中直接注入service或者dao来进行数据库操作是无法注入的。我的解决办法是新建一个专门操作数据库的类并在applicationContext.xml中进行配置。比如我的这个类叫GetDataServiceImpl，首先要在 applicationContext.xml 中加入<bean id="getDataServiceImpl" class="com.cn.service.impl.GetDataServiceImpl"></bean>在websocket中通过beanid获取该类

private GetDataServiceImpl getDataServiceImpl = (GetDataServiceImpl) ContextLoader.getCurrentWebApplicationContext().getBean("getDataServiceImpl");

         以上就是我参考网上相关资料后根据自己的理解在SSM框架的基础上整合的websocket代码，并且在项目中已经在使用，目前为止还没有发现什么问题。可以看到，代码其实并不多，而且使用注解方式也比较简单，容易理解。如有不妥之处，欢迎交流指正。