io.mosip.preregistration.batchjob.impl.SlotAvailabilityPartitioner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pre-registration-batchjob Show documentation
There is a newer version: 1.3.0-beta.1
Show newest version
package io.mosip.preregistration.batchjob.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.springframework.batch.core.partition.support.Partitioner;
import org.springframework.batch.item.ExecutionContext;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import io.mosip.kernel.core.logger.spi.Logger;
import io.mosip.preregistration.batchjob.code.PreRegBatchContants;
import io.mosip.preregistration.batchjob.helper.RegCenterIdsHolder;
import io.mosip.preregistration.batchjob.helper.RestHelper;
import io.mosip.preregistration.batchjob.model.RegistrationCenterDto;
import io.mosip.preregistration.core.config.LoggerConfiguration;

/**
 * @author Mahammed Taheer
 * @since 1.2.0
 *
 */

@Component
public class SlotAvailabilityPartitioner implements Partitioner {
    
    private Logger LOGGER = LoggerConfiguration.logConfig(SlotAvailabilityPartitioner.class);

    @Autowired
	private RestHelper restHelper;
    
    @Override
	public Map partition(int gridSize) {
        Map regCentersMap = new HashMap(gridSize);

        RegCenterIdsHolder idsHolder = RegCenterIdsHolder.getInstance();
        LOGGER.info(PreRegBatchContants.SESSIONID, PreRegBatchContants.PRE_REG_BATCH, PreRegBatchContants.EMPTY,
                        "Ids before deleting all elements");
        idsHolder.printAllRegCenterIds();
        // call delete all elements
        idsHolder.deleteAllRegCenterIds();
        LOGGER.info(PreRegBatchContants.SESSIONID, PreRegBatchContants.PRE_REG_BATCH, PreRegBatchContants.EMPTY,
                        "Ids after deleting all elements");
        idsHolder.printAllRegCenterIds();

        int totalNoOfPages = restHelper.getRegistrationCenterTotalPages();
		LOGGER.info(PreRegBatchContants.SESSIONID, PreRegBatchContants.PRE_REG_BATCH, PreRegBatchContants.EMPTY, 
		 				"Total Number of Pages Found in Master Data: <" + totalNoOfPages + ">");

        //List regCentersList = restHelper.getRegistrationCenterDetails(null);
        //int regCentersCount = regCentersList.size();
        int partitionSize = getPartitionSize(totalNoOfPages, gridSize);

        LOGGER.info(PreRegBatchContants.SESSIONID, PreRegBatchContants.PRE_REG_BATCH, PreRegBatchContants.EMPTY, 
		 				"Total Number of Pages Found in Master Data: <" + totalNoOfPages + 
                         ">, GridSize Configured: <" + gridSize + ">, partitionSize:: <" + partitionSize + ">");
        
        int regCenterCounter = 0;
        for (int i = 0; i < gridSize; i++) {
            ExecutionContext execContext = new ExecutionContext();

            List partRegCentersList = new ArrayList<>();
            for (int j = 0; j < partitionSize; j++) {
                if (regCenterCounter >= totalNoOfPages){
                    break;
                }
                partRegCentersList.add(Integer.toString(regCenterCounter++));
            } 
            execContext.put("regCenterIdsPartList", partRegCentersList);
            execContext.putString("name", "regCenterIdsPartList-" + i);

            regCentersMap.put("regCenterPartition" + i, execContext);
            if (regCenterCounter >= totalNoOfPages){
				break;
			}
        }

        return regCentersMap;
    }

    private int getPartitionSize(int regCentersCount, int gridSize) {

        if (regCentersCount <= gridSize)
            return 1;
        
        return (regCentersCount/gridSize) + 1;
    }
}