springboot如何通过不同的策略动态调用不同的实现类

通过不同的策略动态调用不同的实现类

经常遇到这样的一个需求，前端传的实体类型相同，后端需要根据实体类中的某一个字符串，动态地调用某一个类的方法。

在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));
    }
}

以上为个人经验，希望能给大家一个参考，也希望大家多多支持猪先飞。