Subversion Repositories SmartDukaan

package in.shop2020.warehouse.util;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

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.mail.MessagingException;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.thrift.transport.TTransportException;

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

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

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

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.InventoryClient;

import in.shop2020.thrift.clients.UserClient;

import in.shop2020.utils.GmailUtils;

public class OosStatusMarker {

	private static Log log = LogFactory.getLog(OosStatusMarker.class);

	private static final Integer OOS_STATUS_CUTOFF_HOUR = 23;

	private static final Integer OOS_STATUS_CUTOFF_MINUTE = 59;

	private static java.text.SimpleDateFormat sdf = new java.text.SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

	private static final String[] tomail = {"aman.kumar@shop2020.in", "himanshu.pandey@shop2020.in", "khushal.bhatia@shop2020.in", "amit.gupta@shop2020.in","vikram.raghav@shop2020.in","kshitij.sood@shop2020.in"};

	//private static final String[] tomail = {"amar.kumar@shop2020.in"};

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

		File file = new File("/home/manish/alerts/oosmarker.txt");

		Calendar oosEvaluationDate = Calendar.getInstance();

		Calendar cartAdditionStartDate = Calendar.getInstance();

		oosEvaluationDate.add(Calendar.DATE, -1);

		oosEvaluationDate.set(oosEvaluationDate.get(Calendar.YEAR), 

				oosEvaluationDate.get(Calendar.MONTH), oosEvaluationDate.get(Calendar.DATE), 

				OOS_STATUS_CUTOFF_HOUR , OOS_STATUS_CUTOFF_MINUTE, 0);

		cartAdditionStartDate.set(oosEvaluationDate.get(Calendar.YEAR), 

				oosEvaluationDate.get(Calendar.MONTH), oosEvaluationDate.get(Calendar.DATE) -1, 

				OOS_STATUS_CUTOFF_HOUR , OOS_STATUS_CUTOFF_MINUTE, 0);

		Map<Long, Boolean> oosStatusMap = new HashMap<Long, Boolean>();

		List<Item> items = null;

		try {

			Client catalogClient = new CatalogClient().getClient();

			items = catalogClient.getAllAliveItems();

		} catch (Exception e) {

			try {

				Client catalogClient = new CatalogClient().getClient();

				items = catalogClient.getAllAliveItems();

			} catch (Exception ex) {

				log.error("Exception while getting all items ", ex);

				sendMailForError("Exception while getting all items \n" + ex);

			}

		}

		System.out.println("P1");

		String failedItemsList = "";

		in.shop2020.model.v1.user.UserContextService.Client client = new UserClient().getClient();

		try {

			List<Long> cartItems = client.getProductsAddedToCart(cartAdditionStartDate.getTimeInMillis(), oosEvaluationDate.getTimeInMillis());

			for(Item item : items) {

				if (item.getItemStatus() == status.ACTIVE) {

					oosStatusMap.put(item.getId(), false);

				} else {

					if(cartItems.contains(item.getId())) {

						oosStatusMap.put(item.getId(), false);

					} else {

						oosStatusMap.put(item.getId(), true);

					}

				}

			}

		} catch (Exception e) {

			log.error("Exception while getting all items list added to cart", e);

			sendMailForError("Exception while getting all items list added to cart \n");

		}

		System.out.println("P2");

		try {

			in.shop2020.model.v1.inventory.InventoryService.Client invClient = new InventoryClient().getClient();

			invClient.addOosStatusForItem(oosStatusMap, oosEvaluationDate.getTimeInMillis());

			System.out.println("P3");

		} catch(Exception e) {

			log.error("Error while adding OOS Status", e);

			sendMailForError("Error while adding OOS Status");

			return;

		}

		BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter(file));

		bufferedWriter.write(StringUtils.join(new String[] {"OosStatusUpdationDate",sdf.format(new Date())}, '\t'));

		bufferedWriter.close();

		sendMailForSuccess(failedItemsList);

	}

	private static void sendMailForError(String errorMessage) {

        GmailUtils g = new GmailUtils();

        try {

	        g.sendSSLMessage(tomail, "Error while Marking OutOfStock Statuses for Items", errorMessage, "cnc.center@shop2020.in", "5h0p2o2o", new ArrayList<File>());

        } catch(MessagingException mex) {

        	try {

        		g.sendSSLMessage(tomail, "Error while Marking OutOfStock Statuses for Items", errorMessage, "cnc.center@shop2020.in", "5h0p2o2o", new ArrayList<File>());

        	} catch (Exception e) {

        		//TODO Logging

        	}

        }

	}

	private static void sendMailForSuccess(String message) {

        GmailUtils g = new GmailUtils();

        try {

	        g.sendSSLMessage(tomail, "Marked OutOfStock Statuses ", message, "cnc.center@shop2020.in", "5h0p2o2o", new ArrayList<File>());

        } catch(MessagingException mex) {

        		//TODO Logging

        }

	}

}