Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

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

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

import in.shop2020.serving.utils.FileUtils;

import in.shop2020.serving.utils.Utils;

import in.shop2020.thrift.clients.CatalogServiceClient;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.apache.log4j.Logger;

import org.apache.struts2.convention.annotation.Action;

import org.apache.thrift.TException;

/**

 * 

 * @author rajveer

 *

 */

@SuppressWarnings("serial")

public class HomeController extends BaseController {

	private static Logger logger = Logger.getLogger(Class.class);

	private List<String> bestDealSnippets = null;

	private List<String> bestSellerSnippets = null;

	private List<String> latestArrivalSnippets = null;

	public HomeController(){

		super();

	}

	@Action("/")

    public String index() throws SecurityException, IOException {

    	logger.info("userinfo:" + userinfo.toString());

		List<Long> bestDealCatalogIds = null;

		List<Long> bestSellerCatalogIds = null;

 		List<Long> latestArrivalCatalogIds = null;

 		int bestSellerCount = 4;

 		int latestArrivalCount = 4;

		try {

			CatalogServiceClient catalogServiceClient = new CatalogServiceClient();

			Client client = catalogServiceClient.getClient();

			//Get top 4 best deals 

			bestDealCatalogIds = client.getBestDealsCatalogIds(0, 4, null, -1);

			//Get top 8 best deals b'coz 4 of them may overlap with best deals.

			bestSellerCatalogIds = client.getBestSellersCatalogIds(0, 8, null, -1);

			bestSellerCatalogIds.removeAll(bestDealCatalogIds);

			if(bestSellerCatalogIds.size() < bestSellerCount)

				bestSellerCount = bestSellerCatalogIds.size();

			//Get top 12 latest arrivals b'coz 4 of them may overlap with best deals

			//while another 4 may overlap with best sellers.

			latestArrivalCatalogIds = client.getLatestArrivalsCatalogIds(0, 12, null, 10003);

			latestArrivalCatalogIds.removeAll(bestDealCatalogIds);

			latestArrivalCatalogIds.removeAll(bestSellerCatalogIds.subList(0, bestSellerCount)); //We're only considering the first 4 best sellers for removal.

			if(latestArrivalCatalogIds.size() < latestArrivalCount)

				latestArrivalCount = latestArrivalCatalogIds.size();

		} catch (InventoryServiceException e) {

			logger.error("Error while fetching data from the catalog service", e);

		} catch (TException e) {

			logger.error("Error while fetching data from the catalog service", e);

		} catch (Exception e) {

			logger.error("Unexpected exception", e);

		}

		bestDealSnippets = getSnippets(bestDealCatalogIds);

		bestSellerSnippets = getSnippets(bestSellerCatalogIds.subList(0, bestSellerCount));

		latestArrivalSnippets = getSnippets(latestArrivalCatalogIds).subList(0, latestArrivalCount);

    	htmlSnippets.put("MAIN_BANNER", pageLoader.getMainBannerHtml());

    	return "index";

    }

	/**

	 * Returns a list of snippets to be used on the home page corresponding to

	 * the given catalog ids. The snippets are in the same order as the ids in

	 * the input list. In case a snippet is not found for an entity, this method

	 * simply logs the problem and moves on.

	 * 

	 * @param catalogIds

	 *            Ids of the entities which we want to show.

	 * @return The snippet corresponding to each catalog entity

	 */

	private List<String> getSnippets(List<Long> catalogIds) {

		List<String> snippets = new ArrayList<String>();

		if(catalogIds == null)

			return snippets;

		for(Long item: catalogIds){

			try{

				snippets.add(FileUtils.read( Utils.EXPORT_ENTITIES_PATH + item + File.separator +"HomeSnippet.html"));

			}catch(IOException ioex){

				logger.error("Unable to get home page snippet for " + item, ioex);

			}

		}

		return snippets;

	}

	public String getMainBannerSnippet(){

		return htmlSnippets.get("MAIN_BANNER");

	}

	public List<String> getBestDealSnippets(){

		return bestDealSnippets; 

	}

	public List<String> getBestSellerSnippets(){

		return bestSellerSnippets; 

	}

	public List<String> getLatestArrivalSnippets(){

		return latestArrivalSnippets;

	}

}