Polish "Add Quartz Scheduler support"

Closes gh-4299

Author: snicoll

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/quartz/JobStoreType.java

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2012-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.boot.autoconfigure.quartz;
+
+/**
+ * Define the supported Quartz {@code JobStore}.
+ *
+ * @author Stephane Nicoll
+ * @since 2.0.0
+ */
+public enum JobStoreType {
+
+	/**
+	 * Store jobs in memory.
+	 */
+	MEMORY,
+
+	/**
+	 * Store jobs in the database.
+	 */
+	JDBC
+
+}

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/quartz/QuartzAutoConfiguration.java

@@ -28,21 +28,18 @@
 import org.quartz.Scheduler;
 import org.quartz.Trigger;
 
-import org.springframework.beans.BeansException;
 import org.springframework.beans.factory.ObjectProvider;
 import org.springframework.boot.autoconfigure.AutoConfigureAfter;
 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
-import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnSingleCandidate;
 import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
 import org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaAutoConfiguration;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.context.ApplicationContext;
-import org.springframework.context.ApplicationContextAware;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.core.annotation.AnnotationAwareOrderComparator;
 import org.springframework.core.io.ResourceLoader;
 import org.springframework.scheduling.quartz.SchedulerFactoryBean;
 import org.springframework.transaction.PlatformTransactionManager;
@@ -51,6 +48,7 @@
  * {@link EnableAutoConfiguration Auto-configuration} for Quartz Scheduler.
  *
  * @author Vedran Pavic
+ * @author Stephane Nicoll
  * @since 2.0.0
  */
 @Configuration
@@ -59,7 +57,7 @@
 @EnableConfigurationProperties(QuartzProperties.class)
 @AutoConfigureAfter({ DataSourceAutoConfiguration.class,
 		HibernateJpaAutoConfiguration.class })
-public class QuartzAutoConfiguration implements ApplicationContextAware {
+public class QuartzAutoConfiguration {
 
 	private final QuartzProperties properties;
 
@@ -73,23 +71,25 @@ public class QuartzAutoConfiguration implements ApplicationContextAware {
 
 	private final Trigger[] triggers;
 
-	private ApplicationContext applicationContext;
+	private final ApplicationContext applicationContext;
 
 	public QuartzAutoConfiguration(QuartzProperties properties,
 			ObjectProvider<List<SchedulerFactoryBeanCustomizer>> customizers,
 			ObjectProvider<Executor> taskExecutor, ObjectProvider<JobDetail[]> jobDetails,
 			ObjectProvider<Map<String, Calendar>> calendars,
-			ObjectProvider<Trigger[]> triggers) {
+			ObjectProvider<Trigger[]> triggers,
+			ApplicationContext applicationContext) {
 		this.properties = properties;
 		this.customizers = customizers.getIfAvailable();
 		this.taskExecutor = taskExecutor.getIfAvailable();
 		this.jobDetails = jobDetails.getIfAvailable();
 		this.calendars = calendars.getIfAvailable();
 		this.triggers = triggers.getIfAvailable();
+		this.applicationContext = applicationContext;
 	}
 
 	@Bean
-	@ConditionalOnBean(DataSource.class)
+	@ConditionalOnSingleCandidate(DataSource.class)
 	@ConditionalOnMissingBean
 	public QuartzDatabaseInitializer quartzDatabaseInitializer(DataSource dataSource,
 			ResourceLoader resourceLoader) {
@@ -98,7 +98,7 @@ public QuartzDatabaseInitializer quartzDatabaseInitializer(DataSource dataSource
 
 	@Bean
 	@ConditionalOnMissingBean
-	public SchedulerFactoryBean schedulerFactoryBean() {
+	public SchedulerFactoryBean quartzScheduler() {
 		SchedulerFactoryBean schedulerFactoryBean = new SchedulerFactoryBean();
 		schedulerFactoryBean.setJobFactory(new AutowireCapableBeanJobFactory(
 				this.applicationContext.getAutowireCapableBeanFactory()));
@@ -122,12 +122,6 @@ public SchedulerFactoryBean schedulerFactoryBean() {
 		return schedulerFactoryBean;
 	}
 
-	@Override
-	public void setApplicationContext(ApplicationContext applicationContext)
-			throws BeansException {
-		this.applicationContext = applicationContext;
-	}
-
 	private Properties asProperties(Map<String, String> source) {
 		Properties properties = new Properties();
 		properties.putAll(source);
@@ -136,23 +130,29 @@ private Properties asProperties(Map<String, String> source) {
 
 	private void customize(SchedulerFactoryBean schedulerFactoryBean) {
 		if (this.customizers != null) {
-			AnnotationAwareOrderComparator.sort(this.customizers);
 			for (SchedulerFactoryBeanCustomizer customizer : this.customizers) {
 				customizer.customize(schedulerFactoryBean);
 			}
 		}
 	}
 
 	@Configuration
-	@ConditionalOnBean(DataSource.class)
-	protected static class QuartzSchedulerDataSourceConfiguration {
+	@ConditionalOnSingleCandidate(DataSource.class)
+	protected static class JdbcStoreTypeConfiguration {
 
 		@Bean
-		public SchedulerFactoryBeanCustomizer dataSourceCustomizer(DataSource dataSource,
-				PlatformTransactionManager transactionManager) {
+		public SchedulerFactoryBeanCustomizer dataSourceCustomizer(
+				QuartzProperties properties, DataSource dataSource,
+				ObjectProvider<PlatformTransactionManager> transactionManager) {
 			return schedulerFactoryBean -> {
-				schedulerFactoryBean.setDataSource(dataSource);
-				schedulerFactoryBean.setTransactionManager(transactionManager);
+				if (properties.getJobStoreType() == JobStoreType.JDBC) {
+					schedulerFactoryBean.setDataSource(dataSource);
+					PlatformTransactionManager txManager =
+							transactionManager.getIfUnique();
+					if (txManager != null) {
+						schedulerFactoryBean.setTransactionManager(txManager);
+					}
+				}
 			};
 		}
 

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/quartz/QuartzDatabaseInitializer.java

@@ -41,12 +41,12 @@ public QuartzDatabaseInitializer(DataSource dataSource, ResourceLoader resourceL
 
 	@Override
 	protected boolean isEnabled() {
-		return this.properties.getInitializer().isEnabled();
+		return this.properties.getJdbc().isInitializeSchema();
 	}
 
 	@Override
 	protected String getSchemaLocation() {
-		return this.properties.getSchema();
+		return this.properties.getJdbc().getSchema();
 	}
 
 	@Override

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/quartz/QuartzProperties.java

@@ -25,60 +25,69 @@
  * Configuration properties for the Quartz Scheduler integration.
  *
  * @author Vedran Pavic
+ * @author Stephane Nicoll
  * @since 2.0.0
  */
 @ConfigurationProperties("spring.quartz")
 public class QuartzProperties {
 
-	private static final String DEFAULT_SCHEMA_LOCATION = "classpath:org/quartz/impl/"
-			+ "jdbcjobstore/tables_@@platform@@.sql";
-
-	private final Initializer initializer = new Initializer();
-
 	/**
-	 * Additional Quartz Scheduler properties.
+	 * Quartz job store type.
 	 */
-	private Map<String, String> properties = new HashMap<>();
+	private JobStoreType jobStoreType = JobStoreType.MEMORY;
 
 	/**
-	 * Path to the SQL file to use to initialize the database schema.
+	 * Additional Quartz Scheduler properties.
 	 */
-	private String schema = DEFAULT_SCHEMA_LOCATION;
+	private final Map<String, String> properties = new HashMap<>();
 
-	public Initializer getInitializer() {
-		return this.initializer;
+	private final Jdbc jdbc = new Jdbc();
+
+	public JobStoreType getJobStoreType() {
+		return this.jobStoreType;
+	}
+
+	public void setJobStoreType(JobStoreType jobStoreType) {
+		this.jobStoreType = jobStoreType;
 	}
 
 	public Map<String, String> getProperties() {
 		return this.properties;
 	}
 
-	public void setProperties(Map<String, String> properties) {
-		this.properties = properties;
+	public Jdbc getJdbc() {
+		return this.jdbc;
 	}
 
-	public String getSchema() {
-		return this.schema;
-	}
+	public static class Jdbc {
 
-	public void setSchema(String schema) {
-		this.schema = schema;
-	}
+		private static final String DEFAULT_SCHEMA_LOCATION = "classpath:org/quartz/impl/"
+				+ "jdbcjobstore/tables_@@platform@@.sql";
 
-	public class Initializer {
+		/**
+		 * Path to the SQL file to use to initialize the database schema.
+		 */
+		private String schema = DEFAULT_SCHEMA_LOCATION;
 
 		/**
-		 * Create the required Quartz Scheduler tables on startup if necessary. Enabled
-		 * automatically if the schema is configured.
+		 * Create the required Quartz Scheduler tables on startup.
 		 */
-		private boolean enabled = true;
+		private boolean initializeSchema;
+
+		public String getSchema() {
+			return this.schema;
+		}
+
+		public void setSchema(String schema) {
+			this.schema = schema;
+		}
 
-		public boolean isEnabled() {
-			return this.enabled && QuartzProperties.this.getSchema() != null;
+		public boolean isInitializeSchema() {
+			return this.initializeSchema;
 		}
 
-		public void setEnabled(boolean enabled) {
-			this.enabled = enabled;
+		public void setInitializeSchema(boolean initializeSchema) {
+			this.initializeSchema = initializeSchema;
 		}
 
 	}

spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/quartz/SchedulerFactoryBeanCustomizer.java

@@ -28,6 +28,10 @@
  */
 public interface SchedulerFactoryBeanCustomizer {
 
+	/**
+	 * Customize the {@link SchedulerFactoryBean}.
+	 * @param schedulerFactoryBean the scheduler to customize
+	 */
 	void customize(SchedulerFactoryBean schedulerFactoryBean);
 
 }

spring-boot-autoconfigure/src/main/resources/META-INF/additional-spring-configuration-metadata.json

@@ -357,6 +357,10 @@
     "name": "spring.mvc.locale-resolver",
     "defaultValue": "accept-header"
   },
+  {
+    "name": "spring.quartz.job-store-type",
+    "defaultValue": "memory"
+  },
   {
     "name": "spring.rabbitmq.cache.connection.mode",
     "defaultValue": "channel"