/**
* 将常驻作业放入待执行队列.
*
* @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 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");
}
}
/**
* 待运行作业队列服务.
*
* @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));
}
}
}
}
