package com.example.socketserver.websocket;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.ConcurrentHashMap;
/**
* @author Administrator
* @description 接收客户端请求连接
* @date 2021-01-07 10:09
*/
@Component
@ServerEndpoint(value = "/{ip}")
public class WebSocketServer {
// 在多线程访问的时候，使用线程安全的ConCurrentHashMap对象
private static ConcurrentHashMap<String, Session> connections = new ConcurrentHashMap<>();
private static ApplicationContext applicationContext;
public static void setApplicationContext(ApplicationContext applicationContext) {
WebSocketServer.applicationContext = applicationContext;
}
/**
* 打开连接
* @param session
* @param ip
*/
@OnOpen
public void onOpen(Session session, @PathParam("ip") String ip) {
System.out.println("=====接受到的请求地址=====" + ip);
// 接收到客户端的请求，可以做一些其他业务逻辑处理，比如可以把该IP存储到数据库
// 避免当前服务断开后，与客户端服务失去连接
// 这时就可以使用到预加载处理，项目当中自定义的MyApplicationRunner类
connections.put(ip, session);
}
/**
* 接收消息
* @param text
*/
@OnMessage
public void onMessage(String text) {
}
/**
* 异常处理
* @param throwable
*/
@OnError
public void onError(Throwable throwable) {
throwable.printStackTrace();
}
/**
* 关闭连接
* @param ip
*/
@OnClose
public void onClosing(@PathParam("ip") String ip) throws IOException {
System.out.println("=====关闭连接地址=====" + ip);
connections.remove(ip);
}
/**
* 根据IP发送消息
* @param ip
* @param text
*/
public void send(String ip, String text) {
try {
Session session = connections.get(ip);
if (session != null && session.isOpen()) {
session.getAsyncRemote().sendText(text);
}
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 遍历群发消息
* @param text
*/
public void send(String text) {
for (ConcurrentHashMap.Entry<String, Session> entry : connections.entrySet()) {
send(entry.getKey(), text);
}
}
}
SpringBoot+nacos+websocket+redis+mysql+mybatis-plus微服务项目实战.zip （27个子文件） 
SpringBoot+nacos+websocket+redis+mysql+mybatis-plus微服务项目实战 src test java com example socketserver 
SocketServerApplicationTests.java 276B main resources application-uat.yml 18B application-dev.yml 21B application-prod.yml 21B mapper UserMapper.xml 649B application-test.yml 1KB application.yml 1KB java com example socketserver service impl UserServiceImpl.java 1KB FeignServiceImpl.java 391B UserService.java 225B FeignService.java 186B dto UserDTO.java 145B controller FeignController.java 758B UserController.java 1KB mapper UserMapper.java 421B websocket WebSocketServer.java 3KB util RedisUtils.java 2KB config WebSocketConfig.java 526B properties RedisConfigProperties.java 1KB MyApplicationRunner.java 794B Swagger2Config.java 1KB MybatisPlusConfig.java 1KB RedissonConfig.java 2KB entity UserEntity.java 153B vo UserVo.java 413B SocketServerApplication.java 656B pom.xml 7KB资源评论
