예제 #1
0
 /**
  * 将常驻作业放入待执行队列.
  *
  * @param jobName 作业名称
  */
 public void addDaemon(final String jobName) {
   if (regCenter.getNumChildren(ReadyNode.ROOT)
       > env.getFrameworkConfiguration().getJobStateQueueSize()) {
     log.warn(
         "Cannot add daemon job, caused by read state queue size is larger than {}.",
         env.getFrameworkConfiguration().getJobStateQueueSize());
     return;
   }
   Optional<CloudJobConfiguration> cloudJobConfig = configService.load(jobName);
   if (!cloudJobConfig.isPresent()
       || JobExecutionType.DAEMON != cloudJobConfig.get().getJobExecutionType()) {
     return;
   }
   regCenter.persist(ReadyNode.getReadyJobNodePath(jobName), "1");
 }
예제 #2
0
 /**
  * 将瞬时作业放入待执行队列.
  *
  * @param jobName 作业名称
  */
 public void addTransient(final String jobName) {
   if (regCenter.getNumChildren(ReadyNode.ROOT)
       > env.getFrameworkConfiguration().getJobStateQueueSize()) {
     log.warn(
         "Cannot add transient job, caused by read state queue size is larger than {}.",
         env.getFrameworkConfiguration().getJobStateQueueSize());
     return;
   }
   Optional<CloudJobConfiguration> cloudJobConfig = configService.load(jobName);
   if (!cloudJobConfig.isPresent()
       || JobExecutionType.TRANSIENT != cloudJobConfig.get().getJobExecutionType()) {
     return;
   }
   String readyJobNode = ReadyNode.getReadyJobNodePath(jobName);
   String times = regCenter.getDirectly(readyJobNode);
   if (cloudJobConfig.get().getTypeConfig().getCoreConfig().isMisfire()) {
     regCenter.persist(
         readyJobNode, Integer.toString(null == times ? 1 : Integer.parseInt(times) + 1));
   } else {
     regCenter.persist(ReadyNode.getReadyJobNodePath(jobName), "1");
   }
 }
예제 #3
0
/**
 * 待运行作业队列服务.
 *
 * @author zhangliang
 */
@Slf4j
public class ReadyService {

  private final BootstrapEnvironment env = BootstrapEnvironment.getInstance();

  private final CoordinatorRegistryCenter regCenter;

  private final ConfigurationService configService;

  private final RunningService runningService;

  public ReadyService(final CoordinatorRegistryCenter regCenter) {
    this.regCenter = regCenter;
    configService = new ConfigurationService(regCenter);
    runningService = new RunningService();
  }

  /**
   * 将瞬时作业放入待执行队列.
   *
   * @param jobName 作业名称
   */
  public void addTransient(final String jobName) {
    if (regCenter.getNumChildren(ReadyNode.ROOT)
        > env.getFrameworkConfiguration().getJobStateQueueSize()) {
      log.warn(
          "Cannot add transient job, caused by read state queue size is larger than {}.",
          env.getFrameworkConfiguration().getJobStateQueueSize());
      return;
    }
    Optional<CloudJobConfiguration> cloudJobConfig = configService.load(jobName);
    if (!cloudJobConfig.isPresent()
        || JobExecutionType.TRANSIENT != cloudJobConfig.get().getJobExecutionType()) {
      return;
    }
    String readyJobNode = ReadyNode.getReadyJobNodePath(jobName);
    String times = regCenter.getDirectly(readyJobNode);
    if (cloudJobConfig.get().getTypeConfig().getCoreConfig().isMisfire()) {
      regCenter.persist(
          readyJobNode, Integer.toString(null == times ? 1 : Integer.parseInt(times) + 1));
    } else {
      regCenter.persist(ReadyNode.getReadyJobNodePath(jobName), "1");
    }
  }

  /**
   * 将常驻作业放入待执行队列.
   *
   * @param jobName 作业名称
   */
  public void addDaemon(final String jobName) {
    if (regCenter.getNumChildren(ReadyNode.ROOT)
        > env.getFrameworkConfiguration().getJobStateQueueSize()) {
      log.warn(
          "Cannot add daemon job, caused by read state queue size is larger than {}.",
          env.getFrameworkConfiguration().getJobStateQueueSize());
      return;
    }
    Optional<CloudJobConfiguration> cloudJobConfig = configService.load(jobName);
    if (!cloudJobConfig.isPresent()
        || JobExecutionType.DAEMON != cloudJobConfig.get().getJobExecutionType()) {
      return;
    }
    regCenter.persist(ReadyNode.getReadyJobNodePath(jobName), "1");
  }

  /**
   * 设置禁用错过重执行.
   *
   * @param jobName 作业名称
   */
  public void setMisfireDisabled(final String jobName) {
    Optional<CloudJobConfiguration> cloudJobConfig = configService.load(jobName);
    if (cloudJobConfig.isPresent()
        && null != regCenter.getDirectly(ReadyNode.getReadyJobNodePath(jobName))) {
      regCenter.persist(ReadyNode.getReadyJobNodePath(jobName), "1");
    }
  }

  /**
   * 从待执行队列中获取所有有资格执行的作业上下文.
   *
   * @param ineligibleJobContexts 无资格执行的作业上下文
   * @return 有资格执行的作业上下文集合
   */
  public Collection<JobContext> getAllEligibleJobContexts(
      final Collection<JobContext> ineligibleJobContexts) {
    if (!regCenter.isExisted(ReadyNode.ROOT)) {
      return Collections.emptyList();
    }
    Collection<String> ineligibleJobNames =
        Collections2.transform(
            ineligibleJobContexts,
            new Function<JobContext, String>() {

              @Override
              public String apply(final JobContext input) {
                return input.getJobConfig().getJobName();
              }
            });
    List<String> jobNames = regCenter.getChildrenKeys(ReadyNode.ROOT);
    List<JobContext> result = new ArrayList<>(jobNames.size());
    for (String each : jobNames) {
      if (ineligibleJobNames.contains(each)) {
        continue;
      }
      Optional<CloudJobConfiguration> jobConfig = configService.load(each);
      if (!jobConfig.isPresent()) {
        regCenter.remove(ReadyNode.getReadyJobNodePath(each));
        continue;
      }
      if (!runningService.isJobRunning(each)
          || JobExecutionType.DAEMON == jobConfig.get().getJobExecutionType()) {
        result.add(JobContext.from(jobConfig.get(), ExecutionType.READY));
      }
    }
    return result;
  }

  /**
   * 从待执行队列中删除相关作业.
   *
   * @param jobNames 待删除的作业名集合
   */
  public void remove(final Collection<String> jobNames) {
    for (String each : jobNames) {
      String readyJobNode = ReadyNode.getReadyJobNodePath(each);
      String timesStr = regCenter.getDirectly(readyJobNode);
      int times = null == timesStr ? 0 : Integer.parseInt(timesStr);
      if (times <= 1) {
        regCenter.remove(readyJobNode);
      } else {
        regCenter.persist(readyJobNode, Integer.toString(times - 1));
      }
    }
  }
}