Subversion Repositories SmartDukaan

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

	private String virtualWarehouseId;

	public String getVirtualWarehouseId() {

		return virtualWarehouseId;

	}

	public void setVirtualWarehouseId(String virtualWarehouseId) {

		this.virtualWarehouseId = virtualWarehouseId;

	}

	public String loadDownloadVirtualDiv(){

		if(!ReportsUtils.canAccessReport((Long)session.getAttribute(ReportsUtils.ROLE), request.getRequestURI().substring(request.getContextPath().length()).split("/")[1].split("!")[0])) {

			logger.info(request.getRequestURI().substring(request.getContextPath().length()).split("/")[1].split("!")[0]);

			return "authfail";

		}

		return "download-virtual-div";

	}

	public String downloadVirtualInventoryForWarehouse() throws NumberFormatException, TException, IOException{

		try{

			InventoryClient inventoryServiceClient = new InventoryClient();

			in.shop2020.model.v1.inventory.InventoryService.Client inventoryClient = inventoryServiceClient.getClient();

			Map<Long, ItemInventory> inventoryMap = inventoryClient.getInventorySnapshot(Long.valueOf(virtualWarehouseId));

			File file = new File("/tmp/currentinventory.xls");

			HSSFWorkbook hwb=new HSSFWorkbook();

			HSSFSheet sheet =  hwb.createSheet("Inventory");

			HSSFRow rowhead=   sheet.createRow((short)0);

			rowhead.createCell((short) 0).setCellValue("item_id");

			rowhead.createCell((short) 1).setCellValue("Availability");

			rowhead.createCell((short) 2).setCellValue("Reserved");

			rowhead.createCell((short) 3).setCellValue("Held");

			int iterator= 1;

			for (Map.Entry<Long,ItemInventory> entry : inventoryMap.entrySet()) {

				HSSFRow row = sheet.createRow((short)iterator);

				row.createCell((short) 0).setCellValue(entry.getKey());

				row.createCell((short) 1).setCellValue(entry.getValue().getAvailability().get(Long.valueOf(virtualWarehouseId)));

				row.createCell((short) 2).setCellValue(entry.getValue().getReserved().get(Long.valueOf(virtualWarehouseId)));

				row.createCell((short) 3).setCellValue(entry.getValue().getHeld().get(Long.valueOf(virtualWarehouseId)));

				iterator++;

			}

			FileOutputStream fileOut = null;

			try {

				fileOut = new FileOutputStream(file);

			} catch (FileNotFoundException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

			try {

				hwb.write(fileOut);

			} catch (IOException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

			try {

				fileOut.close();

			} catch (IOException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

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

			InputStream input = null;

			try {

				int totalBytesRead = 0;

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

				while(totalBytesRead < buffer.length){

					int bytesRemaining = buffer.length - totalBytesRead;

					int bytesRead = input.read(buffer, totalBytesRead, bytesRemaining); 

					if (bytesRead > 0){

						totalBytesRead = totalBytesRead + bytesRead;

					}

				}

			}

			finally {

				input.close();

				file.delete();

			}

			response.setHeader("Content-Disposition", "attachment; filename=\"Inventory.xls\"");

			response.setContentType("application/octet-stream");

			ServletOutputStream sos;

			try {

				sos = response.getOutputStream();

				sos.write(buffer);

				sos.flush();

			} catch (IOException e) {

				System.out.println("Unable to stream the manifest file");

			}

		}

		catch(Exception e){

			setResult("Something went wrong");

			return "item-details-json";

		}

		setResult("Streaming done");

		return "item-details-json";

	}

	public static void main(String args[]) throws NumberFormatException, TException, IOException{

		BulkAddController b =  new BulkAddController();

		b.setVirtualWarehouseId("3295");

		b.downloadVirtualInventoryForWarehouse();

	}