|
| 1 | +package com.back.global.appConfig; |
| 2 | + |
| 3 | +import com.back.global.standard.util.DecimalToFractionConverter; |
| 4 | +import lombok.RequiredArgsConstructor; |
| 5 | +import org.springframework.batch.core.Job; |
| 6 | +import org.springframework.batch.core.Step; |
| 7 | +import org.springframework.batch.core.configuration.annotation.EnableBatchProcessing; |
| 8 | +import org.springframework.batch.core.job.builder.JobBuilder; |
| 9 | +import org.springframework.batch.core.repository.JobRepository; |
| 10 | +import org.springframework.batch.core.step.builder.StepBuilder; |
| 11 | +import org.springframework.batch.item.ItemProcessor; |
| 12 | +import org.springframework.batch.item.file.FlatFileItemReader; |
| 13 | +import org.springframework.batch.item.file.FlatFileItemWriter; |
| 14 | +import org.springframework.batch.item.file.builder.FlatFileItemReaderBuilder; |
| 15 | +import org.springframework.batch.item.file.builder.FlatFileItemWriterBuilder; |
| 16 | +import org.springframework.batch.item.file.mapping.PassThroughLineMapper; |
| 17 | +import org.springframework.batch.item.file.transform.PassThroughLineAggregator; |
| 18 | +import org.springframework.context.annotation.Bean; |
| 19 | +import org.springframework.context.annotation.Configuration; |
| 20 | +import org.springframework.core.io.FileSystemResource; |
| 21 | +import org.springframework.transaction.PlatformTransactionManager; |
| 22 | + |
| 23 | +// DecimalToFractionConverter가 있다고 가정합니다. |
| 24 | +// import com.example.DecimalToFractionConverter; |
| 25 | + |
| 26 | +@Configuration |
| 27 | +@EnableBatchProcessing |
| 28 | +@RequiredArgsConstructor |
| 29 | +public class CocktailBatchConfig { |
| 30 | + |
| 31 | + private final JobRepository jobRepository; |
| 32 | + private final PlatformTransactionManager transactionManager; |
| 33 | + |
| 34 | + private static final String INPUT_PATH = "src/main/resources/cocktails.csv"; |
| 35 | + private static final String OUTPUT_PATH = "src/main/resources/cocktails_clean.csv"; |
| 36 | + |
| 37 | + // 1.Reader: CSV 파일 한 줄씩 읽기 |
| 38 | + @Bean |
| 39 | + public FlatFileItemReader<String> reader() { |
| 40 | + return new FlatFileItemReaderBuilder<String>() |
| 41 | + .name("cocktailReader") |
| 42 | + .resource(new FileSystemResource(INPUT_PATH)) |
| 43 | + .lineMapper(new PassThroughLineMapper()) |
| 44 | + .build(); |
| 45 | + } |
| 46 | + |
| 47 | + // 2️⃣ Processor: ingredient 컬럼(6번째)만 DecimalToFractionConverter로 변환 |
| 48 | + @Bean |
| 49 | + public ItemProcessor<String, String> processor() { |
| 50 | + return line -> { |
| 51 | + // CSV 한 줄을 안전하게 split |
| 52 | + String[] columns = line.split(",(?=(?:[^\"]*\"[^\"]*\")*[^\"]*$)", -1); |
| 53 | + |
| 54 | + // Header인지 확인: "ingredient"라는 컬럼 이름으로 간단 체크 |
| 55 | + if (columns.length > 6 && !columns[6].equalsIgnoreCase("ingredient")) { |
| 56 | + // DecimalToFractionConverter는 별도로 구현되어 있어야 합니다. |
| 57 | + columns[6] = DecimalToFractionConverter.convert(columns[6]); |
| 58 | + } |
| 59 | + |
| 60 | + // 다시 CSV 문자열로 합치기 |
| 61 | + return String.join(",", columns); |
| 62 | + }; |
| 63 | + } |
| 64 | + |
| 65 | + // 3️⃣ Writer: 변환된 CSV 출력 (변화 없음) |
| 66 | + @Bean |
| 67 | + public FlatFileItemWriter<String> writer() { |
| 68 | + return new FlatFileItemWriterBuilder<String>() |
| 69 | + .name("cocktailWriter") |
| 70 | + .resource(new FileSystemResource(OUTPUT_PATH)) |
| 71 | + .lineAggregator(new PassThroughLineAggregator<>()) |
| 72 | + .build(); |
| 73 | + } |
| 74 | + |
| 75 | + // 4️⃣ Step: Chunk 단위 처리 (StepBuilderFactory 대체) |
| 76 | + @Bean |
| 77 | + public Step convertStep() { |
| 78 | + // StepBuilder를 직접 생성하고, jobRepository와 transactionManager를 주입합니다. |
| 79 | + return new StepBuilder("convertStep", jobRepository) |
| 80 | + // chunk 메서드에 transactionManager를 인수로 전달합니다. |
| 81 | + .<String, String>chunk(5, transactionManager) |
| 82 | + .reader(reader()) |
| 83 | + .processor(processor()) |
| 84 | + .writer(writer()) |
| 85 | + .build(); |
| 86 | + } |
| 87 | + |
| 88 | + // 5️⃣ Job 정의 (JobBuilderFactory 대체) |
| 89 | + @Bean |
| 90 | + public Job convertJob() { |
| 91 | + // JobBuilder를 직접 생성하고, jobRepository를 주입합니다. |
| 92 | + return new JobBuilder("convertJob", jobRepository) |
| 93 | + .start(convertStep()) |
| 94 | + .build(); |
| 95 | + } |
| 96 | +} |
0 commit comments