Spring boot集成websocket（一）----传统注册方式

修改配置类

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer{
 
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        
        //handler是webSocket的核心，配置入口
        registry.addHandler(new CTIHandler(), "/webscoket/call/{ID}").setAllowedOrigins("*").addInterceptors(new WebSocketInterceptor());
        
    }

}

增加拦截器的配置

public class WebSocketInterceptor extends HttpSessionHandshakeInterceptor {
	/**
	 * 配置日志
	 */
	private final static Logger logger = LoggerFactory.getLogger(WebSocketInterceptor.class);

	@Override
	public boolean beforeHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse seHttpResponse,
			WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
//		HttpServletRequest request = ((ServletServerHttpRequest) serverHttpRequest).getServletRequest();
		String userName = serverHttpRequest.getURI().toString().split("ID=")[1];
		attributes.put("userName", userName);
		logger.info("握手之前");
		//从request里面获取对象，存放attributes
		return super.beforeHandshake(serverHttpRequest, seHttpResponse, wsHandler, attributes);
	}

	@Override
	public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
			Exception ex) {
		logger.info("握手之后");
		super.afterHandshake(request, response, wsHandler, ex);
	}
}

修改handler类

@Service
public class CTIHandler extends ParentHandler implements WebSocketHandler,RedisReceiveService {
	/**
	 * 配置日志
	 */
	private final static Logger logger = LoggerFactory.getLogger(CTIHandler.class);
	
    //新增socket
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
    	logger.info("外呼成功建立连接");
    	//获取用户信息
    	String userName = (String) session.getAttributes().get("userName");
    	logger.info("获取当前"+socketMap.get(userName));
    	if(socketMap.get(userName)==null) {
    		socketMap.put(userName,session);
    	}
    	logger.info("链接成功");
    }

    /**
     * 接收socket信息
     */
	@Override
    public void handleMessage(WebSocketSession webSocketSession, WebSocketMessage<?> webSocketMessage) throws Exception {
    	String userName = (String) webSocketSession.getAttributes().get("userName");
    	String payLoad = Pub_Tools.getString_TrimZeroLenAsNull(webSocketMessage.getPayload());
    	logger.info("收到的消息"+payLoad);
    }

	/**
	 * 连接出错
	 */
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        if (session.isOpen()) {
            session.close();
        }
        logger.info("连接出错");
    }
    /**
     * 连接关闭
     */
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
    	//获取用户信息
    	String userName = (String) session.getAttributes().get("userName");
    	if(socketMap.get(userName)!=null) {
    		socketMap.remove(userName);
    		//并且通过redis发布和订阅广播给其他的的机器，或者通过消息队列
    	}
        logger.info("连接已关闭：" + status);
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
    /**
     * 接受订阅信息
     */
	@Override
	public void receiveMessage(String message) {
		JSONObject sendMsg = JSONObject.fromObject(message.substring(message.indexOf("{")));
		String clientId = sendMsg.getString("userName");
		TextMessage receiveMessage = new TextMessage(sendMsg.getString("message"));
		boolean flag = sendMessageToUser(clientId, receiveMessage);
		if(flag) {
			logger.info("我发送消息成功了！");
		}
	}

}