ParamPromoteurSequencesExportStepConfig.java

package com.sintia.ffl.admin.dentaire.services.batch.parampromoteur.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 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.List;

@Configuration
public class ParamPromoteurSequencesExportStepConfig extends SequencePromoteurExportStepConfig {

    protected ParamPromoteurSequencesExportStepConfig() {
        super(TypeStaging.PARAM_PROMOTEUR);
    }

    @Bean
    public Step paramPromoteurSequencesExportStep(StepBuilderFactory stepBuilderFactory, FileUtils fileUtils, DataSource dataSource) throws IOException {
        return super.buildStep(
                stepBuilderFactory,
                fileUtils,
                dataSource,
                new IntegerSequenceRowMapper()
        );
    }

    @Bean("paramPromoteurSequencesReader")
    @StepScope
    @Override
    public AbstractItemStreamItemReader<Sequence<Integer>> reader(
            DataSource dataSource,
            IntegerSequenceRowMapper rowMapper,
            @Value("#{jobParameters['codePromoteur']}") String codePromoteur
    ) {
        return super.buildReader(
                dataSource,
                List.of(
                    "assureur_id_seq",
                    "config_edition_id_config_edition_seq",
                    "promoteur_id_seq"
                ),
                rowMapper
        );
    }

    @Bean("paramPromoteurSequencesWriter")
    @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);
    }

}