Scheduled task introspection through ScheduledTaskHolder interface

Issue: SPR-15982

Author: jhoeller

spring-context/src/main/java/org/springframework/scheduling/annotation/ScheduledAnnotationBeanPostProcessor.java

@@ -60,8 +60,10 @@
 import org.springframework.scheduling.TaskScheduler;
 import org.springframework.scheduling.Trigger;
 import org.springframework.scheduling.config.CronTask;
-import org.springframework.scheduling.config.IntervalTask;
+import org.springframework.scheduling.config.FixedDelayTask;
+import org.springframework.scheduling.config.FixedRateTask;
 import org.springframework.scheduling.config.ScheduledTask;
+import org.springframework.scheduling.config.ScheduledTaskHolder;
 import org.springframework.scheduling.config.ScheduledTaskRegistrar;
 import org.springframework.scheduling.support.CronTrigger;
 import org.springframework.scheduling.support.ScheduledMethodRunnable;
@@ -96,7 +98,7 @@
  * @see AsyncAnnotationBeanPostProcessor
  */
 public class ScheduledAnnotationBeanPostProcessor
-		implements MergedBeanDefinitionPostProcessor, DestructionAwareBeanPostProcessor,
+		implements ScheduledTaskHolder, MergedBeanDefinitionPostProcessor, DestructionAwareBeanPostProcessor,
 		Ordered, EmbeddedValueResolverAware, BeanNameAware, BeanFactoryAware, ApplicationContextAware,
 		SmartInitializingSingleton, ApplicationListener<ContextRefreshedEvent>, DisposableBean {
 
@@ -400,7 +402,7 @@ protected void processScheduled(Scheduled scheduled, Method method, Object bean)
 			if (fixedDelay >= 0) {
 				Assert.isTrue(!processedSchedule, errorMessage);
 				processedSchedule = true;
-				tasks.add(this.registrar.scheduleFixedDelayTask(new IntervalTask(runnable, fixedDelay, initialDelay)));
+				tasks.add(this.registrar.scheduleFixedDelayTask(new FixedDelayTask(runnable, fixedDelay, initialDelay)));
 			}
 			String fixedDelayString = scheduled.fixedDelayString();
 			if (StringUtils.hasText(fixedDelayString)) {
@@ -417,7 +419,7 @@ protected void processScheduled(Scheduled scheduled, Method method, Object bean)
 						throw new IllegalArgumentException(
 								"Invalid fixedDelayString value \"" + fixedDelayString + "\" - cannot parse into long");
 					}
-					tasks.add(this.registrar.scheduleFixedDelayTask(new IntervalTask(runnable, fixedDelay, initialDelay)));
+					tasks.add(this.registrar.scheduleFixedDelayTask(new FixedDelayTask(runnable, fixedDelay, initialDelay)));
 				}
 			}
 
@@ -426,7 +428,7 @@ protected void processScheduled(Scheduled scheduled, Method method, Object bean)
 			if (fixedRate >= 0) {
 				Assert.isTrue(!processedSchedule, errorMessage);
 				processedSchedule = true;
-				tasks.add(this.registrar.scheduleFixedRateTask(new IntervalTask(runnable, fixedRate, initialDelay)));
+				tasks.add(this.registrar.scheduleFixedRateTask(new FixedRateTask(runnable, fixedRate, initialDelay)));
 			}
 			String fixedRateString = scheduled.fixedRateString();
 			if (StringUtils.hasText(fixedRateString)) {
@@ -443,7 +445,7 @@ protected void processScheduled(Scheduled scheduled, Method method, Object bean)
 						throw new IllegalArgumentException(
 								"Invalid fixedRateString value \"" + fixedRateString + "\" - cannot parse into integer");
 					}
-					tasks.add(this.registrar.scheduleFixedRateTask(new IntervalTask(runnable, fixedRate, initialDelay)));
+					tasks.add(this.registrar.scheduleFixedRateTask(new FixedRateTask(runnable, fixedRate, initialDelay)));
 				}
 			}
 
@@ -467,6 +469,24 @@ protected void processScheduled(Scheduled scheduled, Method method, Object bean)
 	}
 
 
+	/**
+	 * Return all currently scheduled tasks, from {@link Scheduled} methods
+	 * as well as from programmatic {@link SchedulingConfigurer} interaction.
+	 * @since 5.0.2
+	 */
+	@Override
+	public Set<ScheduledTask> getScheduledTasks() {
+		Set<ScheduledTask> result = new LinkedHashSet<>();
+		synchronized (this.scheduledTasks) {
+			Collection<Set<ScheduledTask>> allTasks = this.scheduledTasks.values();
+			for (Set<ScheduledTask> tasks : allTasks) {
+				result.addAll(tasks);
+			}
+		}
+		result.addAll(this.registrar.getScheduledTasks());
+		return result;
+	}
+
 	@Override
 	public void postProcessBeforeDestruction(Object bean, String beanName) {
 		Set<ScheduledTask> tasks;

spring-context/src/main/java/org/springframework/scheduling/config/CronTask.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2013 the original author or authors.
+ * Copyright 2002-2017 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -26,8 +26,7 @@
  * @author Chris Beams
  * @since 3.2
  * @see org.springframework.scheduling.annotation.Scheduled#cron()
- * @see ScheduledTaskRegistrar#setCronTasksList(java.util.List)
- * @see org.springframework.scheduling.TaskScheduler
+ * @see ScheduledTaskRegistrar#addCronTask(CronTask)
  */
 public class CronTask extends TriggerTask {
 
@@ -37,7 +36,7 @@ public class CronTask extends TriggerTask {
 	/**
 	 * Create a new {@code CronTask}.
 	 * @param runnable the underlying task to execute
-	 * @param expression cron expression defining when the task should be executed
+	 * @param expression the cron expression defining when the task should be executed
 	 */
 	public CronTask(Runnable runnable, String expression) {
 		this(runnable, new CronTrigger(expression));
@@ -54,6 +53,9 @@ public CronTask(Runnable runnable, CronTrigger cronTrigger) {
 	}
 
 
+	/**
+	 * Return the cron expression defining when the task should be executed.
+	 */
 	public String getExpression() {
 		return this.expression;
 	}

spring-context/src/main/java/org/springframework/scheduling/config/FixedDelayTask.java

@@ -0,0 +1,39 @@
+/*
+ * Copyright 2002-2017 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.scheduling.config;
+
+/**
+ * Specialization of {@link IntervalTask} for fixed-delay semantics.
+ *
+ * @author Juergen Hoeller
+ * @since 5.0.2
+ * @see org.springframework.scheduling.annotation.Scheduled#fixedDelay()
+ * @see ScheduledTaskRegistrar#addFixedDelayTask(IntervalTask)
+ */
+public class FixedDelayTask extends IntervalTask {
+
+	/**
+	 * Create a new {@code FixedDelayTask}.
+	 * @param runnable the underlying task to execute
+	 * @param interval how often in milliseconds the task should be executed
+	 * @param initialDelay the initial delay before first execution of the task
+	 */
+	public FixedDelayTask(Runnable runnable, long interval, long initialDelay) {
+		super(runnable, interval, initialDelay);
+	}
+
+}

spring-context/src/main/java/org/springframework/scheduling/config/FixedRateTask.java

@@ -0,0 +1,39 @@
+/*
+ * Copyright 2002-2017 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.scheduling.config;
+
+/**
+ * Specialization of {@link IntervalTask} for fixed-rate semantics.
+ *
+ * @author Juergen Hoeller
+ * @since 5.0.2
+ * @see org.springframework.scheduling.annotation.Scheduled#fixedRate()
+ * @see ScheduledTaskRegistrar#addFixedRateTask(IntervalTask)
+ */
+public class FixedRateTask extends IntervalTask {
+
+	/**
+	 * Create a new {@code FixedRateTask}.
+	 * @param runnable the underlying task to execute
+	 * @param interval how often in milliseconds the task should be executed
+	 * @param initialDelay the initial delay before first execution of the task
+	 */
+	public FixedRateTask(Runnable runnable, long interval, long initialDelay) {
+		super(runnable, interval, initialDelay);
+	}
+
+}

spring-context/src/main/java/org/springframework/scheduling/config/IntervalTask.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2013 the original author or authors.
+ * Copyright 2002-2017 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -23,11 +23,8 @@
  *
  * @author Chris Beams
  * @since 3.2
- * @see org.springframework.scheduling.annotation.Scheduled#fixedRate()
- * @see org.springframework.scheduling.annotation.Scheduled#fixedDelay()
- * @see ScheduledTaskRegistrar#setFixedRateTasksList(java.util.List)
- * @see ScheduledTaskRegistrar#setFixedDelayTasksList(java.util.List)
- * @see org.springframework.scheduling.TaskScheduler
+ * @see ScheduledTaskRegistrar#addFixedRateTask(IntervalTask)
+ * @see ScheduledTaskRegistrar#addFixedDelayTask(IntervalTask)
  */
 public class IntervalTask extends Task {
 
@@ -40,7 +37,7 @@ public class IntervalTask extends Task {
 	 * Create a new {@code IntervalTask}.
 	 * @param runnable the underlying task to execute
 	 * @param interval how often in milliseconds the task should be executed
-	 * @param initialDelay initial delay before first execution of the task
+	 * @param initialDelay the initial delay before first execution of the task
 	 */
 	public IntervalTask(Runnable runnable, long interval, long initialDelay) {
 		super(runnable);
@@ -58,10 +55,16 @@ public IntervalTask(Runnable runnable, long interval) {
 	}
 
 
+	/**
+	 * Return how often in milliseconds the task should be executed.
+	 */
 	public long getInterval() {
 		return this.interval;
 	}
 
+	/**
+	 * Return the initial delay before first execution of the task.
+	 */
 	public long getInitialDelay() {
 		return this.initialDelay;
 	}

spring-context/src/main/java/org/springframework/scheduling/config/ScheduledTask.java

@@ -21,24 +21,37 @@
 import org.springframework.lang.Nullable;
 
 /**
- * A representation of a scheduled task,
+ * A representation of a scheduled task at runtime,
  * used as a return value for scheduling methods.
  *
  * @author Juergen Hoeller
  * @since 4.3
- * @see ScheduledTaskRegistrar#scheduleTriggerTask
- * @see ScheduledTaskRegistrar#scheduleFixedRateTask
+ * @see ScheduledTaskRegistrar#scheduleCronTask(CronTask)
+ * @see ScheduledTaskRegistrar#scheduleFixedRateTask(FixedRateTask)
+ * @see ScheduledTaskRegistrar#scheduleFixedDelayTask(FixedDelayTask)
  */
 public final class ScheduledTask {
 
+	private final Task task;
+
 	@Nullable
 	volatile ScheduledFuture<?> future;
 
 
-	ScheduledTask() {
+	ScheduledTask(Task task) {
+		this.task = task;
 	}
 
 
+	/**
+	 * Return the underlying task (typically a {@link CronTask},
+	 * {@link FixedRateTask} or {@link FixedDelayTask}).
+	 * @since 5.0.2
+	 */
+	public Task getTask() {
+		return this.task;
+	}
+
 	/**
 	 * Trigger cancellation of this scheduled task.
 	 */
@@ -49,4 +62,9 @@ public void cancel() {
 		}
 	}
 
+	@Override
+	public String toString() {
+		return this.task.toString();
+	}
+
 }

spring-context/src/main/java/org/springframework/scheduling/config/ScheduledTaskHolder.java

@@ -0,0 +1,36 @@
+/*
+ * Copyright 2002-2017 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.scheduling.config;
+
+import java.util.Set;
+
+/**
+ * Common interface for exposing locally scheduled tasks.
+ *
+ * @author Juergen Hoeller
+ * @since 5.0.2
+ * @see ScheduledTaskRegistrar
+ * @see org.springframework.scheduling.annotation.ScheduledAnnotationBeanPostProcessor
+ */
+public interface ScheduledTaskHolder {
+
+	/**
+	 * Return an overview of the tasks that have been scheduled by this instance.
+	 */
+	Set<ScheduledTask> getScheduledTasks();
+
+}