springboot如何通過不同的策略動態呼叫不同的實現類

2022-02-26 13:00:27

通過不同的策略動態呼叫不同的實現類

經常遇到這樣的一個需求，前端傳的實體型別相同，後端需要根據實體類中的某一個字串，動態地呼叫某一個類的方法。

在SpringBoot中，我們可以理解成，一個Controller介面對應多個ServiceImpl，使用這種方式，如果後期需要新增一個功能，僅僅建立一個ServiceImpl就可以滿足需求，而不用再額外建立一個Controller介面。

現在假設一個情景，前端傳入不同的使用者型別，後端返回該使用者的任務。

你可能問我，為什麼不直接把（使用者型別，使用者任務）存入資料庫？

現在只是一個簡單的場景而已，實際中更為複雜，無法直接存入資料庫。

程式碼演示

我們先定義一個介面

public interface UserService { 
    //返回使用者的主要任務
    String task();
}

兩個實現類

@Service("student")
public class StudentServiceImpl implements UserService {
    @Override
    public String task() {
        return "學習";
    }
}

@Service("teacher")
public class TeacherServiceImpl implements UserService {
    @Override
    public String task() {
        return "教書";
    }
}

實現動態呼叫的核心類

@Service
public class UserContext { 
    @Autowired
    Map<String, UserService> userMap;
 
    public UserService getUserService(String type) {
        return userMap.get(type);
    }
}

Spring會自動地將形如(@Service後面的名稱，實現該介面的類)注入到該userMap中

在啟動後，userMap中就存在兩個元素，("student"，StudentServiceImpl)與("teacher"，TeacherServiceImpl)

getUserService方法返回userMap中key=type的UserService物件

實體類

public class User { 
    private String type; 
    private String task; 
    public String getType() {
        return type;
    }
 
    public void setType(String type) {
        this.type = type;
    }
 
    public String getTask() {
        return task;
    }
 
    public void setTask(String task) {
        this.task = task;
    }
}

Controller層介面

@RestController
@RequestMapping("/user")
public class UserController { 
    @Autowired
    UserContext userContext;
 
    @PostMapping("/getTask")
    public String getTask(@RequestBody User user) {
        UserService userService = userContext.getUserService(user.getType());
        return userService.task();
    }
}

測試樣例：

可能用到的場景舉例

關於庫存的儀表盤統計

前端傳入區域id，倉庫id，物品id等資訊

後端依據引數動態地選擇某一個物品實現類，最後返回統計的資訊。

這裡有幾個問題，為什麼不一次性將所有物品id傳入，一次性獲取所有物品的庫存？

一次性傳入，可能後端處理時間邊長，失敗率也高，一旦失敗，整個儀表盤沒有任何資料。而且後期可能面臨的一個需求，不同的物品，需要有不同的介面重新整理速度，暢銷的物品介面呼叫頻率快。所以可能需要將物品分組，一個小組是同一種型別，使用一個實現類。

比如，這裡有100種物品，按型別或者其他屬性分成了10組，每個組之間，有一個不同的屬性groupId，但10組共用一個介面，進入介面後，再進入10個不同的實現類，在實現類中呼叫具體的計算邏輯。

spring中動態選擇實現類

在spring中當一個介面有多個實現類的時候，通過建立簡單工廠類，根據傳入的不同的引數獲取不同的介面實現類。

public interface ExecuteService {    
    ExecuteEnum getCode();
    // 業務方法
    void execute();
}

@Service
public class FirstExecuteServiceImpl implements ExecuteService {    
    @Override
    public ExecuteEnum getCode() {
        return ExecuteEnum.FIRST;
    }
    
    public void execute() {
        System.out.println("11111111111");
    }
}

@Service
public class SecondExecuteServiceImpl implements ExecuteService {    
    @Override
    public ExecuteEnum getCode() {
        return ExecuteEnum.SECOND;
    }
    
    public void execute() {
        System.out.println("222222222");
    }
}

    public enum ExecuteEnum {
        FIRST,
        SECOND,;
    }

方案一

@Component
public class ExecuteServiceFactory implements ApplicationContextAware {
    
    private final static Map<ExecuteEnum, ExecuteService> EXECUTE_SERVICES = new HashMap<>();
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        Map<String, ExecuteService> types = applicationContext.getBeansOfType(ExecuteService.class);
        types.values().forEach(e -> EXECUTE_SERVICES.putIfAbsent(e.getCode(), e));
    }    
}

方案二

@Component
public class ExecuteServiceFactory implements InitializingBean {
    @Autowired
    private List<ExecuteService> executeServices;
    public final static Map<ExecuteEnum, ExecuteService> EXECUTE_SERVICES = new HashMap<>();
    @Override
    public void afterPropertiesSet() throws Exception {
        executeServices.forEach(l -> EXECUTE_SERVICES.putIfAbsent(l.getCode(), l));
    }
}

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。

springboot如何通過不同的策略動態呼叫不同的實現類

目錄

通過不同的策略動態呼叫不同的實現類

程式碼演示

可能用到的場景舉例

spring中動態選擇實現類

方案一

方案二

熱門文章