Spring Cloud Task 集成Spring Cloud Task Batch（一）

原创

堕落飞鸟

发布于 2023-04-17 02:39:02

1.1K00

代码可运行

文章被收录于专栏：飞鸟的专栏飞鸟的专栏

运行总次数：0

代码可运行

介绍

Spring Cloud Task和Spring Batch都是Spring生态系统中强大的工具。Spring Batch提供了一个框架，用于编写和执行大规模批处理作业，而Spring Cloud Task提供了一种机制，可以将短期的任务作为单独的执行单元来运行。这两个工具在不同的场景下都非常有用，因此将它们结合起来可以提供更广泛的应用程序开发和部署选择。

添加Spring Batch依赖项

在构建文件中，我们需要添加Spring Batch依赖项：

<dependency>
    <groupId>org.springframework.batch</groupId>
    <artifactId>spring-batch-core</artifactId>
    <version>${spring-batch.version}</version>
</dependency>

创建Spring Batch作业

我们将创建一个简单的Spring Batch作业，该作业将读取一个文件，并将文件中的每一行打印到控制台上。首先，我们需要定义一个JobBuilderFactory，并在其中创建一个新的Job：

import org.springframework.batch.core.Job;
import org.springframework.batch.core.JobParametersBuilder;
import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.annotation.EnableBatchProcessing;
import org.springframework.batch.core.configuration.annotation.JobBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.item.file.FlatFileItemReader;
import org.springframework.batch.item.file.mapping.PassThroughLineMapper;
import org.springframework.batch.item.file.transform.DefaultFieldSet;
import org.springframework.batch.item.file.transform.FieldSet;
import org.springframework.batch.item.file.transform.LineTokenizer;
import org.springframework.batch.item.file.transform.PassThroughLineTokenizer;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.Resource;

@Configuration
@EnableBatchProcessing
public class BatchConfiguration {

    private final JobBuilderFactory jobBuilderFactory;

    private final StepBuilderFactory stepBuilderFactory;

    @Value("file:${input.file}")
    private Resource inputFile;

    public BatchConfiguration(JobBuilderFactory jobBuilderFactory,
                               StepBuilderFactory stepBuilderFactory) {
        this.jobBuilderFactory = jobBuilderFactory;
        this.stepBuilderFactory = stepBuilderFactory;
    }

    @Bean
    public FlatFileItemReader<FieldSet> reader() {
        FlatFileItemReader<FieldSet> reader = new FlatFileItemReader<>();
        reader.setResource(inputFile);
        LineTokenizer tokenizer = new PassThroughLineTokenizer();
        reader.setLineMapper(new PassThroughLineMapper() {
            @Override
            public FieldSet mapLine(String line, int lineNumber) throws Exception {
                return new DefaultFieldSet(line.split(","), tokenizer.tokenize(line));
            }
        });
        return reader;
    }

    @Bean
    public SimpleItemWriter writer() {
        return new SimpleItemWriter();
    }

    @Bean
    public Step step() {
        return stepBuilderFactory.get("step")
                .<FieldSet, String>chunk(10)
                .reader(reader())
                .writer(writer())
                .build();
    }

    @Bean
    public Job job() {
        return jobBuilderFactory.get("job")
                .start(step())
                .build();
    }

}