解决 SpringMVC 非spring管理的工具类使用@Autowired注解注入DAO为null的问题

有时候在工具类中需要要操作数据库,即需要注入springMVC中的DAO层,但是此时工具类并非spring管理,所以使用@Autowired注入会报空指针异常,在试了好多方案都不行时,找到了比较适合springMVC的解决方法:

(1)在配置文件(springmvc.xml)中加入工具类所在包的扫描配置

(2)使用@Component注解将工具类声明为spring组件,并静态初始化

代码截取:

@Component
@ServerEndpoint("/websocket/{nowUid}")
public class WebSocketTest {
	
	@Autowired
	private ShareDao shareDao;   
        //...省略
    //静态初始化  
    public static WebSocketTest webSocketTest;
    //保证Bean初始化前已经装配了属性
    @PostConstruct
    public void init() {
    	webSocketTest = this;
    }
    //...
    webSocketTest.shareDao.addToChatLog(nowUid, toUid, message, 1); //使用


参考如下代码:

package com.yyy.school.share.util;

import java.io.IOException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.ConcurrentHashMap;

import javax.annotation.PostConstruct;
import javax.websocket.EndpointConfig;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import com.yyy.school.share.dao.ShareDao;

/**
 * @ServerEndpoint 注解是一个类层次的注解,它的功能主要是将目前的类定义成一个websocket服务器端,
 * 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
 * @ServerEndpoint 可以把当前类变成websocket服务类
 */
@Component
@ServerEndpoint("/websocket/{nowUid}")
public class WebSocketTest {
	
	@Autowired
	private ShareDao shareDao;
	
    //静态变量,用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;
    //concurrent包的线程安全Set,用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话,可以使用Map来存放,其中Key可以为用户标识
    private static ConcurrentHashMap<String, WebSocketTest> webSocketSet = new ConcurrentHashMap<String, WebSocketTest>();
    //与某个客户端的连接会话,需要通过它来给客户端发送数据
    private Session WebSocketsession;
    //当前发消息的人员编号
    private String nowUid = ""; 
    //静态初使化当前类
    public static WebSocketTest webSocketTest;
    //在方法上加上注解@PostConstruct,保证Bean初始化前已经装配了属性(注:Bean初始化包括,实例化Bean,并装配Bean的属性(依赖注入))
    @PostConstruct
    public void init() {
    	webSocketTest = this;
    }
    

    /**
     * 连接建立成功调用的方法
     *
     * @param session 可选的参数。session为与某个客户端的连接会话,需要通过它来给客户端发送数据
     */
	@OnOpen
    public void onOpen(@PathParam(value = "nowUid") String param, Session WebSocketsession, EndpointConfig config) {
        System.out.println(param);
        nowUid = param;//接收到发送消息的人员编号
        this.WebSocketsession = WebSocketsession;
        webSocketSet.put(param, this);//加入map中
        addOnlineCount();           //在线数加1
        System.out.println("有新连接加入!当前在线人数为" + getOnlineCount());
       
    }


    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose() {
        if (!nowUid.equals("")) {
            webSocketSet.remove(nowUid);  //从set中删除
            subOnlineCount();           //在线数减1
            System.out.println("有一连接关闭!当前在线人数为" + getOnlineCount());
        }
    }


    /**
     * 收到客户端消息后调用的方法
     *
     * @param message 客户端发送过来的消息
     * @param session 可选的参数
     */
    @SuppressWarnings("unused")
	@OnMessage
    public void onMessage(String message, Session session) {
        System.out.println("来自客户端的消息:" + message);
        sendToUser(message);
//        //群发消息
//        if (1 < 2) {
//            sendAll(message);
//        } else {
//            //给指定的人发消息
//            sendToUser(message);
//        }
    }


    /**
     * 给指定的人发送消息
     * @param message
     */
    public void sendToUser(String messageStr) {
    	String toUid = messageStr.split("[|]")[1];
        String message = messageStr.split("[|]")[0];
        String now = getNowTime();
        try {
            if (webSocketSet.get(toUid) != null) {
                webSocketSet.get(toUid).sendMessage(now + "用户" + nowUid + "发来消息:" + " <br/> " + message);
                webSocketTest.shareDao.addToChatLog(nowUid, toUid, message, 1);
            } else {
                System.out.println(toUid + "当前用户不在线"); 
                webSocketTest.shareDao.addToChatLog(nowUid, toUid, message, 0);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }




	/**
     * 给所有人发消息
     * @param message
     */
    private void sendAll(String message) {
        String now = getNowTime();
        String sendMessage = message.split("[|]")[0];
        //遍历HashMap
        for (String key : webSocketSet.keySet()) {
            try {
                //判断接收用户是否是当前发消息的用户
                if (!nowUid.equals(key)) {
                    webSocketSet.get(key).sendMessage(now + "用户" + nowUid + "发来消息:" + " <br/> " + sendMessage);
                    System.out.println("key = " + key);
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }




    /**
     * 获取当前时间
     *
     * @return
     */
    private String getNowTime() {
        Date date = new Date();
        DateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        String time = format.format(date);
        return time;
    }
    /**
     * 发生错误时调用
     *
     * @param session
     * @param error
     */
    @OnError
    public void onError(Session session, Throwable error) {
        System.out.println("发生错误");
        error.printStackTrace();
    }


    /**
     * 这个方法与上面几个方法不一样。没有用注解,是根据自己需要添加的方法。
     *
     * @param message
     * @throws IOException
     */
    public void sendMessage(String message) throws IOException {
        this.WebSocketsession.getBasicRemote().sendText(message);
        //this.session.getAsyncRemote().sendText(message);
    }


    public static synchronized int getOnlineCount() {
        return onlineCount;
    }


    public static synchronized void addOnlineCount() {
        WebSocketTest.onlineCount++;
    }


    public static synchronized void subOnlineCount() {
        WebSocketTest.onlineCount--;
    }


}


猜你喜欢

转载自blog.csdn.net/z595746437/article/details/80340836