使用spring動態獲取接口的不同實現類

Posted on by

spring動態獲取接口的不同實現類

最近做項目，有個需求是和外部對接，從接口獲取新聞數據，把數據和緩存中的數據對比，多瞭的添加到數據庫，少瞭的刪掉。

接口有兩個，一開始我是在業務類裡寫瞭兩個方法，代碼太長，簡單說就是兩個部分：

public Object saveANews() {
	//1、獲取A接口新聞列表
    //2、和緩存對比，存數據到數據庫
}
 
public Object saveBNews() {
	//1、獲取B新聞列表
    //2、和緩存對比，存數據到數據庫
}

寫完後我發現，因為操作的是數據庫的同一張表，2的部分代碼完全一模一樣，隻有1的部分不同，而1的部分其實就隻有一行代碼。。。

這必須得復用啊，而且是一個業務，也沒必要分別用兩個方法，於是我改成瞭這樣：

//業務接口實現方法
public Object saveNews(NewsUtilService service) {
	//1、獲取接口新聞列表
    List<NewsVO> list = service.queryNews();
    //2、和緩存對比，存數據到數據庫
}
 
//定義新聞數據接口
public interface NewsUtilService {
    List<NewsVO> queryNews();
}
 
//接口的兩個實現類
@Service
public class ANewsDataServiceImpl implements NewsUtilService { 
    @Autowired
	private NewsDataMapper newsDataMapper;
 
    @Override
	public List<NewsVO> queryNews(){
		//對接數據
	}
}
 
@Service
public class BNewsDataServiceImpl implements NewsUtilService {
    @Override
	public List<NewsVO> queryNews(){
		//對接數據
	}
}
 
//定義工廠類
@Service
public class NewsUtilServiceFactory {	
	/**
	 * 
	 * Method Name:  getNewsUtilService
	 * @param source
	 * @return 
	 */
	public NewsUtilService getNewsUtilService(String source){
		switch(source){
			case "a":
				return new ANewsDataServiceImpl();
			case "b":
				return new BNewsDataServiceImpl();
			default:
				return null;
		}
	}
}
 
//控制層調用
@RestController
public class NewsDataController {
    @Resource
    private NewsDataService newsDataService;
    
    @Resource
    private NewsUtilServiceFactory factory;
 
    public Object getNewsData(){
        String[] sources = {"a","b"};
    		for (int i = 0; i < sources.length; i++) {
    			NewsUtilService newsUtilService = factory.getNewsUtilService(sources[i]);
    			newsDataService.saveNews(newsUtilService);
			}
    }
}

本來以為這就大工告成瞭，誰知道運行後控制臺居然報錯瞭：

經過一步步調試，總算發現瞭是什麼問題：

其中一個實現類中註入的Mapper沒有實例化，是null。

一開始我還以為是構造函數調用和註入的順序問題，查瞭半天才明白不是，問題在這裡：

使用new關鍵字實例化的對象不是被spring創建的，不歸spring管，所以A類實現類中Mapper註入的註解根本不生效！

但是因為業務需要，那個mapper又需要用到，怎麼辦呢？

當時想到瞭兩種解決辦法

1、在接口的方法參數裡加入mapper，把mapper作為參數傳進去，但這實在太奇怪瞭，先不說B類實現類根本用不到mapper，而且一個接口定義出來後根本不管它的實現類吧，因為實現類的問題去改接口，，，似乎有點非呀。

於是決定用第二種，修改工廠類變成如下：

//定義工廠類
@Service
public class NewsUtilServiceFactory {	
    @Autowired
    private ANewsDataServiceImpl aNewsDataServiceImpl;
    @Autowired
    private BNewsDataServiceImpl bNewsDataServiceImpl; 
	public NewsUtilService getNewsUtilService(String source){
		switch(source){
			case "a":
				return aNewsDataServiceImpl;
			case "b":
				return bNewsDataServiceImpl;
			default:
				return null;
		}
	}
}

代碼寫出來自己都無語瞭，先把所有的實現類都實例化出來，在根據輸入返回。這不是工廠模式，是商店模式吧。。。但是當時也想不到其他辦法，就先這麼寫瞭，但一直覺得肯定有其他解決方案，直到今天有空，去查瞭一下，才發現自己都多low。。。

其實spring可以動態獲取實現類的~~~

@Service
public class NewsUtilServiceFactory {	
	@Autowired
	private ApplicationContext applicationContext;	
	public NewsUtilService getNewsUtilService(String source){
		switch(source){
			case "web":
				return applicationContext.getBean(WebNewsDataServiceImpl.class);
			case "oa":
				return applicationContext.getBean(OANewDataServiceImpl.class);
			default:
				return null;
		}
	}
}

這才是正確寫法有木有！

總算弄出來瞭，趕緊記錄下來先~

獲取某接口所有實現類

在springboot項目中，為瞭方便，我們可能需要獲取某一個接口下面的所有實現類，根據名稱進行匹配使用。

正文

1、ServiceLocator.java

package com.yang.config;
import com.yang.workOrder.service.IRootService;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;
import java.util.Map;
/**
 * explain：獲取應用上下文並獲取相應的接口實現類
 *
 * @author yang
 * @date 2021/1/5
 */
@Component
public class ServiceLocator implements ApplicationContextAware {
    /**
     * 用於保存接口實現類名及對應的類
     */
    private Map<String, IRootService> map;
    /**
     * 獲取應用上下文並獲取相應的接口實現類
     * @param applicationContext
     * @throws BeansException
     */
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        //根據接口類型返回相應的所有bean
        map = applicationContext.getBeansOfType(IRootService.class);
    }
    /**
     * 獲取所有實現集合
     * @return
     */
    public Map<String, IRootService> getMap() {
        return map;
    }
    /**
     * 獲取對應服務
     * @param key
     * @return
     */
    public IRootService getService(String key) {
        return map.get(key);
    }
}

2、IRootService.java

package com.yang.workOrder.service;
import com.alibaba.fastjson.JSONObject;
import com.yang.workOrder.entity.WorkOrder;
/**
 * explain：基礎流程操作服務接口
 *
 * @author yang
 * @date 2021/1/5
 */
public interface IRootService {
    /**
     * 開始流程
     * @param workOrder
     * @return
     */
    boolean startProcess(WorkOrder workOrder);
}

3、RootA001ServiceImpl.java

package com.yang.workOrder.service.impl;
import com.alibaba.fastjson.JSONObject;
import com.yang.workOrder.entity.WorkOrder;
import com.yang.workOrder.service.IRootService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;
/**
 * explain：A_001流程審批實現類
 *
 * @author yang
 * @date 2021/1/5
 */
@Service("A_001")
public class RootA001ServiceImpl implements IRootService {
    private static final Logger LOGGER = LoggerFactory.getLogger(RootA001ServiceImpl.class);
    @Override
    public boolean startProcess(WorkOrder workOrder) {
        return false;
    }
}

4、RootA002ServiceImpl.java

package com.yang.workOrder.service.impl;
import com.alibaba.fastjson.JSONObject;
import com.yang.workOrder.entity.WorkOrder;
import com.yang.workOrder.service.IRootService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;
/**
 * explain：A_002流程審批實現類
 *
 * @author yang
 * @date 2021/1/5
 */
@Service("A_002")
public class RootA002ServiceImpl implements IRootService {
    private static final Logger LOGGER = LoggerFactory.getLogger(RootA002ServiceImpl.class);
    @Override
    public boolean startProcess(WorkOrder workOrder) {
        return false;
    }
}

結果

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

推薦閱讀：