SupplementVerreAutreIteStepConfig.java

package com.sintia.ffl.admin.optique.services.batch.refpromoteur.export.ite;

import com.sintia.ffl.admin.optique.dal.entities.SupplementVerreAutreIte;
import com.sintia.ffl.admin.staging.FileUtils;
import com.sintia.ffl.admin.staging.GenerericJDBCItemReadFromDB;
import com.sintia.ffl.admin.staging.step.export.jdbc.AbstractSinglePromoteurJDBCExportStepConfig;
import com.sintia.ffl.adminui.commons.enums.TypeStaging;
import com.sintia.ffl.core.commons.enums.CodePromoteur;
import org.springframework.batch.core.Step;

import org.springframework.batch.core.configuration.annotation.StepScope;
import org.springframework.batch.integration.async.AsyncItemWriter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.batch.core.repository.JobRepository;
import org.springframework.transaction.PlatformTransactionManager;


import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Timestamp;

@Configuration
public class SupplementVerreAutreIteStepConfig extends
        AbstractSinglePromoteurJDBCExportStepConfig<SupplementVerreAutreIte> {

    private static final String QUERY = "select id_supplement_verre_autre_ite, c_supplement_autre, l_supplement_autre, id_type_supplement_ite, " +
            "d_creation, d_maj " +
            "from supplement_verre_autre_ite";

    public SupplementVerreAutreIteStepConfig() {
        super(TypeStaging.REF_PROMOTEUR, (rs, rowNum) -> {
            Timestamp dCreation = rs.getTimestamp("d_creation");
            Timestamp dMaj = rs.getTimestamp("d_maj");
            return new SupplementVerreAutreIte(
                    rs.getInt("id_supplement_verre_autre_ite"),
                    rs.getString("c_supplement_autre"),
                    rs.getString("l_supplement_autre"),
                    rs.getInt("id_type_supplement_ite"),
                    dCreation == null ? null : dCreation.toLocalDateTime(),
                    dMaj == null ? null : dMaj.toLocalDateTime()
            );
        }, CodePromoteur.ITE);
    }

    @Bean
    public Step supplementVerreAutreIteStep(JobRepository jobRepository, PlatformTransactionManager transactionManager, FileUtils fileUtils,
                                            DataSource dataSource) throws IOException {
        return super.buildStep( jobRepository,  transactionManager, fileUtils, dataSource);
    }


    @Bean("supplementVerreAutreIteReader")
    @StepScope
    @Override
    protected GenerericJDBCItemReadFromDB<SupplementVerreAutreIte> reader(
            DataSource dataSource,
            @Value("#{jobParameters['idPromoteur']}") Integer idPromoteur,
            @Value("#{jobParameters['codePromoteur']}") String codePromoteur) throws IOException {
        return super.buildReader(dataSource, QUERY, codePromoteur);
    }


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