快捷導航

SpringBoot批處理的實現(xiàn)示例

更新時間：2022年03月15日 10:58:50 作者：清雨未盡時

SpringBatch是一個開源的、全面的、輕量級的批處理框架，通過SpringBoot可以實現(xiàn)強大的批處理應(yīng)用程序開發(fā)，本文主要介紹了SpringBoot批處理的實現(xiàn)示例，感興趣的可以了解一下

1.添加依賴

<dependency>
?? ?<groupId>org.springframework.boot</groupId>
?? ?<artifactId>spring-boot-starter-batch</artifactId>
</dependency>
<dependency>
?? ?<groupId>org.springframework.boot</groupId>
?? ?<artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>
<dependency>
?? ?<groupId>com.alibaba</groupId>
?? ?<artifactId>druid-spring-boot-starter</artifactId>
?? ?<version>1.1.9</version>
</dependency>
<dependency>
?? ?<groupId>mysql</groupId>
?? ?<artifactId>mysql-connector-java</artifactId>
</dependency>

2.添加配置信息

server.port=9000
spring.http.encoding.force-response=true
spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.url=jdbc:mysql:///batch
spring.datasource.username  = root
spring.datasource.password  = kangxg198811
#項目啟動時創(chuàng)建數(shù)據(jù)表的SQL腳本
spring.datasource.schema=classpath:/org/springframework/batch/core/schema-mysql.sql
#項目啟動時執(zhí)行建表SQL
spring.batch.initialize-schema=always
#配置后不會自動執(zhí)行
spring.batch.job.enabled=false

3.在項目啟動類上添加@EnableBatchProcessing注解開開啟Spring Batch支持

@SpringBootApplication
@EnableBatchProcessing
public class BatchApplication {
 
    public static void main(String[] args) {
        SpringApplication.run(BatchApplication.class, args);
    }
 
}

4.配置批處理

@Configuration
public class CsvBatchJobConfig {
    @Autowired
    JobBuilderFactory jobBuilderFactory;
 
    @Autowired
    StepBuilderFactory stepBuilderFactory;
 
    @Autowired
    DataSource dataSource;
 
    class CustomLineCallbackHandler implements LineCallbackHandler{
        @Override
        public void handleLine(String result)
        {
            System.out.println(result);
        }
    }
    @Bean
    ItemReader<User> itemReader()
    {
         FlatFileItemReader<User> reader = new FlatFileItemReader<User>();
         reader.setLinesToSkip(1);
         reader.setResource(new ClassPathResource("data.csv"));
         reader.setLineMapper(new DefaultLineMapper<User>(){
             {
                 setLineTokenizer(new DelimitedLineTokenizer(){{
                     setNames(new String[]{"id","username","address","gender"});
                     //setDelimiter("\t");
                 }});
                 setFieldSetMapper(new BeanWrapperFieldSetMapper<User>(){{
                     setTargetType(User.class);
                 }});
 
             }
         });
         reader.setSkippedLinesCallback(new CustomLineCallbackHandler());
         return reader;
    }
 
    @Bean
    JdbcBatchItemWriter jdbcBatchItemWriter()
    {
        JdbcBatchItemWriter writer = new JdbcBatchItemWriter();
        writer.setDataSource(dataSource);
        writer.setSql("insert into user(id,username,address,gender)" + "values(:id,:username,:address,:gender)");
        writer.setItemSqlParameterSourceProvider(new BeanPropertyItemSqlParameterSourceProvider());
        return  writer;
    }
 
    @Bean
    Step csvStep()
    {
        return  stepBuilderFactory.get("csvStep")
                .<User,User>chunk(2)
                .reader(itemReader())
                .writer(jdbcBatchItemWriter())
                .build();
    }
 
    @Bean
    Job csvJob()
    {
        return jobBuilderFactory.get("csvJob")
                .start(csvStep())
                .build();
    }
 
}

5.實體類

public class User {
    private Integer id;
 
    private String username;
 
    private  String address;
 
    private  String gender;
    public User(Integer id,String username,String address,String gender)
    {
        this.username= username;
        this.id = id;
        this.address = address;
        this.gender = gender;
    }
 
    public  User()
    {
        super();
    }
    public void setId(Integer id) {
        this.id = id;
    }
    public void setUsername(String username) {
        this.username = username;
    }
    public void setGender(String gender) {
        this.gender = gender;
    }
    public void setAddress(String address) {
        this.address = address;
    }
 
    public String getUsername() {
        return username;
    }
 
    public String getGender() {
        return gender;
    }
    
    public Integer getId() {
        return id;
    }
    
    public String getAddress() {
        return address;
    }
 
    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", username='" + username + '\'' +
                ", address='" + address + '\'' +
                ", gender='" + gender + '\'' +
                '}';
    }
}

6.創(chuàng)建Controller

@RestController
public class BatchController {
    @Autowired
    JobLauncher jobLauncher;
 
    @Autowired
    Job job;
 
    @GetMapping("/batch")
    public void batch()
    {
        try {
            jobLauncher.run(job,new JobParametersBuilder().toJobParameters());
 
        }catch (Exception e)
        {
            e.printStackTrace();
        }
    }
 
}

這里配置就結(jié)束了，數(shù)據(jù)庫會增加一些自動創(chuàng)建到表，但是User表不能自動創(chuàng)建和字段需要提前設(shè)置，否則會出現(xiàn)錯誤，還需要進一步處理，待完成。暫做標記！

到此這篇關(guān)于SpringBoot批處理的實現(xiàn)示例的文章就介紹到這了,更多相關(guān)SpringBoot批處理內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: