快捷導(dǎo)航

使用Spring自定義注解實(shí)現(xiàn)任務(wù)路由的方法

更新時(shí)間：2017年07月31日 11:41:06 作者：wangyan9110

本篇文章主要介紹了使用Spring自定義注解實(shí)現(xiàn)任務(wù)路由的方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

在Spring mvc的開(kāi)發(fā)中，我們可以通過(guò)RequestMapping來(lái)配，當(dāng)前方法用于處理哪一個(gè)URL的請(qǐng)求.同樣我們現(xiàn)在有一個(gè)需求，有一個(gè)任務(wù)調(diào)度器，可以按照不同的任務(wù)類型路由到不同的任務(wù)執(zhí)行器。其本質(zhì)就是通過(guò)外部參數(shù)進(jìn)行一次路由和Spring mvc做的事情類似。簡(jiǎn)單看了Spring mvc的實(shí)現(xiàn)原理之后，決定使用自定義注解的方式來(lái)實(shí)現(xiàn)以上功能。

自定義TaskHandler注解

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Component
public @interface TaskHandler {

  String taskType() default "";
}

以上定義了任務(wù)處理器的注解，其中@Component表示在spring 啟動(dòng)過(guò)程中，會(huì)掃描到并且注入到容器中。taskType表示類型。

任務(wù)處理器定義

public abstract class AbstractTaskHandler {

  /**
   * 任務(wù)執(zhí)行器
   *
   * @param task 任務(wù)
   * @return 執(zhí)行結(jié)果
   */
   public abstract BaseResult execute(Task task);
}

以上定義了一個(gè)任務(wù)執(zhí)行的處理器，其他所有的具體的任務(wù)執(zhí)行器繼承實(shí)現(xiàn)這個(gè)方法。其中Task表示任務(wù)的定義，包括任務(wù)Id，執(zhí)行任務(wù)需要的參數(shù)等。

任務(wù)處理器實(shí)現(xiàn)

接下來(lái)，我們可以實(shí)現(xiàn)一個(gè)具體的任務(wù)處理器。

@TaskHandler(taskType = "UserNameChanged")
public class UserNameChangedSender extends AbstractTaskHandler {
  @Override
  public BaseResult execute(Task task) {
   return new BaseResult();
  }
}

以上我們就實(shí)現(xiàn)一個(gè)用戶名修改通知的任務(wù)處理器，具體的業(yè)務(wù)邏輯這里沒(méi)有實(shí)現(xiàn)。

其中：@TaskHandler(taskType = "UserNameChanged")，這里我們指定這個(gè)Handler用于處理用戶名變更的任務(wù)

任務(wù)處理Handler注冊(cè)

public class TaskHandlerRegister extends ApplicationObjectSupport {

  private final static Map<String, AbstractTaskHandler> TASK_HANDLERS_MAP = new HashMap<>();

  private static final Logger LOGGER = LoggerFactory.getLogger(TaskHandlerRegister.class);

  @Override
  protected void initApplicationContext(ApplicationContext context) throws BeansException {
    super.initApplicationContext(context);
    Map<String, Object> taskBeanMap = context.getBeansWithAnnotation(TaskHandler.class);
    taskBeanMap.keySet().forEach(beanName -> {
      Object bean = taskBeanMap.get(beanName);
      Class clazz = bean.getClass();
      if (bean instanceof AbstractTaskHandler && clazz.getAnnotation(TaskHandler.class) != null) {
        TaskHandler taskHandler = (TaskHandler) clazz.getAnnotation(TaskHandler.class);
        String taskType = taskHandler.taskType();
        if (TASK_HANDLERS_MAP.keySet().contains(taskType)) {
          throw new RuntimeException("TaskType has Exits. TaskType=" + taskType);
        }
        TASK_HANDLERS_MAP.put(taskHandler.taskType(), (AbstractTaskHandler) taskBeanMap.get(beanName));
        LOGGER.info("Task Handler Register. taskType={},beanName={}", taskHandler.taskType(), beanName);
      }
    });
  }

  public static AbstractTaskHandler getTaskHandler(String taskType) {
    return TASK_HANDLERS_MAP.get(taskType);
  }
}

這里繼承了Spring的ApplicationObjectSupport類，具體的注冊(cè)過(guò)程如下

Spring完成bean的初始化
查找spring的容器中，所有帶有TaskHandler注解的bean
校驗(yàn)bean是否為AbstractTaskHandler類型，獲取到taskType
把該bean放到TASK_HANDLERS_MAP容器中，即注冊(cè)完成

任務(wù)執(zhí)行

接下來(lái)我們來(lái)看下任務(wù)執(zhí)行

public class TaskExecutor implements Job {

  private static final String TASK_TYPE = "taskType";

  @Override
  public BaseResult execute(Task task){
    String taskType=task.getTaskType();
    if (TaskHandlerRegister.getTaskHandler(taskType) == null) {
      throw new RuntimeException("can't find taskHandler,taskType=" + taskType);
    }
    AbstractTaskHandler abstractHandler = TaskHandlerRegister.getTaskHandler(taskType);
    return abstractHandler.execute(task);
  }
}

這里發(fā)起任務(wù)執(zhí)行的是一個(gè)Job，具體過(guò)程如下

校驗(yàn)該任務(wù)類型，有沒(méi)有在注冊(cè)中心注冊(cè)相關(guān)Handler
從任務(wù)注冊(cè)中心獲取到對(duì)應(yīng)的處理的Handelr
執(zhí)行該Handelr

以上過(guò)程就完成了，可以實(shí)現(xiàn)基于注解的一個(gè)任務(wù)路由過(guò)程。其實(shí)現(xiàn)思路來(lái)自于Spring mvc的RequestMapping的設(shè)計(jì)思路.希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: