java websocket 群发_java集成WebSocket向所有用户发送消息

package com.reading.controller.library;

import org.springframework.stereotype.Controller;

import org.springframework.web.socket.server.standard.SpringConfigurator;

import javax.websocket.*;

import javax.websocket.server.PathParam;

import javax.websocket.server.ServerEndpoint;

import java.util.concurrent.CopyOnWriteArraySet;

/**

* Created by zz on 2018/9/6.

*/

@ServerEndpoint(value = "/websocket/{id}",configurator = SpringConfigurator.class)

public class MyWebSocket {

//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。

private static int onlineCount = 0;

public MyWebSocket() {

}

//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。

// 若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识

private static CopyOnWriteArraySet webSocketSet = new CopyOnWriteArraySet();

//与客户端的连接会话，需要通过它来给客户端发送数据

private Session session;

/**

* 连接建立成功调用的方法

* @param session  可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据

*/

@OnOpen

public void onOpen(Session session,EndpointConfig config,@PathParam("id") Integer id){

this.session = session;

//加入set中

webSocketSet.add(this);

//在线数加1

addOnlineCount();

System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());

}

/**

* 连接关闭调用的方法

*/

@OnClose

public void onClose(){

//从set中删除

webSocketSet.remove(this);

//在线数减1

subOnlineCount();

System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());

}

/**

* 收到客户端消息后调用的方法

* @param message 客户端发送过来的消息

* @param session 可选的参数

*/

@OnMessage

public void onMessage(String message, Session session) {

System.out.println("来自客户端的消息:" + message);

//群发消息

for(MyWebSocket item: webSocketSet){

//            try {

//                item.sendMessage(message);

//            } catch (IOException e) {

//                e.printStackTrace();

//                continue;

//            }

}

}

/**

* 发生错误时调用

* @param session

* @param error

*/

@OnError

public void onError(Session session, Throwable error){

System.out.println("发生错误");

error.printStackTrace();

}

/**

* 这个方法与上面几个方法不一样。没有用注解，是根据自己需要添加的方法。

* @param message

* @throws IOException

*/

public void sendMessage(String message){

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

}

public static synchronized int getOnlineCount() {

return onlineCount;

}

public static synchronized void addOnlineCount() {

MyWebSocket.onlineCount++;

}

public static synchronized void subOnlineCount() {

MyWebSocket.onlineCount--;

}

}

T