ForfaitAudioStepConfig.java

package com.sintia.ffl.admin.audio.services.batch.refpromoteur.export;

import com.sintia.ffl.admin.audio.dal.entities.ForfaitAudio;
import com.sintia.ffl.admin.audio.dal.repositories.ForfaitAudioRepository;
import com.sintia.ffl.admin.staging.FileUtils;
import com.sintia.ffl.admin.staging.step.export.AbstractPromoteurExportStepConfig;
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 java.io.IOException;

@Configuration
public class ForfaitAudioStepConfig extends AbstractPromoteurExportStepConfig<ForfaitAudio, Integer, ForfaitAudioRepository> {

    public ForfaitAudioStepConfig(){
        super(TypeStaging.REF_PROMOTEUR, "findByPromoteurId");
    }

    @Bean
    public Step forfaitAudioStep(ForfaitAudioRepository forfaitAudioRepository, StepBuilderFactory stepBuilderFactory, FileUtils fileUtils) throws IOException {
        return super.buildStep(stepBuilderFactory, forfaitAudioRepository, fileUtils);
    }

    @Bean("forfaitAudioReader")
    @StepScope
    @Override
    public AbstractItemStreamItemReader<ForfaitAudio> reader(
            ForfaitAudioRepository repository,
            @Value("#{jobParameters['idPromoteur']}") Integer idPromoteur,
            @Value("#{jobParameters['codePromoteur']}") String codePromoteur
    ) throws IOException {
        return super.buildReader(repository, idPromoteur, codePromoteur);
    }

    @Bean("forfaitAudioWriter")
    @StepScope
    @Override
    public AsyncItemWriter<ForfaitAudio> 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);
    }

}