ParamPromoteurSequencesUpdateStepConfig.java

package com.sintia.ffl.admin.optique.services.batch.parampromoteur.load;

import com.sintia.ffl.admin.staging.FileUtils;
import com.sintia.ffl.admin.staging.sequences.IntegerSequence;
import com.sintia.ffl.admin.staging.step.imports.sequences.SequencePromoteurImportStepConfig;
import com.sintia.ffl.adminui.commons.enums.TypeStaging;
import com.sintia.ffl.core.commons.enums.CodePromoteur;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.annotation.JobScope;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepScope;
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 org.springframework.core.task.AsyncTaskExecutor;

import javax.sql.DataSource;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

@Slf4j
@Configuration
public class ParamPromoteurSequencesUpdateStepConfig extends SequencePromoteurImportStepConfig {

    public ParamPromoteurSequencesUpdateStepConfig(){
        super(
                TypeStaging.PARAM_PROMOTEUR
        );
    }

    @Bean
    @JobScope
    public Step paramPromoteurSequencesUpdateStep(
            StepBuilderFactory stepBuilderFactory,
            FileUtils fileUtils,
            DataSource dataSource,
            AsyncTaskExecutor asyncExecutor,
            @Value("#{jobParameters['codePromoteur']}") String codePromoteur
    ) throws IOException {

        Set<String> sequencesNames = new HashSet<>(List.of(
                "assureur_id_assureur_seq",
                "config_edition_id_config_edition_seq",
                "promoteur_id_promoteur_seq"
        ));

        // Table spécifique ITE
        if(StringUtils.equals(CodePromoteur.ITE.name(), codePromoteur)){
            sequencesNames.add(
                    "information_specifique_ite_id_information_specifique_ite_seq"
            );
            sequencesNames.add("distributeur_fabricant_asso_i_id_distributeur_fabricant_ass_seq");
            sequencesNames.add("distributeur_ite_id_distributeur_ite_seq");
            sequencesNames.add("fabricant_ite_id_fabricant_ite_seq");
            sequencesNames.add("information_specifique_ite_id_information_specifique_ite_seq");
            sequencesNames.add("modele_verre_ite_id_modele_verre_ite_seq");
            sequencesNames.add("supplement_verre_autre_ite_id_supplement_verre_autre_ite_seq");
            sequencesNames.add("supplement_verre_ite_id_supplement_verre_ite_seq");
            sequencesNames.add("traitement_integre_ite_id_trait_integre_ite_seq");
            sequencesNames.add("type_materiau_ite_id_type_materiau_ite_seq");
            sequencesNames.add("type_supplement_ite_id_type_supplement_ite_seq");
            sequencesNames.add("type_verre_ite_id_type_verre_ite_seq");
            sequencesNames.add("verre_supplement_asso_ite_id_verre_supplement_asso_ite_seq");
            sequencesNames.add("verre_trait_integres_asso_ite_id_verre_trait_integres_asso__seq");
        }

        return super.buildStep(
                stepBuilderFactory,
                fileUtils,
                asyncExecutor,
                dataSource,
                sequencesNames
        );
    }

    @Bean("paramPromoteurUpdateSequencesReader")
    @StepScope
    @Override
    public AbstractItemStreamItemReader<IntegerSequence> reader(
            FileUtils fileUtils,
            @Value("#{jobParameters['envSource'] ?: 'pprod'}") String envSource,
            @Value("#{jobParameters['envCible'] ?: 'pprod'}") String envCible,
            @Value("#{jobParameters['codePromoteur']}") String codePromoteur
    ) throws IOException {
        return super.buildReader(fileUtils, envSource, envCible, codePromoteur);
    }

}