Spring Boot WebSocket使用及配置

 1--依赖

<!-- webSocket依赖-->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2--配置bean

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

3--.使webSocket可以发送Object

//必要的
public class UserEncoder implements javax.websocket.Encoder.Text<ResultJson> {
    @Override
    public String encode(ResultJson resultJson) throws EncodeException {
        try {
            return JSONObject.toJSONString(resultJson);
        } catch (Exception e) {
            log.error("", e);
        }
        return null;
    }
    @Override
    public void init(EndpointConfig endpointConfig) {}
    @Override
    public void destroy() {}
}
//必要的
public class GetHeepSessionConfigurator extends ServerEndpointConfig.Configurator {
    @Override
    public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
        HttpSession httpSession = (HttpSession) request.getHttpSession();
        sec.getUserProperties().put(HttpSession.class.getName(), httpSession);
    }
}
//非必要的
public class UserDecoder implements javax.websocket.Decoder.Text<SysUser> {
    @Override
    public void destroy() {}
    @Override
    public void init(EndpointConfig arg0) {}
    @Override
    public SysUser decode(String user) throws DecodeException {
        return JSON.parseObject(user, SysUser.class);
    }
    @Override
    public boolean willDecode(String arg0) {
        return true;
    }
}

3.实现类

//value后面的{}里面的数据为前台传的参数，下面可以通过加入@PathParam注解来指定接收
//decoders 和encoders 第二个是对前台返回的Object类型进行编译
@Component
@ServerEndpoint(value = "/chatWebSocket/{userId}", decoders = {UserDecoder.class}, encoders = {UserEncoder.class})
public class ChatWebSocket {
    //websocket解决无法注入Service
    private static UserFeedBackService userFeedBackService;
    @Autowired //这里可以写多个Service
    public void getServiceImpl(UserFeedBackService userFeedBackService) {
        this.userFeedBackService = userFeedBackService;
    }
    //通过工具类注入redis工具类
    private StringRedisTemplate stringRedisTemplate = SpringUtils.getBean(RedisUtil.class).getStringRedisTemplate();
    
    @OnOpen
    public void onOpen(Session session, @PathParam("userId") Long userId) throws IOException, EncodeException {
        //向前台发送的方法
        session.getBasicRemote().sendObject(ResultJson.error("这里可以随便放Object类"));
        //第二种
        //session.getAsyncRemote().sendObject(ResultJson.ok(Constant.LOGIN_OK, json));
        //这两种区别待找：字面意思是基本的和异步的；也可以发送文本sendText
        System.out.println("这是前台发起连接的时候调用");   
    }
    @OnMessage
    public void onMessage(String message, Session session) {
         System.out.println("这是前台发送消息的时候调用");   
        //用Redis发布订阅消息
        stringRedisTemplate.convertAndSend(Constant.REDIS_CHANNEL_CHAT, message);
    }
    @OnClose
    public void onClose(Session session, @PathParam("userId") Long userId) {
       System.out.println("这是前台关闭session的时候调用");   
    }
}