spring websocket 模拟发送消息

实现以下功能：

1、各个客户端之间的消息互发；

2、客户端向后台发送消息；

3、后台向客户端发送消息






注：完整源代码下载地址  http://download.csdn.net/detail/u010994277/9705588

代码如下：

1）、配置处理器

/**
* WebScoket配置处理器
*
* @author www
* @Date 2016年12月8日
*/
@Component
@EnableWebSocket
@Configuration
@EnableWebMvc
public class WebSocketConfig implements WebSocketConfigurer {

@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(new MyWebSocketHandler(), "/portfolio2")// 添加一个处理器还有定义处理器的处理路径
.addInterceptors(new MyHandShakeInterceptor());
registry.addHandler(new MyWebSocketHandler(), "/sockjs/portfolio2")// 添加一个处理器还有定义处理器的处理路径
.addInterceptors(new MyHandShakeInterceptor()).withSockJS();
}

}2、配置拦截器
/**
* Socket建立连接（握手）和断开
*
* @author www
* @Date 2016年12月8日
*/
public class MyHandShakeInterceptor implements HandshakeInterceptor {

/**
* 握手之前，若返回false，则不建立链接
*/
@Override
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
Map<String, Object> attributes) throws Exception {
// TODO Auto-generated method stub
System.out.println("beforeHandshake-->");
if (request instanceof ServletServerHttpRequest) {
try {
ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
HttpSession session = servletRequest.getServletRequest().getSession(false);
if (session != null) {
String userId = (String) session.getAttribute(Constant.SESSION_USER_ID);
if(userId==null||userId.equals("")){
throw new Exception("用户ID为空");
}
attributes.put(Constant.SESSION_USER_ID,userId);
}else{
throw new Exception("session为空");
}
} catch (Exception e) {
attributes.put(Constant.SESSION_USER_ID,"test-user");
}
}
return true;
}

/**
* 握手之后
*/
@Override
public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
Exception exception) {
// TODO Auto-generated method stub
System.out.println("afterHandshake-->");
}

3、socket处理器
/**
* Socket处理器
*
* @author www
* @Date 2016年12月8日
*/
@Component
public class MyWebSocketHandler implements WebSocketHandler {

public static final Map<String, WebSocketSession> userSocketSessionMap;

static {
userSocketSessionMap = new HashMap<String, WebSocketSession>();
}

/**
* webscoket建立好链接之后的处理函数
*
* @param session
* 当前websocket的会话id，打开一个websocket通过都会生成唯一的一个会话，
* 可以通过该id进行发送消息到浏览器客户端
*/
@Override
public void afterConnectionEstablished(WebSocketSession session)
throws Exception {
// TODO Auto-generated method stub
String uid = (String) session.getAttributes().get(Constant.SESSION_USER_ID);
System.out.println("用户ID:" + uid + ";已建立连接");
// 每个链接来的客户端都把WebSocketSession保存进来
if (userSocketSessionMap.get(uid) == null) {
userSocketSessionMap.put(uid, session);
}else{
userSocketSessionMap.remove(uid);
userSocketSessionMap.put(uid, session);
}

}

/**
* 客户端发送服务器的消息时，的处理函数，在这里收到消息之后可以分发消息
*/
@Override
public void handleMessage(WebSocketSession session,
WebSocketMessage<?> message) throws Exception {
// TODO Auto-generated method stub
Iterator<Map.Entry<String, WebSocketSession>> entries = userSocketSessionMap.entrySet().iterator();
while (entries.hasNext()) {
Map.Entry<String, WebSocketSession> entry = entries.next();
System.out.println("Key = " + entry.getKey() + ", Value = " + entry.getValue());
if(entry.getKey().equals(session.getAttributes().get("userId"))){
String msg = "后台已收到信息[" + message.getPayload().toString()+ "];用户ID为：" + entry.getKey();
System.out.println(msg);
message = new TextMessage(msg);
entry.getValue().sendMessage(message);
}
}
}

/**
* 消息传输过程中出现的异常处理函数
*/
@Override
public void handleTransportError(WebSocketSession session,
Throwable exception) throws Exception {
// TODO Auto-generated method stub
String userId=(String) session.getAttributes().get("userId");
userSocketSessionMap.remove(userId);

}

/**
* websocket链接关闭的回调
*/
@Override
public void afterConnectionClosed(WebSocketSession session,
CloseStatus closeStatus) throws Exception {
// TODO Auto-generated method stub
String userId=(String) session.getAttributes().get("userId");
userSocketSessionMap.remove(userId);
}

/**
* 是否支持处理拆分消息，返回true返回拆分消息
*/
@Override
public boolean supportsPartialMessages() {
// TODO Auto-generated method stub
return false;
}

/**
* 给所有在线用户发送消息
*
* @param message
* @throws IOException
*/
public void broadcast(final String msg) throws IOException {
final TextMessage message=new TextMessage(msg);
Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap.entrySet().iterator();
// 多线程群发
while (it.hasNext()) {

final Entry<String, WebSocketSession> entry = it.next();

if (entry.getValue().isOpen()) {
// entry.getValue().sendMessage(message);
new Thread(new Runnable() {

public void run() {
try {
if (entry.getValue().isOpen()) {
entry.getValue().sendMessage(message);
}
} catch (IOException e) {
e.printStackTrace();
}
}

}).start();
}

}
}

/**
* 给某个用户发送消息
*
* @param userName
* @param message
* @throws IOException
*/
public void sendMessageToUser(String uid, String msg)
throws IOException {
TextMessage message=new TextMessage(msg);
WebSocketSession session = userSocketSessionMap.get(uid);
if (session != null && session.isOpen()) {
session.sendMessage(message);
}
}