WEB Socket 通讯服务

import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;

import javax.servlet.Servlet;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.web.socket.server.standard.SpringConfigurator;

import com.mysql.jdbc.interceptors.SessionAssociationInterceptor;
 
 
/**
 * 
* @ClassName: WebSocketChat
* @Description: websocket连接URL地址和可被调用配置
* @author tangxuefeng
* @date 2019年9月7日
* 
 */
@ServerEndpoint(value="/chat/{userId}",configurator = SpringConfigurator.class)
public class WebSocketChat {
    
    private static final Logger logger = LoggerFactory.getLogger(WebSocketChat.class);
     //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;
    
    private static Set<WebSocketChat> userSocket =new HashSet<>(); 
    //记录每个用户下多个终端的连接
    private static Map<String, Set<WebSocketChat>> map = new HashMap<>();
    //需要session来对用户发送数据, 获取连接特征userId
    private Session session;
    private String userId;
   
    public Map<String, Set<WebSocketChat>> getMap(){
        return map;
    }
    /**
     * @Title: onOpen
     * @Description: websocekt连接建立时的操作
     * @param @param userId 用户id
     * @param @param session websocket连接的session属性
     * @param @throws IOException
     */
    @OnOpen
    public void onOpen(@PathParam("userId") String userId,Session session) throws IOException{
        this.session = session;
        this.userId = userId;
        onlineCount++;
        //根据该用户当前是否已经在别的终端登录进行添加操作
        if (map.containsKey(this.userId)) {
            logger.info("当前用户id:{"+this.userId+"}已有其他终端登录");
            map.get(this.userId).add(this);
        }else {
            logger.info("当前用户id:{"+this.userId+"}第一个终端登录");
            Set<WebSocketChat> addUserSet = new HashSet<>();
            addUserSet.add(this);
            map.put(this.userId, addUserSet);
        }
    }
   
    /**
     * @throws IOException 
     * @Title: onClose
     * @Description: 连接关闭的操作
     */
    
    @OnClose
    public void onClose() throws IOException{      
   
        onlineCount--;
        System.out.println("连接关闭的操作:"+this.userId);
        //移除当前用户终端登录的websocket信息,如果该用户的所有终端都下线了，则删除该用户的记录
        if (map.get(this.userId).size() == 0) {
            userSocket.remove(this.userId);
            map.remove(this.userId);
        }else{
            map.get(this.userId).remove(this);
        }
        logger.info("用户{"+this.userId+"}登录的终端个数是为{"+map.get(this.userId).size()+"}");
        logger.info("所有终端个数为：{"+onlineCount+"}");
        
    }
   
    /**
     * @Title: onMessage
     * @Description: 收到消息后的操作
     * @param @param message 收到的消息
     * @param @param session 该连接的session属性
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        System.out.println("收到消息后的操作:"+this.userId);
        logger.info("收到来自用户id为：{"+this.userId+"}的消息：{"+message+"}");
        if(session ==null)  logger.info("session null");
        //测试向客户端发送消息发送
        sendMessageToUser(this.userId,"服务器收到你的消息："+ message);
    }
   
    /**
     * @Title: onError·
     * @Description: 连接发生错误时候的操作
     * @param @param session 该连接的session
     * @param @param error 发生的错误
     */
    @OnError
    public void onError(Session session, Throwable error){
        
        if (map.get(this.userId).size() == 0) {
            userSocket.remove(this.userId);
            map.remove(this.userId);
            onlineCount--;
        }else{
            map.get(this.userId).remove(this);
        }
        System.out.println("连接发生错误时候的操作:"+this.userId);
        logger.info("用户id为：{"+this.userId+"}的连接发送错误");
        error.printStackTrace();
        
    }
   
  /**
   * @Title: sendMessageToUser
   * @Description: 发送消息给用户下的所有终端
   * @param @param userId 用户id
   * @param @param message 发送的消息
   * @param @return 发送成功返回true，反则返回false
   */
    public Boolean sendMessageToUser(String userId,String message){
        System.out.println(map.size());
        if (map.containsKey(userId)) {
            logger.info(" 给用户id为：{"+userId+"}的所有终端发送消息：{"+message+"}");
            for (WebSocketChat WS : map.get(userId)) {
                logger.info("sessionId为:{"+WS.session.getId()+"}");
                try { 
                 WS.session.getBasicRemote().sendText(message);
                } catch (IOException e) {
                    e.printStackTrace();
                    logger.info(" 给用户id为：{"+userId+"}发送消息失败");
                    return false;
                }
            }
            return true;
        }
        System.out.println("发送错误：当前连接没有:"+this.userId);
        logger.info("发送错误：当前连接不包含id为：{"+userId+"}的用户");
        return false;
    }

    public Map<String, Object> seleMap() {
        Map<String, Object> hasemap=new HashMap<String, Object>();
        List list=new ArrayList<>();
        int i=0;
        for(String key : map.keySet()){
            list.add(key);
            i++;
          }
        hasemap.put("num", i);
        hasemap.put("stringArray", list);
        return hasemap;
    }
 
}