Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileOutputStream;

import java.io.FileWriter;

import java.io.Writer;

import java.util.List;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Map;

import in.shop2020.model.v1.catalog.InventoryService.Client;

import in.shop2020.model.v1.catalog.Warehouse;

import in.shop2020.thrift.clients.CatalogServiceClient;

import org.apache.commons.collections.map.HashedMap;

import org.apache.struts2.rest.DefaultHttpHeaders;

import org.apache.struts2.rest.HttpHeaders;

import com.opensymphony.xwork2.ModelDriven;

public class UpdatesController implements ModelDriven<Object>{

	private InventoryUpdate inventoryUpdate = new InventoryUpdate();

	private String timestamp;

	private int errorCode = 0;

	private String errorMessage;

	private String id;

	public UpdatesController(){

	}

	@Override

	public Object getModel() {

		return this.inventoryUpdate;

	}

	public HttpHeaders show(){

		System.out.println("Warehouse Id is:  " + getId());

		CatalogServiceClient catalogServiceClient;

		Client client;

		try {

			catalogServiceClient = new CatalogServiceClient();

			client = catalogServiceClient.getClient();

			long warehouseId = Long.parseLong(getId());

			Warehouse warehouse = client.getWarehouse(warehouseId);

			setTimestamp(warehouse.getVendorString());

		} catch (Exception e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		return new DefaultHttpHeaders("lsuccess");

	}

	public HttpHeaders create(){

//		Writer wr = new BufferedWriter(new FileWriter("inventory-updates.txt"));

//		//wr.write(availability.toString());

//		wr.write(inventoryUpdate);

//		wr.close();

		System.out.println(inventoryUpdate);

		// need to store all data from object to service

		CatalogServiceClient catalogServiceClient;

		Client client;

		try {

			catalogServiceClient = new CatalogServiceClient();

			client = catalogServiceClient.getClient();

			Long warehouseId = Long.parseLong(inventoryUpdate.getwarehouseId());

			String timestamp = inventoryUpdate.getCurrentTimestamp();

			Map<String, Long> availability = new HashMap<String, Long>();

			List<Update> updates = inventoryUpdate.getUpdates();

			for(Update update: updates){

				String skuId = update.getSkuId();

				long quantity = (long) Double.parseDouble(update.getQuantity());

				if(availability.containsKey(skuId)){

					quantity = quantity + availability.get(skuId);

				}

				availability.put(skuId, quantity);

			}

			client.updateInventory(warehouseId, timestamp, availability);

		} catch (Exception e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		return new DefaultHttpHeaders("psuccess");

	}

	public int getErrorCode() {

		return errorCode;

	}

	public String getErrorMessage() {

		return errorMessage;

	}

	public String getId(){

		return id;

	}

	public void setId(String id){

		this.id = id;

	}

	public InventoryUpdate getInventoryUpdate() {

		return inventoryUpdate;

	}

	public void setInventoryUpdate(InventoryUpdate inventoryUpdate) {

		this.inventoryUpdate = inventoryUpdate;

	}

	public void setTimestamp(String timestamp) {

		this.timestamp = timestamp;

	}

	public String getTimestamp() {

		return timestamp;

	}

}