web socket

@ServerEndpoint("/my-websocket")

@Component

public class MyWebSocket {

protected final Logger logger = LoggerFactory.getLogger(this.getClass());

private static int onlineCount = 0;

private static CopyOnWriteArraySet<MyWebSocket> webSocketSet = new CopyOnWriteArraySet<>();

private Session session;

public String user;

@OnOpen

public void onOpen(Session session) throws IOException{

this.session = session;

System.out.println(session.getQueryString());

user=session.getRequestParameterMap().get("token").toString();

webSocketSet.add(this);

incrOnlineCount();

for(MyWebSocket item : webSocketSet){

if(!item.equals(this)) { //send to others only.

item.sendMessage("someone just joined in.");

}

}

logger.info("new connection...current online count: {}", getOnlineCount());

}

@OnClose

public void onClose() throws IOException{

webSocketSet.remove(this);

decOnlineCount();

for(MyWebSocket item : webSocketSet){

item.sendMessage("someone just closed a connection.");

}

logger.info("one connection closed...current online count: {}", getOnlineCount());

}

@OnMessage

public void onMessage(String message, Session session) throws IOException {

logger.info("message received: {}", message);

// broadcast received message

for(MyWebSocket item : webSocketSet){

item.sendMessage(message);

if(item.user.equals("[access_token]")){

System.out.println("end it ");

// item.session.close();

}

}

}

public void sendMessage(String message) throws IOException {

this.session.getBasicRemote().sendText(message);

}

public static synchronized int getOnlineCount(){

return MyWebSocket.onlineCount;

}

public static synchronized void incrOnlineCount(){

MyWebSocket.onlineCount++;

}

public static synchronized void decOnlineCount(){

MyWebSocket.onlineCount--;

}

}