Subversion Repositories SmartDukaan

package in.shop2020.utils;

import in.shop2020.model.v1.order.FlipkartFaSalesSnapshot;

import java.io.FileReader;

import java.net.UnknownHostException;

import java.util.ArrayList;

import java.util.Calendar;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.json.JSONException;

import au.com.bytecode.opencsv.CSVReader;

import com.google.gson.Gson;

import com.mongodb.BasicDBObject;

import com.mongodb.DB;

import com.mongodb.DBCollection;

import com.mongodb.DBCursor;

import com.mongodb.DBObject;

import com.mongodb.MongoClient;

import com.mongodb.util.JSON;

public class FlipkartConsumer{

	private static MongoClient mongo;

    private static final String FLIPKART_SALES_SNAPSHOT = "FlipkartSaleSnapshot";

    private static final String FLIPKART_DAILY_SALE_SNAPSHOT = "FlipkartDailySaleSnapshot";

    static {

        try {

            mongo = new MongoClient( "localhost" , 27017 );

        } catch (UnknownHostException e) {

            e.printStackTrace();

        }

    }

    public static boolean bulkAddOrUpdateFlipkartFbaSalesSnapshot(List<FlipkartFaSalesSnapshot> sales) throws JSONException{

    	DB db = mongo.getDB(FLIPKART_SALES_SNAPSHOT);

        DBCollection collection = db.getCollection(FLIPKART_DAILY_SALE_SNAPSHOT);

        Gson gs = new Gson();

        for (FlipkartFaSalesSnapshot sale :sales){

        	DBObject dbObject = (DBObject) JSON.parse(gs.toJson(sale));

        	dbObject.put("_id", generateId(sale.getDateOfSale(), sale.getItem_id(), sale.getFcLocation().getValue()));

        	BasicDBObject searchObject = new BasicDBObject();

        	searchObject.put("_id",generateId(sale.getDateOfSale(), sale.getItem_id(), sale.getFcLocation().getValue()));

        	DBCursor cursor = collection.find(searchObject);

        	if (cursor.count() > 0){

        		BasicDBObject updateObject = new BasicDBObject();

        		updateObject.put("totalSale",sale.getTotalSale());

                updateObject.put("totalOrderCount",sale.getTotalOrderCount());

                if (sale.getTotalOrderCount() > 0){

                    updateObject.put("isOutOfStock",false);

                }

                collection.update(searchObject, new BasicDBObject("$set", updateObject));

        	}

        	else{

                collection.insert(dbObject);

            }

        }

        collection.createIndex((DBObject) JSON.parse("{'dateOfSale':1}"));

        return true;

    }

    public static List<FlipkartFaSalesSnapshot> getFlipkartFbaSalesSnapshotForDays(int interval){

    	DB db = mongo.getDB(FLIPKART_SALES_SNAPSHOT);

        DBCollection collection = db.getCollection(FLIPKART_DAILY_SALE_SNAPSHOT);

        Calendar cal = Calendar.getInstance();

        cal.add(Calendar.DATE, -1);cal.set(Calendar.HOUR_OF_DAY, 00);cal.set(Calendar.MINUTE,00);cal.set(Calendar.SECOND,00);cal.set(Calendar.MILLISECOND,00);

        long toDate = cal.getTimeInMillis();

        cal.add(Calendar.DATE, -interval);

        long fromDate = cal.getTimeInMillis();

        BasicDBObject query = new BasicDBObject();

        BasicDBObject obj = new BasicDBObject();

        obj.put("$gte", fromDate);

        obj.put("$lte", toDate);

        query.put("dateOfSale",obj);

        System.out.println(fromDate +" "+ toDate);

        BasicDBObject orderBy = new BasicDBObject();

        orderBy.put("dateOfSale", 1);

        Gson gson = new Gson();

        List<FlipkartFaSalesSnapshot> salesSnapshot = new ArrayList<FlipkartFaSalesSnapshot>();

        DBCursor cursor = collection.find(query).sort(orderBy);

        while (cursor.hasNext()) {

        	FlipkartFaSalesSnapshot salesObj = gson.fromJson(cursor.next().toString(), FlipkartFaSalesSnapshot.class);

        	salesSnapshot.add(salesObj);

        }

        return salesSnapshot;

    }

    public static FlipkartFaSalesSnapshot getFlipkartFaSalesLatestSnapshotForItem(long item_id){

    	FlipkartFaSalesSnapshot salesObj = new FlipkartFaSalesSnapshot();

    	DB db = mongo.getDB(FLIPKART_SALES_SNAPSHOT);

        DBCollection collection = db.getCollection(FLIPKART_DAILY_SALE_SNAPSHOT);

        BasicDBObject andQuery = new BasicDBObject();

        List<BasicDBObject> obj = new ArrayList<BasicDBObject>();

        obj.add(new BasicDBObject("item_id", item_id));

        Calendar cal = Calendar.getInstance();

        cal.add(Calendar.DATE, -1);cal.set(Calendar.HOUR_OF_DAY, 00);cal.set(Calendar.MINUTE,00);cal.set(Calendar.SECOND,00);cal.set(Calendar.MILLISECOND,00);

        obj.add(new BasicDBObject("dateOfSale", cal.getTimeInMillis()));

        andQuery.put("$and", obj);

        BasicDBObject orderBy = new BasicDBObject();

        orderBy.put("dateOfSale", -1);

        DBCursor cursor = collection.find(andQuery).sort(orderBy).limit(1);

        Gson gson = new Gson();

        while (cursor.hasNext()) {

            salesObj = gson.fromJson(cursor.next().toString(), FlipkartFaSalesSnapshot.class);

        }

        return salesObj;

    }

    private static long generateId(long dateOfSale, long item_id, int warehouseLocation){

        return Long.valueOf(String.valueOf(dateOfSale)+String.valueOf(item_id)+String.valueOf(warehouseLocation));

    }

    public static Map<Long, Long> getCurrentFAInventory(){

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

    	try {

    		int i=1;

			CSVReader csvInventoryReader = new CSVReader(new FileReader("/home/FAInventory.csv"));

			String[] values;

			while ((values = csvInventoryReader.readNext()) != null) {

				if(i==1){

					i++;

					continue;

				}

				long itemId = Long.parseLong(values[0]);

				long available = Long.parseLong(values[4]);

				currentFAInventoryMap.put(itemId, available);

				i++;

			}

    	} catch(Exception e){

			e.printStackTrace();

		}

    	return currentFAInventoryMap;

    }

    public static Map<Long, Double> getCurrentFAInventorySellingPrice(){

    	Map<Long,Double> currentFAInventorySellingPriceMap = new HashMap<Long, Double>();

    	try {

    		int i=1;

			CSVReader csvInventoryReader = new CSVReader(new FileReader("/home/FAInventory.csv"));

			String[] values;

			while ((values = csvInventoryReader.readNext()) != null) {

				if(i==1){

					i++;

					continue;

				}

				long itemId = Long.parseLong(values[0]);

				double sellingPrice = Double.parseDouble(values[2]);

				currentFAInventorySellingPriceMap.put(itemId, sellingPrice);

				i++;

			}

    	} catch(Exception e){

			e.printStackTrace();

		}

    	return currentFAInventorySellingPriceMap;

    }

}