Spring注入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具体的业务逻辑
    }
}

@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;
        }
    }
}