RefPromoteurSequencesExportStepConfig.java

package com.sintia.ffl.admin.optique.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<>(List.of(
                "distributeur_promoteur_id_distributeur_promoteur_seq",
                "fabricant_promoteur_id_fabricant_promoteur_seq",
                "identifiant_flux_sic_id_identifiant_flux_sic_seq",
                "lentille_controle_remise_prom_id_lentille_controle_remise_p_seq",
                "monture_controle_remise_promo_id_monture_controle_remise_pr_seq"
        ));

        // Tables spécifiques GSA
        if(StringUtils.equals(CodePromoteur.GSA.name(), codePromoteur)){
            sequencesNames.addAll(List.of(
                    "fabricant_offre_specifique_gs_id_fabricant_offre_specifique_seq",
                    "offre_specifique_gsa_id_offre_specifique_gsa_seq"
            ));
        }

        // Tables spécifiques ITE
        if(StringUtils.equals(CodePromoteur.ITE.name(), codePromoteur)){
            sequencesNames.addAll(List.of(
                    "distributeur_fabricant_asso_i_id_distributeur_fabricant_ass_seq",
                    "distributeur_ite_id_distributeur_ite_seq",
                    "fabricant_ite_id_fabricant_ite_seq",
                    // Pas de séquence pour la table incompatibilite_supplement_trt_integre_ite ?
                    "modele_verre_ite_id_modele_verre_ite_seq",
                    "supplement_verre_autre_ite_id_supplement_verre_autre_ite_seq",
                    "supplement_verre_ite_id_supplement_verre_ite_seq",
                    "traitement_integre_ite_id_trait_integre_ite_seq",
                    "type_materiau_ite_id_type_materiau_ite_seq",
                    "type_supplement_ite_id_type_supplement_ite_seq",
                    "type_verre_ite_id_type_verre_ite_seq",
                    "verre_supplement_asso_ite_id_verre_supplement_asso_ite_seq",
                    "verre_trait_integres_asso_ite_id_verre_trait_integres_asso__seq"
            ));
        }

        // Tables spécifiques SCS
        if(StringUtils.equals(CodePromoteur.SCS.name(), codePromoteur)){
            sequencesNames.addAll(List.of(
                    "marque_modele_monture_scs_id_marque_modele_monture_scs_seq",
                    "remise_monture_scs_id_remise_monture_scs_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);
    }

}