springboot實現在工具類(util)中調用註入service層方法

Posted on by

一、新建BeanUtil類

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import java.util.ArrayList;
import java.util.List;
 
@Slf4j
public class BeanUtil implements ApplicationContextAware, DisposableBean {
    private static ApplicationContext applicationContext = null;
 
    /**
     * 從靜態變量applicationContext中取得Bean, 自動轉型為所賦值對象的類型.
     */
    public static <T> T getBean(Class<T> requiredType) {
        if(applicationContext==null){
            throw new IllegalStateException("applicaitonContext屬性未註入, 請在SpringBoot啟動類中註冊BeanUtil.");
        }
        return applicationContext.getBean(requiredType);
    }
  
    @Override
    public void destroy() {
        applicationContext = null;
    }
 
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        if (BeanUtil.applicationContext != null) {
            log.warn("BeanUtil中的ApplicationContext被覆蓋, 原有ApplicationContext為:" + BeanUtil.applicationContext);
        }
        BeanUtil.applicationContext = applicationContext;
    }
}

二、在啟動類中註入BeanUtil

@SpringBootApplication
public class StartApplication {
 
    public static void main(String[] args) {
        SpringApplication.run(StartApplication.class, args);
    }
  
    @Bean
    public BeanUtil beanUtil() {
        return new BeanUtil();
    }
}

三、在Util類中獲取Service

public class MyUtils {
   
    public static UserDetails getCurrentUser() {
        MyUserDetailsService userDetailsService = BeanUtil.getBean(MyUserDetailsService.class);
        //TODO 調用service中的方法處理自己的業務
        return null;
    }
}

springBoot中普通類直接調用spring管理的dao.service等bean

在項目中經常會用到一些不是controller和service層的去訪問dao層，或者是普通類訪問service層。

前提：

首先確保dao，serivce全部可以在項目啟動的時候掃描到

編寫一個工具類，通過工具類去獲取ApplicationContext對象，然後通過applicationContext對象去加載spring管理的bean

將該工具類在項目啟動類進行引入，否則會報空指針異常。

具體使用方法如下：

1）編寫工具

package com.bonc.springboot.utils;
/*
 * @desc:提供非SPRING管理類調用管理類的功能
 * 註意在服務啟動的時候進行import,apllication中引入
 */
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
public class SpringUtil implements ApplicationContextAware {
       private static ApplicationContext applicationContext = null;
       @Override
 
       public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
 
              if(SpringUtil.applicationContext == null){
                     SpringUtil.applicationContext  = applicationContext;
              }
              System.out.println("---------------------------------------------------------------------");
              System.out.println("========ApplicationContext配置成功,在普通類可以通過調用SpringUtils.getAppContext()獲取applicationContext對象,applicationContext="+SpringUtil.applicationContext+"========");
              System.out.println("---------------------------------------------------------------------");
       }
     
       //獲取applicationContext
       public static ApplicationContext getApplicationContext() {
              return applicationContext;
       }
      
       //通過name獲取 Bean.
       public static Object getBean(String name){
              return getApplicationContext().getBean(name);
       }
 
       //通過class獲取Bean.
       public static <T> T getBean(Class<T> clazz){
              return getApplicationContext().getBean(clazz);
       }
 
       //通過name,以及Clazz返回指定的Bean
       public static <T> T getBean(String name,Class<T> clazz){
              return getApplicationContext().getBean(name, clazz);
       }
}

2）將該工具類，在服務啟動類application 中導入

3）普通類中使用APPlicationContent對象獲取對應的bean進行使用：

ApplicationContext appCtx = SpringUtil.getApplicationContext();
public  TestDao dao = appCtx.getBean(TestDao.class);

也可寫成

public  TestDao dao = SpringUtil.getBean(TestDao.class);

具體測試如下：

例：比如有一個請求中調用瞭實體類，實體類的構造方法調用的dao,進行的數據庫的訪問

請求：

//普通類進行構造的時候，可以調用dao 對象
@RequestMapping(value="/testClassQuery")
public  void  testClassQuery(){
  TestClassQuery testClassQuery =new TestClassQuery();
}

普通類代碼：

**
 * @Description: 測試普通類如何調用mapper
 */
 
public  class TestClassQuery {
    ApplicationContext appCtx = SpringUtil.getApplicationContext();
    public  TestDao dao = appCtx.getBean(TestDao.class);
 
    private  User user;
 
    public User getUser() {
        return user;
    }
    public void setUser(User user) {
        this.user = user;
    }
 
    public  TestClassQuery(){
        testQuery();
    }
 
    public  void testQuery(){
        user = dao.queryUser("1"); //通過id獲取值
        System.out.println(user.toString());
    }
}

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

推薦閱讀：