SpringBoot集成WebSocket【基于纯H5】进行点对点[一对一]和广播[一对多]实时推送

 
 
  org.springframework.boot 
  spring-boot-starter-websocket 
 

package com.cloud.sbjm.configure;
 
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import com.cloud.sbjm.security.WebSocketInterceptor;
import com.cloud.sbjm.service.Imp.MyHandler;
 
 
//实现接口来配置Websocket请求的路径和拦截器。
@Configuration
@EnableWebSocket
public class WebSocketH5Config implements WebSocketConfigurer{
 
	@Override
	public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
		
	  //handler是webSocket的核心，配置入口
	  registry.addHandler(new MyHandler(), "/myHandler/{ID}").setAllowedOrigins("*").addInterceptors(new WebSocketInterceptor());
		
	}
 
	
}

package com.cloud.sbjm.security;
 
import java.util.Map;
 
import javax.servlet.http.HttpSession;
 
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.HandshakeInterceptor;
 
public class WebSocketInterceptor implements HandshakeInterceptor {
 
  //进入hander之前的拦截
  @Override
  public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map map) throws Exception {
    if (request instanceof ServletServerHttpRequest) {
      String ID = request.getURI().toString().split("ID=")[1];
      System.out.println("当前session的ID="+ID);
      ServletServerHttpRequest serverHttpRequest = (ServletServerHttpRequest) request;
      HttpSession session = serverHttpRequest.getServletRequest().getSession();
      map.put("WEBSOCKET_USERID",ID);
    }
    return true;
  }
 
  @Override
  public void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e) {
  	System.out.println("进来webSocket的afterHandshake拦截器！");
  }
}

package com.cloud.sbjm.service.Imp;
 
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import net.sf.json.JSONObject;
import org.springframework.stereotype.Service;
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;
 
 
@Service
public class MyHandler implements WebSocketHandler {
	
  //在线用户列表
  private static final Map users;
 
  
  static {
    users = new HashMap<>();
  }
  //新增socket
  @Override
  public void afterConnectionEstablished(WebSocketSession session) throws Exception {
  	 System.out.println("成功建立连接");
  	 String ID = session.getUri().toString().split("ID=")[1];
     System.out.println(ID);
     if (ID != null) {
       users.put(ID, session);
       session.sendMessage(new TextMessage("成功建立socket连接"));
       System.out.println(ID);
       System.out.println(session);
     }
     System.out.println("当前在线人数："+users.size());
  }
 
  //接收socket信息
  @Override
	public void handleMessage(WebSocketSession webSocketSession, WebSocketMessage<?> webSocketMessage) throws Exception {
  	try{
	  	JSONObject jsonobject = JSONObject.fromObject(webSocketMessage.getPayload());  	
	  	System.out.println(jsonobject.get("id"));
	  	System.out.println(jsonobject.get("message")+":来自"+(String)webSocketSession.getAttributes().get("WEBSOCKET_USERID")+"的消息");
	  	sendMessageToUser(jsonobject.get("id")+"",new TextMessage("服务器收到了，hello!"));
  	 }catch(Exception e){
   	  e.printStackTrace();
     }
		
	}
 
  /**
   * 发送信息给指定用户
   * @param clientId
   * @param message
   * @return
   */
  public boolean sendMessageToUser(String clientId, TextMessage message) {
    if (users.get(clientId) == null) return false;
    WebSocketSession session = users.get(clientId);
    System.out.println("sendMessage:" + session);
    if (!session.isOpen()) return false;
    try {
      session.sendMessage(message);
    } catch (IOException e) {
      e.printStackTrace();
      return false;
    }
    return true;
  }
 
  /**
   * 广播信息
   * @param message
   * @return
   */
  public boolean sendMessageToAllUsers(TextMessage message) {
    boolean allSendSuccess = true;
    Set clientIds = users.keySet();
    WebSocketSession session = null;
    for (String clientId : clientIds) {
      try {
        session = users.get(clientId);
        if (session.isOpen()) {
          session.sendMessage(message);
        }
      } catch (IOException e) {
        e.printStackTrace();
        allSendSuccess = false;
      }
    }
 
    return allSendSuccess;
  }
 
 
  @Override
  public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
    if (session.isOpen()) {
      session.close();
    }
    System.out.println("连接出错");
    users.remove(getClientId(session));
  }
 
  @Override
  public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
    System.out.println("连接已关闭：" + status);
    users.remove(getClientId(session));
  }
 
  @Override
  public boolean supportsPartialMessages() {
    return false;
  }
 
  /**
   * 获取用户标识
   * @param session
   * @return
   */
  private Integer getClientId(WebSocketSession session) {
    try {
      Integer clientId = (Integer) session.getAttributes().get("WEBSOCKET_USERID");
      return clientId;
    } catch (Exception e) {
      return null;
    }
  }
}

 
  
	
  
  
  
  
 
 
 
 
	
	Welcome

Send  Close