Spring Batch 独自のItemReaderの実装

概要

Spring BatchではChankItemReaderを任意に入れ替えることができます。

CSVの読み込み

Spring BootのCSVファイルを読み込むライブラリを用いてItemReaderを作成します。

ItemReader

読み込むCSVファイルのデータ構造を作成します。
namevalueというふたつのフィールドを持った簡単なデータ構造を定義します。

  • SourceItemDto.java
    package com.sios.ex.springbatch;
    
    import lombok.Getter;
    import lombok.NoArgsConstructor;
    import lombok.Setter;
    
    public class SourceItemDto {
    
    	@Setter
    	@Getter
    	private String name;
    
    	@Setter
    	@Getter
    	private String value;
    
    	public SourceItemDto() {
    
    	}
    
    }
    

CSVファイルを読み込んでSourceItemDtoを生成するItemReaderを作成します。

  • SpringBatchSamp2e1Configuration.javaより抜粋
    @Bean
    @StepScope
    public FlatFileItemReader<SourceItemDto> sourceItemReader(
        @Value("#{jobParameters['filename']}") String filename) {
      FlatFileItemReader<SourceItemDto> reader = new FlatFileItemReader<SourceItemDto>();
      reader.setResource(new FileSystemResource(filename));
      reader.setLineMapper(new DefaultLineMapper<SourceItemDto>() {{
        setLineTokenizer(new DelimitedLineTokenizer() {{
          setNames(new String[] {"name", "value"});
        }});
        setFieldSetMapper(new BeanWrapperFieldSetMapper<SourceItemDto>() {{
          setTargetType(SourceItemDto.class);
        }});
      }});
      return reader;
    }
    

「@Value(“#{jobParameters[‘filename’]}”) String filename」はJobに指定したパラメータを受け取ります。
この場合@StepScodeアノテーションを指定する必要があります。

ItemProcessor

読み込んだCSVをデータベースに出力するためにデータベースのテーブルにエンティティを定義します。

  • TargetItemEntity.java
    package com.sios.ex.springbatch;
    
    import javax.persistence.Column;
    import javax.persistence.Entity;
    import javax.persistence.Id;
    import javax.persistence.Table;
    
    import lombok.Data;
    import lombok.Getter;
    
    @Data
    @Entity
    @Table(name = "target_item")
    public class TargetItemEntity {
    
    	@Id
    	@Column(name = "name", nullable = false)
    	@Getter
    	private String name;
    
    	@Column(name = "value", nullable = false)
    	@Getter
    	private String value;
    
    	public TargetItemEntity() {
    
    	}
    
    	public TargetItemEntity(
    			String name,
    			String value) {
    		this.name = name;
    		this.value = value;
    	}
    }
    

ItemProcessorはCSVから読み込んだSourceItemDtoのオブジェクトを受け取りTargetItemEntityのオブジェクトを返します。

  • TargetItemProcessor.java
    package com.sios.ex.springbatch;
    
    import org.springframework.batch.item.ItemProcessor;
    
    public class TargetItemProcessor implements ItemProcessor<SourceItemDto, TargetItemEntity> {
    
    	@Override
    	public TargetItemEntity process(SourceItemDto item) throws Exception {
    		return new TargetItemEntity(item.getName(), item.getValue());
    	}
    
    }
    [/cdoe]
    
    ### ItemWriter
    
    **ItemWriter**はデータベースの**target_item**テーブルに出力するシンプルなものになります。
    
    [code]
      @Bean
      public ItemWriter<TargetItemEntity> targetItemWriter() {
        RepositoryItemWriter<TargetItemEntity> writer = new RepositoryItemWriter<>();
        writer.setRepository(targetItemRepository);
        writer.setMethodName("save");
        return writer;
      }
    

ItemReaderの作成

ItemReaderItemReader<T>を継承して作成することができます。
上記の例で使用したCSVファイルを読み込むItemReaderを実装する場合は次の通りです。

  • CsvSourceItemReader.java
    package com.sios.ex.springbatch;
    
    import java.io.BufferedReader;
    import java.io.FileNotFoundException;
    import java.io.FileReader;
    import java.io.IOException;
    import java.io.Reader;
    import java.util.ArrayList;
    
    import org.springframework.batch.item.ItemReader;
    import org.springframework.batch.item.NonTransientResourceException;
    import org.springframework.batch.item.ParseException;
    import org.springframework.batch.item.UnexpectedInputException;
    
    public class CsvSourceItemReader implements ItemReader<SourceItemDto> {
    
    	private String filename;
    	private ArrayList<SourceItemDto> items = null;
    	private int nextIndex;
    
    	public CsvSourceItemReader(String filename) {
    		this.filename = filename;
    	}
    
    	private void readCsvFile() {
    		items = new ArrayList<SourceItemDto>();
    		try {
    			BufferedReader reader = new BufferedReader(new FileReader(filename));
    			try {
    				for (;;) {
    					String line = reader.readLine();
    					if (line == null) break;
    					String[] fields = line.split(",", 2);
    					SourceItemDto item = new SourceItemDto();
    					item.setName(fields[0]);
    					item.setValue(fields[1]);
    					items.add(item);
    				}
    			} finally {
    				reader.close();
    			}
    		} catch (FileNotFoundException e) {
    			e.printStackTrace();
    		} catch (IOException e) {
    			e.printStackTrace();
    		}
    	}
    
    	@Override
    	public SourceItemDto read()
    			throws Exception, UnexpectedInputException, ParseException, NonTransientResourceException {
    		if (items == null) {
    			readCsvFile();
    			nextIndex = 0;
    		}
    		SourceItemDto item = null;
    		if (nextIndex < items.size()) {
    			item = items.get(nextIndex);
    			nextIndex++;
    		}
    		return item;
    	}
    
    }
    

read()メソッドが呼び出されるたびにSourceItemDtoのオブジェクトを返します。
返すSourceItemDtoのオブジェクトが終了したらnullを返します。

StepItemReaderの定義は次のようになります。

  • SpringBatchSamp2e1Configuration.javaより抜粋
    	@Bean
    	@StepScope
    	public CsvSourceItemReader sourceItemReader(
    			@Value("#{jobParameters['filename']}") String filename) {
    		return new CsvSourceItemReader(filename);
    	}
    

これらのように独自のItemReaderを作成することでさまざまな入力に対応したバッチ処理を実現することができます。

ご覧いただきありがとうございます! この投稿はお役に立ちましたか?

役に立った 役に立たなかった

0人がこの投稿は役に立ったと言っています。

コメントを残す

メールアドレスが公開されることはありません。 が付いている欄は必須項目です