| 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.batch.core.repository.JobRepository; |
| 19 | import org.springframework.batch.core.step.builder.StepBuilder; |
| 20 | import org.springframework.transaction.PlatformTransactionManager; |
| 21 | |
| 22 | /** |
| 23 | * Convenient factory for a {@link StepBuilder} which sets the {@link JobRepository} and |
| 24 | * {@link PlatformTransactionManager} automatically. |
| 25 | * |
| 26 | * @author Dave Syer |
| 27 | * |
| 28 | */ |
| 29 | public class StepBuilderFactory { |
| 30 | |
| 31 | private JobRepository jobRepository; |
| 32 | |
| 33 | private PlatformTransactionManager transactionManager; |
| 34 | |
| 35 | public StepBuilderFactory(JobRepository jobRepository, PlatformTransactionManager transactionManager) { |
| 36 | this.jobRepository = jobRepository; |
| 37 | this.transactionManager = transactionManager; |
| 38 | } |
| 39 | |
| 40 | /** |
| 41 | * Creates a step builder and initializes its job repository and transaction manager. Note that if the builder is |
| 42 | * used to create a @Bean definition then the name of the step and the bean name might be different. |
| 43 | * |
| 44 | * @param name the name of the step |
| 45 | * @return a step builder |
| 46 | */ |
| 47 | public StepBuilder get(String name) { |
| 48 | StepBuilder builder = (StepBuilder) new StepBuilder(name).repository(jobRepository).transactionManager( |
| 49 | transactionManager); |
| 50 | return builder; |
| 51 | } |
| 52 | |
| 53 | } |