websocketservice服务类
package cn.oyohotels.iot.bff.kiosk.server;import java.io.IOException;
import java.util.concurrent.CopyOnWriteArraySet;
import java.util.concurrent.atomic.AtomicInteger;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 cn.oyohotels.iot.bff.kiosk.entity.socket.SocketView;
import cn.oyohotels.iot.bff.kiosk.utils.JsonUtil;
import org.springframework.stereotype.Component;
import lombok.extern.slf4j.Slf4j;/*** WebSocket服务端** @author panghui* @version 1.0* @since 2019/11/6*/
@ServerEndpoint("/webSocketServer/{hotelId}")
@Component
@Slf4j
public class WebSocketServer {//静态变量，用来记录当前在线连接数private static AtomicInteger onlineCount = new AtomicInteger(0);//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。private static CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();//与某个客户端的连接会话，需要通过它来给客户端发送数据private Session session;/*** 接收的酒店ID*/private String hotelId="";/*** 连接建立成功** @param session* @param hotelId*/@OnOpenpublic void onOpen(Session session,@PathParam("hotelId") String hotelId) {log.info("当前对象："+this);this.session = session;// 将当前的Socket对象放入集合webSocketSet.add(this);// 增加连接人数 +1onlineCount.incrementAndGet();log.info("有新窗口开始监听:"+hotelId+",当前在线人数为" + onlineCount.get());this.hotelId=hotelId;try {sendMessage("连接成功");} catch (IOException e) {log.error("websocket IO异常");}}/*** 连接关闭调用的方法*/@OnClosepublic void onClose() {// 移除webSocketSet.remove(this);// 在线人数减少 -1onlineCount.decrementAndGet();log.info("有一连接关闭！当前在线人数为" + onlineCount.get());}/*** 收到客户端消息后调用方法** @param message 客户端发送过来的消息* @param session*/@OnMessagepublic void onMessage(String message, Session session) {log.info("收到来自酒店ID："+hotelId+"的信息:"+message);//群发消息for (WebSocketServer webSocketServer : webSocketSet) {try {webSocketServer.sendMessage(message);} catch (IOException e) {e.printStackTrace();}}}@OnErrorpublic void onError(Session session, Throwable error) {log.error("发生错误");error.printStackTrace();}/*** 实现服务器主动推送*/public void sendMessage(String message) throws IOException {this.session.getBasicRemote().sendText(message);}/*** 群发自定义消息*/public static void sendInfo(SocketView socketView, @PathParam("hotelId") String hotelId) throws IOException {String message = JsonUtil.objectToJson(socketView);log.info("推送消息到对应酒店，酒店ID："+hotelId+"，推送内容:"+message);for (WebSocketServer webSocketServer : webSocketSet) {try {//这里可以设定只推送给这个sid的，为null则全部推送if(hotelId==null) {webSocketServer.sendMessage(message);}else if(webSocketServer.hotelId.equals(hotelId)){webSocketServer.sendMessage(message);}} catch (IOException e) {continue;}}}
}
调用WebSocketServer 服务发送消息
WebSocketServer.sendInfo(socketView,bean.getHotelId().toString());
客户端测试建立连接
直接使用在线工具：http://ws.douqq.com/
