java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > @ServerEndpoint不能注入@Autowired

解决@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;
    }

}

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文