Subversion Repositories SmartDukaan

import java.io.FileInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import org.apache.poi.ss.usermodel.Cell;

import org.apache.poi.ss.usermodel.CellType;

import org.apache.poi.ss.usermodel.Row;

import org.apache.poi.ss.usermodel.Row.MissingCellPolicy;

import org.apache.poi.xssf.usermodel.XSSFRow;

import org.apache.poi.xssf.usermodel.XSSFSheet;

import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import com.spice.profitmandi.common.model.PartnerTargetModel;

import com.spice.profitmandi.common.model.ProfitMandiConstants;

	@Override

	public void createOffers(InputStream fileInputStream) throws Exception {

		Map<CreateOfferRequest, List<PartnerTargetModel>> offerPartnerTargetMap = this.parseFromExcel(fileInputStream);

		for(Map.Entry<CreateOfferRequest, List<PartnerTargetModel>> partnerTargetEntry: offerPartnerTargetMap.entrySet()) {

			CreateOfferRequest createOfferRequest = partnerTargetEntry.getKey();

			Map<List<Integer>, List<Integer>> targetPartnerMap = partnerTargetEntry.getValue().stream().collect(Collectors.groupingBy(PartnerTargetModel::getTargets, Collectors.mapping(PartnerTargetModel::getFofoId, Collectors.toList())));

					for (Map.Entry<List<Integer>, List<Integer>> targetPartnersEntry : targetPartnerMap.entrySet()) {

						List<Integer> fofoIds = createOfferRequest.getPartnerCriteria().getFofoIds();

						List<Integer> filterFofoIds = targetPartnersEntry.getValue().stream().filter(x->!fofoIds.contains(x)).collect(Collectors.toList());

						List<Integer> targets = targetPartnersEntry.getKey();

						createOfferRequest.getPartnerCriteria().setFofoIds(filterFofoIds);

						int counter = 0;

						for (com.spice.profitmandi.dao.model.TargetSlab targetSlab : createOfferRequest.getTargetSlabs()) {

							targetSlab.setOnwardsAmount(targets.get(counter));

							counter++;

						}

						this.addOfferService(createOfferRequest);

					}

		}

	}

	private Map<CreateOfferRequest, List<PartnerTargetModel>> parseFromExcel(InputStream inputStream) throws ProfitMandiBusinessException {

		Map<CreateOfferRequest, List<PartnerTargetModel>> offerPartnerTargetMap =  new HashMap<>();

		XSSFWorkbook myWorkBook = null;

		try {

			myWorkBook = new XSSFWorkbook(inputStream);

			myWorkBook.setMissingCellPolicy(MissingCellPolicy.RETURN_BLANK_AS_NULL);

			for(int i=0; i< myWorkBook.getNumberOfSheets();i++) {

				XSSFSheet mySheet = myWorkBook.getSheetAt(i);

				int offerId = Integer.parseInt(mySheet.getSheetName());

				CreateOfferRequest existingOffer = this.getOffer(offerId);

				List<PartnerTargetModel> partnerTargetModels = new ArrayList<>();

				offerPartnerTargetMap.put(existingOffer, partnerTargetModels);

				this.printIterator(mySheet);

				LOGGER.info("rowCellNum {}", mySheet.getLastRowNum());

				for (int rowNumber = 1; rowNumber <= mySheet.getLastRowNum(); rowNumber++) {

					XSSFRow row = mySheet.getRow(rowNumber);

					LOGGER.info("row {}", row);

					PartnerTargetModel partnerTargetModel = new PartnerTargetModel();

					if (row.getCell(0) != null && row.getCell(0).getCellTypeEnum() == CellType.NUMERIC) {

						partnerTargetModel.setFofoId((int) row.getCell(0).getNumericCellValue());

					} else {

						ProfitMandiBusinessException profitMandiBusinessException = new ProfitMandiBusinessException(

								"FOFO ID", row.getCell(0).toString(), "TGLSTNG_VE_1010");

						LOGGER.error("Excel file parse error : ", profitMandiBusinessException);

						throw profitMandiBusinessException;

					}

					List<Integer> targets = new ArrayList<>();

					for (int cellNumber = 1; cellNumber <= existingOffer.getTargetSlabs().size(); cellNumber++) {

						if (row.getCell(cellNumber) != null

								&& row.getCell(cellNumber).getCellTypeEnum() == CellType.NUMERIC) {

							targets.add((int) row.getCell(cellNumber).getNumericCellValue());

						} else {

							ProfitMandiBusinessException profitMandiBusinessException = new ProfitMandiBusinessException(

									"Target Column issue at " + rowNumber + ", " + cellNumber, row.getCell(cellNumber),

									"Invalid target");

							LOGGER.error("Excel file parse error : ", profitMandiBusinessException);

							throw profitMandiBusinessException;

						}

					}

					partnerTargetModel.setTargets(targets);

					partnerTargetModels.add(partnerTargetModel);

				}

			}

			myWorkBook.close();

		} catch ( ProfitMandiBusinessException pbse) {

			throw pbse;

		} catch (IOException ioException) {

			ioException.printStackTrace();

			throw new ProfitMandiBusinessException(ProfitMandiConstants.EXCEL_FILE, ioException.getMessage(),

					"EXL_VE_1000");

		} finally {

			if (myWorkBook != null) {

				try {

					myWorkBook.close();

				} catch (IOException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				}

			}

		}

		LOGGER.info("offerPartnerTargetMap {}", offerPartnerTargetMap);

		return offerPartnerTargetMap;

	}

	private void printIterator (XSSFSheet sheet) {

		 Iterator<Row> rowIterator = sheet.iterator();

         while (rowIterator.hasNext()) 

         {

             Row row = rowIterator.next();

             //For each row, iterate through all the columns

             Iterator<Cell> cellIterator = row.cellIterator();

             while (cellIterator.hasNext()) 

             {

                 Cell cell = cellIterator.next();

                 //Check the cell type and format accordingly

                 switch (cell.getCellType()) 

                 {

                     case Cell.CELL_TYPE_NUMERIC:

                         System.out.print(cell.getNumericCellValue() + "t");

                         break;

                     case Cell.CELL_TYPE_STRING:

                         System.out.print(cell.getStringCellValue() + "t");

                         break;

                 }

             }

             System.out.println("");

         }

	}