View Javadoc

1   /*
2    * Copyright 2012-2013 the original author or authors.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  package org.springframework.batch.core.configuration.annotation;
17  
18  import org.springframework.context.annotation.ImportSelector;
19  import org.springframework.core.annotation.AnnotationAttributes;
20  import org.springframework.core.type.AnnotationMetadata;
21  import org.springframework.util.Assert;
22  
23  /**
24   * Base {@code Configuration} class providing common structure for enabling and using Spring Batch. Customization is
25   * available by implementing the {@link BatchConfigurer} interface.
26   * 
27   * @author Dave Syer
28   * @since 2.2
29   * @see EnableBatchProcessing
30   */
31  public class BatchConfigurationSelector implements ImportSelector {
32  
33  	@Override
34  	public String[] selectImports(AnnotationMetadata importingClassMetadata) {
35  		Class<?> annotationType = EnableBatchProcessing.class;
36  		AnnotationAttributes attributes = AnnotationAttributes.fromMap(importingClassMetadata.getAnnotationAttributes(
37  				annotationType.getName(), false));
38  		Assert.notNull(attributes, String.format("@%s is not present on importing class '%s' as expected",
39  				annotationType.getSimpleName(), importingClassMetadata.getClassName()));
40  
41  		String[] imports;
42  		if (attributes.containsKey("modular") && attributes.getBoolean("modular")) {
43  			imports = new String[] { ModularBatchConfiguration.class.getName() };
44  		}
45  		else {
46  			imports = new String[] { SimpleBatchConfiguration.class.getName() };
47  		}
48  
49  		return imports;
50  	}
51  
52  }