001/* 002 * Copyright 2012-2016 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package org.springframework.boot.cli.compiler.autoconfigure; 018 019import org.codehaus.groovy.ast.ClassNode; 020import org.codehaus.groovy.control.customizers.ImportCustomizer; 021 022import org.springframework.boot.cli.compiler.AstUtils; 023import org.springframework.boot.cli.compiler.CompilerAutoConfiguration; 024import org.springframework.boot.cli.compiler.DependencyCustomizer; 025 026/** 027 * {@link CompilerAutoConfiguration} for Spring Batch. 028 * 029 * @author Dave Syer 030 * @author Phillip Webb 031 */ 032public class SpringBatchCompilerAutoConfiguration extends CompilerAutoConfiguration { 033 034 @Override 035 public boolean matches(ClassNode classNode) { 036 return AstUtils.hasAtLeastOneAnnotation(classNode, "EnableBatchProcessing"); 037 } 038 039 @Override 040 public void applyDependencies(DependencyCustomizer dependencies) { 041 dependencies.ifAnyMissingClasses("org.springframework.batch.core.Job") 042 .add("spring-boot-starter-batch"); 043 dependencies.ifAnyMissingClasses("org.springframework.jdbc.core.JdbcTemplate") 044 .add("spring-jdbc"); 045 } 046 047 @Override 048 public void applyImports(ImportCustomizer imports) { 049 imports.addImports("org.springframework.batch.repeat.RepeatStatus", 050 "org.springframework.batch.core.scope.context.ChunkContext", 051 "org.springframework.batch.core.step.tasklet.Tasklet", 052 "org.springframework.batch.core.configuration.annotation.StepScope", 053 "org.springframework.batch.core.configuration.annotation.JobBuilderFactory", 054 "org.springframework.batch.core.configuration.annotation.StepBuilderFactory", 055 "org.springframework.batch.core.configuration.annotation.EnableBatchProcessing", 056 "org.springframework.batch.core.Step", 057 "org.springframework.batch.core.StepExecution", 058 "org.springframework.batch.core.StepContribution", 059 "org.springframework.batch.core.Job", 060 "org.springframework.batch.core.JobExecution", 061 "org.springframework.batch.core.JobParameter", 062 "org.springframework.batch.core.JobParameters", 063 "org.springframework.batch.core.launch.JobLauncher", 064 "org.springframework.batch.core.converter.JobParametersConverter", 065 "org.springframework.batch.core.converter.DefaultJobParametersConverter"); 066 } 067 068}