基于springmvc的websocket及时聊天

package com.pro.websocket; import java.io.IOException; import java.util.ArrayList; import java.util.List; import java.util.Map; import org.apache.log4j.Logger; 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.Gson; import com.google.gson.reflect.TypeToken; /** * 2.0 * * @author Administrator * */ public class WebSocketHander implements WebSocketHandler { private static final List<WebSocketSession> users = new ArrayList<WebSocketSession>(); private static final Logger log = Logger.getLogger(WebSocketHandler.class); /** * * * @param session * @throws Exception */ public void afterConnectionEstablished(WebSocketSession session) throws Exception { String loginuser=(String) session.getHandshakeAttributes().get("loginuser"); boolean flag = false; for (WebSocketSession preSession : users) { if (preSession.getHandshakeAttributes().containsValue(loginuser)) { flag = true; break; } } if (!flag) { users.add(session); } System.out.println(loginuser); } /** * 澶勭悊娑堟伅 * * @param session * @param message * @throws Exception */ public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception { Gson gson = new Gson(); Map<String, Object> msg = gson.fromJson(message.getPayload().toString(), new TypeToken<Map<String, Object>>() { }.getType()); String msgspeaker = msg.get("postId").toString(); String msgContent = msg.get("msgContent").toString(); TextMessage textMessage = new TextMessage(msgspeaker+"說:"+msgContent, true); sendMessageToUser(textMessage); } /** * 发送消息 * * @param acceptorId * @param message */ public void sendMessageToUser(TextMessage message) { for (WebSocketSession user : users) { try { if (user.isOpen()) { user.sendMessage(message); } } catch (IOException e) { e.printStackTrace(); } } } public void handleTransportError(WebSocketSession session, Throwable throwable) throws Exception { if (session.isOpen()) { session.close(); } users.remove(session); } public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { users.remove(session); } public boolean supportsPartialMessages() { return false; } }