Subversion Repositories SmartDukaan

import in.shop2020.model.v1.inventory.InventoryServiceException;

import java.io.BufferedInputStream;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Date;

import javax.servlet.ServletOutputStream;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;

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

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

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

	public void downloadFBASheet() {

		try {

			WarehouseService.Client warehouseClient = new WarehouseClient().getClient();

			Map<Long, Long> transferLotItems = warehouseClient.getItemsInTransferLot(id);

			File fbaSheetFile = new File("transfer-lot-id-sheet"+new Date());

			FileOutputStream fStream = null;

			try {

				fStream = new FileOutputStream(fbaSheetFile);

			} catch (FileNotFoundException e1) {

				logger.error(e1.getMessage());

			}

	    	ByteArrayOutputStream baosXLS = new ByteArrayOutputStream();

		    Workbook wb = new HSSFWorkbook();

		    Sheet fbaSheet = wb.createSheet("FBA Sheet");

		    Row merchantShipmentName = fbaSheet.createRow((short)0);

		    merchantShipmentName.createCell(0).setCellValue("MerchantShipmentName");

		    Row labelPrepPreference = fbaSheet.createRow((short)1);

		    labelPrepPreference.createCell(0).setCellValue("LabelPrepPreference");

		    Row addressName = fbaSheet.createRow((short)2);

		    addressName.createCell(0).setCellValue("AddressName");

		    Row addressFieldOne = fbaSheet.createRow((short)3);

		    addressFieldOne.createCell(0).setCellValue("AddressFieldOne");

		    Row addressFieldTwo = fbaSheet.createRow((short)4);

		    addressFieldTwo.createCell(0).setCellValue("AddressFieldTwo");

		    Row addressCity = fbaSheet.createRow((short)5);

		    addressCity.createCell(0).setCellValue("AddressCity");

		    Row addressCountryCode = fbaSheet.createRow((short)6);

		    addressCountryCode.createCell(0).setCellValue("AddressCountryCode");

		    Row addressStateOrRegion = fbaSheet.createRow((short)7);

		    addressStateOrRegion.createCell(0).setCellValue("AddressStateOrRegion");

		    Row addressPostalCode = fbaSheet.createRow((short)8);

		    addressPostalCode.createCell(0).setCellValue("AddressPostalCode");

		    Row addressDistrict = fbaSheet.createRow((short)9);

		    addressDistrict.createCell(0).setCellValue("AddressDistrict");

		    Row blankRow = fbaSheet.createRow((short)10);

		    //blankRow.createCell(0).setCellValue("AddressPostalCode");

		    Row skuHeaders = fbaSheet.createRow((short)11);

		    skuHeaders.createCell(0).setCellValue("MerchantSKU");

		    skuHeaders.createCell(1).setCellValue("Quantity");

		    int rowCount = 12;

		    for(Entry<Long, Long> itemQuantity : transferLotItems.entrySet()) {

		    	Row newRow = fbaSheet.createRow((short)rowCount);

		    	newRow.createCell(0).setCellValue("FBA"+itemQuantity.getKey());

			    newRow.createCell(1).setCellValue(itemQuantity.getValue());

			    rowCount++;

		    }

		    try {

			    wb.write(baosXLS);

			    baosXLS.close();

			    baosXLS.writeTo(fStream);

			    fStream.flush();

				fStream.close();

		    } catch(IOException e) {

		    	//TODO

		    }

		    byte[] buffer = null;

            buffer = new byte[(int) fbaSheetFile.length()];

            InputStream input = null;

            try {

                int totalBytesRead = 0;

                input = new BufferedInputStream(new FileInputStream(fbaSheetFile));

                while (totalBytesRead < buffer.length) {

                    int bytesRemaining = buffer.length - totalBytesRead;

                    // input.read() returns -1, 0, or more :

                    int bytesRead = input.read(buffer, totalBytesRead,

                            bytesRemaining);

                    if (bytesRead > 0) {

                        totalBytesRead = totalBytesRead + bytesRead;

                    }

                }

                 /* the above style is a bit tricky: it places bytes into the

                 * 'buffer' array; 'buffer' is an output parameter; the while

                 * loop usually has a single iteration only.*/

            } finally {

                input.close();

            }

            response.setContentType("application/vnd.ms-excel");

            response.setHeader("Content-disposition", "inline; filename="

                    + fbaSheetFile.getName());

            ServletOutputStream sos = response.getOutputStream();

            sos.write(buffer);

            sos.flush();

		} catch (TException tex) {

			logger.error("Error in generating fba sheet for transfer Lot Id " + id,tex);

		}  catch (FileNotFoundException fnfex) {

			logger.error("Error in generating fba sheet for transfer Lot Id " + id,fnfex);

		} catch (IOException ioex) {

			logger.error("Error in generating fba sheet for transfer Lot Id " + id,ioex);

		}

	}