Bootstrap

java websocket使用

当前项目中由于用到了webSocket 于是做一个总结,当前项目中由于需要后台监控日志程序,所以需要一直给前段发送请求,这就用到了webSocket以下是代码

依赖

		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-websocket</artifactId>
		</dependency>

配置bean


@Component
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

service

@Component
@ServerEndpoint("/fileWebSocket")
public class FileWebSocket {
    private Logger log = LoggerFactory.getLogger(FileWebSocket.class);
    private Session session;

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

    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        webSocketSet.add(this);
        log.info("【websocket消息】有新的连接," + "id为:" + session.getId() + " 总数:{}", webSocketSet.size());
    }


    @OnClose
    public void onClose() {
        webSocketSet.remove(this);
        log.info("【websocket消息】连接断开, 总数:{}", webSocketSet.size());
    }

    @OnMessage
    public void onMessage(Session session, String message){
        log.info("【websocket消息】收到" + session.getId() + "发来的消息:{}", message);
        log.info("收到来自" + session.getId() + "的信息" + message);
        sendMessage(message);
    }

    public void sendMessage(String message) {
        for (FileWebSocket webSocket : webSocketSet) {
//            log.info(session.getId() + "发送的消息, message={}", message);
            try {
                webSocket.session.getBasicRemote().sendText(message);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
    
     public void sendMessage(Map message) {
        for (FileWebSocket webSocket : webSocketSet) {
//            log.info(session.getId() + "发送的消息, message={}", message);
            try {
                webSocket.session.getBasicRemote().sendText(JSONUtil.toJsonStr(message));
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}
;