socket.io服务端是java_SpringBoot(23) 集成socket.io服务端和客户端实现通信

@Slf4j

@Service(value = "socketIOService")

public class SocketIOServiceImpl implements ISocketIOService {

/**

* 存放已连接的客户端

*/

private static Map clientMap = new ConcurrentHashMap<>();

/**

* 自定义事件`push_data_event`,用于服务端与客户端通信

*/

private static final String PUSH_DATA_EVENT = "push_data_event";

@Autowired

private SocketIOServer socketIOServer;

/**

* Spring IoC容器创建之后，在加载SocketIOServiceImpl Bean之后启动

*/

@PostConstruct

private void autoStartup() {

start();

}

/**

* Spring IoC容器在销毁SocketIOServiceImpl Bean之前关闭,避免重启项目服务端口占用问题

*/

@PreDestroy

private void autoStop() {

stop();

}

@Override

public void start() {

// 监听客户端连接

socketIOServer.addConnectListener(client -> {

log.debug("************ 客户端： " + getIpByClient(client) + " 已连接 ************");

// 自定义事件`connected` -> 与客户端通信 (也可以使用内置事件，如：Socket.EVENT_CONNECT)

client.sendEvent("connected", "你成功连接上了哦...");

String userId = getParamsByClient(client);

if (userId != null) {

clientMap.put(userId, client);

}

});

// 监听客户端断开连接

socketIOServer.addDisconnectListener(client -> {

String clientIp = getIpByClient(client);

log.debug(clientIp + " *********************** " + "客户端已断开连接");

String userId = getParamsByClient(client);

if (userId != null) {

clientMap.remove(userId);

client.disconnect();

}

});

// 自定义事件`client_info_event` -> 监听客户端消息

socketIOServer.addEventListener(PUSH_DATA_EVENT, String.class, (client, data, ackSender) -> {

// 客户端推送`client_info_event`事件时，onData接受数据，这里是string类型的json数据，还可以为Byte[],object其他类型

String clientIp = getIpByClient(client);

log.debug(clientIp + " ************ 客户端：" + data);

});

// 启动服务

socketIOServer.start();

// broadcast: 默认是向所有的socket连接进行广播，但是不包括发送者自身，如果自己也打算接收消息的话，需要给自己单独发送。

new Thread(() -> {

int i = 0;

while (true) {

try {

// 每3秒发送一次广播消息

Thread.sleep(3000);

socketIOServer.getBroadcastOperations().sendEvent("myBroadcast", "广播消息 " + DateUtil.now());

} catch (InterruptedException e) {

e.printStackTrace();

}

}

}).start();

}

@Override

public void stop() {

if (socketIOServer != null) {

socketIOServer.stop();

socketIOServer = null;

}

}

@Override

public void pushMessageToUser(String userId, String msgContent) {

SocketIOClient client = clientMap.get(userId);

if (client != null) {

client.sendEvent(PUSH_DATA_EVENT, msgContent);

}

}

/**

* 获取客户端url中的userId参数(这里根据个人需求和客户端对应修改即可)

*

* @param client: 客户端

* @return: java.lang.String

*/

private String getParamsByClient(SocketIOClient client) {

// 获取客户端url参数(这里的userId是唯一标识)

Map> params = client.getHandshakeData().getUrlParams();

List userIdList = params.get("userId");

if (!CollectionUtils.isEmpty(userIdList)) {

return userIdList.get(0);

}

return null;

}

/**

* 获取连接的客户端ip地址

*

* @param client: 客户端

* @return: java.lang.String

*/

private String getIpByClient(SocketIOClient client) {

String sa = client.getRemoteAddress().toString();

String clientIp = sa.substring(1, sa.indexOf(":"));

return clientIp;

}

}