spring boot集成websocket

pom

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

ServerEndpointExporter

@Configuration
@EnableWebSocket
@Service
public class WebSocketConfig {

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

工具类

package com.plus.config.message;


import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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.Collection;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@Component
@ServerEndpoint("/websocket/{user}")
public class MyWebSocket {

    private static final Logger logger = LoggerFactory.getLogger(MyWebSocket.class);

    public static Map<String, Session> clients = new ConcurrentHashMap<String, Session>();

    /**
     * 打开连接时触发
     *
     * @param myWebsocket
     * @param session
     */
    @OnOpen
    public void onOpen(@PathParam("user") String myWebsocket, Session session) {
       
        logger.info("进入：++++++++++++++++++++++++++++++++" + myWebsocket);
       
        clients.put(myWebsocket, session);

//        BROADCAST(myWebsocket+"上线了");
    }

    /**
     * 收到客户端消息时触发
     *
     * @param myWebsocket
     * @param message
     * @return
     */
    @OnMessage
    public void onMessage(@PathParam("user") String myWebsocket, String message, Session session) {
		System.out.println(message);
        if(null==clients.get(myWebsocket)) {
            clients.put(myWebsocket, session);
        }

    }

    /**
     * 异常时触发
     *
     * @param myWebsocket
     * @param throwable
     */
    @OnError
    public void onError(@PathParam("user") String myWebsocket, Throwable throwable) {

         
        try {
            BROADCAST(myWebsocket,"keep");
        }catch(Exception e) {
            clients.remove(myWebsocket);
        }
    }

    /**
     * 关闭连接时触发
     *
     * @param myWebsocket
     */
    @OnClose
    public void onClose(@PathParam("user") String myWebsocket) {
       
        System.out.println("websoket关闭：" + myWebsocket);
        clients.remove(myWebsocket);
    }

    /**
     * 将数据传回客户端 异步的方式
     *
     * @param myWebsocket
     * @param message
     */
    public static void BROADCAST(String myWebsocket, String message) {
        try {
            if (clients.containsKey(myWebsocket)) {
              
                clients.get(myWebsocket).getAsyncRemote().sendText(message);
            } else {
                logger.info("websocket未找到" + myWebsocket);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    

    /**
     * 群发  待测试
     * @param message
     */
    public static void BROADCAST(String message) throws IOException {
        Collection<Session> values = clients.values();
        for (Session s:values) {
            if(s!=null){

                s.getBasicRemote().sendText(message);
            }else {
                System.out.println("用户不存在");
            }

        }
    }




}