转载 Spring特殊注入功能(注入Map集合实现策略模式)

Spring提供通过@Resource注解将相同类型的对象注入到Map集合，并将对象的名字作为key，对象作为value封装进入Map，下面我们来具体实现一下：

首先我们定义一个抽象类

public abstract class TaskAbstractHandler {
 
    abstract public boolean handleJob(String message);
}
定义多个对象分别继承上面的抽象类

@Slf4j
@Component("taskA")
public class TaskAHandler extends TaskAbstractHandler {
    @Override
    public boolean handleJob(String message) {
        // TODO 实现taskA具体的业务逻辑
    }
}
@Slf4j
@Component("taskB")
public class TaskBHandler extends TaskAbstractHandler {
    @Override
    public boolean handleJob(String message) {
        // TODO 实现taskB具体的业务逻辑
    }
}
注入Map对象

@Slf4j
@Component
public class ThirdMQListener implements MessageListener {
 
    @Resource
    private Map<String, TaskAbstractHandler> taskHandlerMap;
 
    @Override
    public Action consume(Message message, ConsumeContext consumeContext) {
 
        // 获取消息体
        byte[] body = message.getBody();
        String messageBody = new String(body);
 
        JSONObject json = JSON.parseObject(messageBody);
        // 获取任务编号
        String taskCode = json.getString("taskCode");
 
        // 根据tag获取具体调用方
        TaskAbstractHandler taskHandler = taskHandlerMap.get(taskCode);
 
        if (taskHandler == null) {
            log.error("No object found according to the task code[{}]", taskCode);
            return Action.ReconsumeLater;
        }
 
        boolean isSuccess = taskHandler.handleJob(messageBody);
        if (isSuccess) {
            return Action.CommitMessage;
        } else {
            return Action.ReconsumeLater;
        }
    }
}
上面通过@Resource注解将TaskAbstractHandler类型的对象注入到Map集合中，再根据消息体中的任务编号从taskHandlerMap对象或获取到具体的执行任务对象，从而根据任务编号执行不同的策略。
————————————————
版权声明：本文为CSDN博主「chaojunma」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/chaojunma/article/details/122595678