RefPromoteurSequencesExportStepConfig.java
package com.sintia.ffl.admin.audio.services.batch.refpromoteur.export;
import com.sintia.ffl.admin.staging.FileUtils;
import com.sintia.ffl.admin.staging.sequences.IntegerSequenceRowMapper;
import com.sintia.ffl.admin.staging.sequences.Sequence;
import com.sintia.ffl.admin.staging.step.export.jdbc.SequencePromoteurExportStepConfig;
import com.sintia.ffl.adminui.commons.enums.TypeStaging;
import com.sintia.ffl.core.commons.enums.CodePromoteur;
import org.apache.commons.lang3.StringUtils;
import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepScope;
import org.springframework.batch.integration.async.AsyncItemWriter;
import org.springframework.batch.item.support.AbstractItemStreamItemReader;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.sql.DataSource;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
@Configuration
public class RefPromoteurSequencesExportStepConfig extends SequencePromoteurExportStepConfig {
protected RefPromoteurSequencesExportStepConfig() {
super(TypeStaging.REF_PROMOTEUR);
}
@Bean
public Step refPromoteurSequencesExportStep(StepBuilderFactory stepBuilderFactory, FileUtils fileUtils, DataSource dataSource) throws IOException {
return super.buildStep(
stepBuilderFactory,
fileUtils,
dataSource,
new IntegerSequenceRowMapper()
);
}
@Bean("refPromoteurSequencesReader")
@StepScope
@Override
public AbstractItemStreamItemReader<Sequence<Integer>> reader(
DataSource dataSource,
IntegerSequenceRowMapper rowMapper,
@Value("#{jobParameters['codePromoteur']}") String codePromoteur
) {
List<String> sequencesNames = new ArrayList<>();
sequencesNames.addAll(List.of(
"forfait_audio_identifiant_forfait_audio_seq",
"marque_audio_promoteur_identifiant_marque_audio_promoteur_seq"
));
// Tables spécifiques SCS
if(StringUtils.equals(CodePromoteur.SCS.name(), codePromoteur)){
sequencesNames.addAll(List.of(
"accessoire_audio_identifianaccessoire_audio_seq",
"gamme_audio_identifiant_gamme_audio_seq"
));
}
return super.buildReader(dataSource, sequencesNames, rowMapper);
}
@Bean("refPromoteurSequencesWriter")
@StepScope
@Override
public AsyncItemWriter<Sequence<Integer>> writer(
FileUtils fileUtils,
@Value("#{jobParameters['envSource'] ?: 'pprod'}") String envSource,
@Value("#{jobParameters['envCible'] ?: 'pprod'}") String envCible,
@Value("#{jobParameters['codePromoteur']}") String codePromoteur
) throws IOException {
return super.buildWriter(fileUtils, envSource, envCible, codePromoteur);
}
}