Subversion Repositories SmartDukaan

package in.shop2020.web;

import in.shop2020.model.ComparisonStats;

import in.shop2020.server.ComparisonStatsRepository;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Calendar;

import java.util.Date;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.json.JSONObject;

import com.google.appengine.api.datastore.DatastoreService;

import com.google.appengine.api.datastore.DatastoreServiceFactory;

import com.google.appengine.api.datastore.Entity;

import com.google.appengine.api.datastore.FetchOptions;

import com.google.appengine.api.datastore.PreparedQuery;

import com.google.appengine.api.datastore.Query;

import com.google.appengine.api.datastore.Text;

public class DailyComparisonStatsGenerator extends HttpServlet {

    private static final long serialVersionUID = -8236918415987438049L;

    public void doPost(HttpServletRequest req, HttpServletResponse resp) {

        resp.setContentType("application/json");

        DatastoreService datastore = DatastoreServiceFactory

                .getDatastoreService();

        Calendar cal = Calendar.getInstance();

        Date toDate =  cal.getTime();

        cal.add(Calendar.MONTH, -1);

        Date fromDate = cal.getTime();

        Query q = new Query("DataLog");

        q.addFilter("eventType", Query.FilterOperator.EQUAL, "PRODUCT_COMPARE");

        q.addFilter("date", Query.FilterOperator.GREATER_THAN_OR_EQUAL, fromDate);

        q.addFilter("date", Query.FilterOperator.LESS_THAN_OR_EQUAL, toDate);

        PreparedQuery pq = datastore.prepare(q);

        Map<Long, Map<Long, Long>> comparisonStats = new HashMap<Long, Map<Long, Long>>();

        List<Entity> results = new ArrayList<Entity>();

        for (Entity result : pq.asIterable(FetchOptions.Builder.withChunkSize(500))) {

            results.add(result);

        }

        for (Entity result : results) {

        	List<Long> itemIds;

        	try{

        		itemIds =  (List<Long>) result.getProperty("itemIds");

        	}catch (ClassCastException ce) {

				System.out.println("Unable to cast itemIds to List");

				continue;

			}

            if(itemIds != null && !itemIds.isEmpty()){

            	for(int i=0; i<itemIds.size(); i++){

            		for(int j=0; j<i; j++){

            			Long itemOne = itemIds.get(i);

            			Long itemTwo = itemIds.get(j);

            			Map<Long, Long> comparedItems;

            			comparedItems = comparisonStats.get(itemOne);

            			if(comparedItems==null){

            				comparedItems = new HashMap<Long, Long>();

            				comparisonStats.put(itemOne, comparedItems);

            			}

            			if(comparedItems.containsKey(itemTwo)){

            				comparedItems.put(itemTwo, comparedItems.get(itemTwo)+1);

            			}else{

            				comparedItems.put(itemTwo, 1L);

            			}

            			comparedItems = comparisonStats.get(itemTwo);

            			if(comparedItems==null){

            				comparedItems = new HashMap<Long, Long>();

            				comparisonStats.put(itemTwo, comparedItems);

            			}

            			if(comparedItems.containsKey(itemOne)){

            				comparedItems.put(itemOne, comparedItems.get(itemOne)+1);

            			}else{

            				comparedItems.put(itemOne, 1L);

            			}

            		}

            	}

            }

        }

        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");

        String id = sdf.format(toDate);  

        JSONObject comparisonStatsJSON = new JSONObject(comparisonStats);

        ComparisonStats compStats = new ComparisonStats();

        compStats.setId(id);

        compStats.setComparisonStatsJSON(new Text(comparisonStatsJSON.toString()));

        ComparisonStatsRepository repo = new ComparisonStatsRepository();

        repo.store(compStats);

    }

    public void doGet(HttpServletRequest req, HttpServletResponse resp) {

        doPost(req, resp);

    }

}