spring boot项目启动websocket

1. pom.xml引入:

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

2. 创建配置类:WebSocketConfig

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

3. 自定义配置类(为了保证websocket能够像controller一样使用):SpringContextHelper

// 自定义配置类
@Component
public class SpringContextHelper extends ServerEndpointConfig.Configurator implements ApplicationContextAware {
    //  通过手动注入applicationContext上下文获取Bean
    private static volatile BeanFactory context;
    @Override
    public <T> TgetEndpointInstance(Class<T> clazz) throws InstantiationException {
        return context.getBean(clazz);
}
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException
    {
        SpringContextHelper.context = applicationContext;
  }
}

4、创建websocket监听类:Web1

import com.alibaba.fastjson.JSON;
import com.epf.app.redis.SmsRedisService;
import com.epf.csce.config.SpringContextHelper;
import com.epf.csce.dao.BjMyCollectDao;
import com.epf.csce.dao.BjNewsTrendsDao;
import com.epf.csce.entity.BjMyCollectEntity;
import com.epf.csce.entity.BjNewsTrendsEntity;
import com.epf.csce.entity.WxSocketEntity;
import com.epf.csce.service.BjPopularRecommendService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.CopyOnWriteArraySet;

@ServerEndpoint(value= "/wxsmall/web1", configurator= SpringContextHelper.class)
@Component
public class Web1 {

    private static Logger log = LoggerFactory.getLogger(Web1.class);

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

    //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
    private static CopyOnWriteArraySet<Web1> webSocketSet = new CopyOnWriteArraySet<Web1>();

    //与某个客户端的连接会话，需要通过它来给客户端发送数据
    private Session session;

    @Autowired
    private SmsRedisService redisService;

    @Autowired
    private BjPopularRecommendService bjPopularRecommendService;

    @Resource
    private BjMyCollectDao bjMyCollectDao;

    @Resource
    private BjNewsTrendsDao bjNewsTrendsDao;

    /**
        * 连接建立成功调用的方法
    */
    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        webSocketSet.add(this);    //加入set中
        addOnlineCount();          //在线数加1
        log.info("有新连接加入！当前在线人数为" + getOnlineCount());
        try {
            sendMessage("连接成功");
        } catch (IOException e) {
            log.error("websocket IO异常");
}
}
    /**
        * 连接关闭调用的方法
    */
    @OnClose
    public void onClose() {
        webSocketSet.remove(this);  //从set中删除
        subOnlineCount();          //在线数减1
        log.info("有一连接关闭！当前在线人数为" + getOnlineCount());
}
    /**
        * 收到客户端消息后调用的方法
        * @parammessage 客户端发送过来的消息
    */
    @OnMessage
    public void onMessage(String message, Session session) {
        log.info("来自客户端的消息:" + message);
        //群发消息
        for (Web1 item: webSocketSet) {
            String result= "";
            try {
                item.sendMessage(result);
            } catch (IOException e) {
                e.printStackTrace();
}
}
}
    /**
    * @paramsession
    * @paramerror
    */
    @OnError
    public void onError(Session session, Throwable error) {
        log.error("发生错误");
        error.printStackTrace();
}
    /**
    * 服务器主动推送
    * @parammessage
    * @throwsIOException
    */
    public void sendMessage(String message) throws IOException {
        this.session.getBasicRemote().sendText(message);
}
    /**
    * 群发自定义消息
  */
    public static void sendInfo(String message) throws IOException {
        log.info(message);
        for (Web1 item: webSocketSet) {
            try {
                item.sendMessage(message);
            } catch (IOException e) {
                continue;
}
}
}
    public static synchronized int getOnlineCount() {
        return onlineCount;
}
    public static synchronized void addOnlineCount() {
        Web1.onlineCount++;
}
    public static synchronized void subOnlineCount() {
        Web1.onlineCount--;
  }
}