解决@ServerEndpoint不能注入@Autowired的问题
作者:于小袁
这篇文章主要介绍了解决@ServerEndpoint不能注入@Autowired的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
都会报空指针错误的原因
本质原因
spring管理的都是单例(singleton)和 websocket (多对象)相冲突。
因为WebSocket 是一个多例,因为你需要多个对象来保存链接,所以就和单例无缘了,然而我们交给Spring管理的是单例的东西,那么我们可以从这里知道,WebSocket的对象是不能交给Spring管理的
此时我们有两种解决方案;
方案1: 在上下文获取你需要的对象
package cn.jiabao.util; import org.springframework.beans.BeansException; import org.springframework.beans.factory.config.BeanFactoryPostProcessor; import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; import org.springframework.stereotype.Repository; @Repository public class SpringUtils implements BeanFactoryPostProcessor { //Spring应用上下文环境 private static ConfigurableListableBeanFactory beanFactory; @Override public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException { SpringUtils.beanFactory = beanFactory; } public static ConfigurableListableBeanFactory getBeanFactory() { return beanFactory; } @SuppressWarnings("unchecked") public static <T> T getBean(String name) throws BeansException { return (T) getBeanFactory().getBean(name); } public static <T> T getBean(Class<T> clz) throws BeansException { T result = (T) getBeanFactory().getBean(clz); return result; } } -------------------------------------------------------- private StringRedisTemplate stringRedisTemplate = SpringUtils.getBean(StringRedisTemplate.class); @OnOpen public void OnOpen(Session session, @PathParam(value = "name") String name,@PathParam(value = "password") String password) { stringRedisTemplate.opsForValue().set("1","1"); this.session = session; this.name = name; // name是用来表示唯一客户端,如果需要指定发送,需要指定发送通过name来区分 webSocketSet.put(name, this); log.info("[WebSocket] 连接成功,当前连接人数为:={}", webSocketSet.size()); //接收用户传入的参数,并且进行判断登录是否成功 如果登录成功那 personService.savePersonSql(name,password);么 就保存Sesion 并且 链接成功 否职反之 // personService.savePersonSql(name,"1"); //多例之后 可以执行reids代码等等 // 注意: 不能在这里面直接用Reids 即使多例 直接保存也有问题 }
使用上面的工具类就,可以从上下文获取到你需要的实例对象
方案2: 既然不给那我就自己搞一个注入进去
private static StringRedisTemplate stringRedisTemplate; @Autowired private void setStringRedisTemplate(StringRedisTemplate stringRedisTemplate){ Link.stringRedisTemplate =stringRedisTemplate; }
心得
这里Link的多对象机制和Spring 的controller注解截止可以同时进行,互相没有矛盾。
Spring在初始的时候,创建一个Link的对象,该对象在运行期间不会被使用,同时注入的类进行了静态的完善,这是Spring在这套代码里面的唯一作用,当有用户聊天的时候,java会根据Link类创建对象,每个对象保持对应的用户链接,因为静态类已经在启动的时候被Spring初始化了,所以此时每个对象都可以被正常访问
注意:
@Component 这个注解需要有,不然就不能被加载到Spring 就不能在启动服务器的时候进行正常的链接
package cn.jiabao.link; import cn.jiabao.util.SpringUtils; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.redis.core.StringRedisTemplate; import org.springframework.stereotype.Component; import javax.websocket.OnMessage; import javax.websocket.OnOpen; import javax.websocket.Session; import javax.websocket.server.PathParam; import javax.websocket.server.ServerEndpoint; import java.util.concurrent.ConcurrentHashMap; @Slf4j @ServerEndpoint("/websocket/{name}/{password}") @Component @Controller public class Link { private static StringRedisTemplate stringRedisTemplate; @Autowired private void setStringRedisTemplate(StringRedisTemplate stringRedisTemplate){ Link.stringRedisTemplate =stringRedisTemplate; } /** * 与某个客户端的连接对话,需要通过它来给客户端发送消息 */ private Session session; /** * 标识当前连接客户端的用户名 */ private String name; /** * 用于存所有的连接服务的客户端,这个对象存储是安全的 */ private static ConcurrentHashMap<String, Link> webSocketSet = new ConcurrentHashMap<>(); // private StringRedisTemplate stringRedisTemplate = SpringUtils.getBean(StringRedisTemplate.class); @OnOpen public void OnOpen(Session session, @PathParam(value = "name") String name,@PathParam(value = "password") String password) { stringRedisTemplate.opsForValue().set("1","1"); this.session = session; this.name = name; // name是用来表示唯一客户端,如果需要指定发送,需要指定发送通过name来区分 webSocketSet.put(name, this); log.info("[WebSocket] 连接成功,当前连接人数为:={}", webSocketSet.size()); //接收用户传入的参数,并且进行判断登录是否成功 如果登录成功那 personService.savePersonSql(name,password);么 就保存Sesion 并且 链接成功 否职反之 // personService.savePersonSql(name,"1"); //多例之后 可以执行reids代码等等 // 注意: 不能在这里面直接用Reids 即使多例 直接保存也有问题 } @OnMessage public void OnMessage(String message) { log.info("[WebSocket] 收到消息:{}", message); //判断是否需要指定发送,具体规则自定义 if (message.indexOf("TOUSER") == 0) { String name = message.substring(message.indexOf("TOUSER") + 6, message.indexOf(";")); AppointSending(name, message.substring(message.indexOf(";") + 1, message.length())); } else { GroupSending(message); } } /** * 群发 * * @param message */ public void GroupSending(String message) { for (String name : webSocketSet.keySet()) { try { webSocketSet.get(name).session.getBasicRemote().sendText(message); } catch (Exception e) { e.printStackTrace(); } } } /** * 指定发送 * * @param name * @param message */ public void AppointSending(String name, String message) { try { webSocketSet.get(name).session.getBasicRemote().sendText(message); } catch (Exception e) { e.printStackTrace(); } } public static Link getPerson(String key) { return webSocketSet.get(key); } public static ConcurrentHashMap<String, cn.jiabao.link.Link> getMap() { return webSocketSet; } }
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。