com.AlertMailerWebPage.servlet.GetMonthlyServlet.java Source code

Java tutorial

Introduction

Here is the source code for com.AlertMailerWebPage.servlet.GetMonthlyServlet.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.AlertMailerWebPage.servlet;

import com.mongodb.BasicDBObject;
import com.mongodb.Block;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoClient;
import com.mongodb.ServerAddress;
import com.mongodb.client.AggregateIterable;
import java.awt.List;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import static java.util.Arrays.asList;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Properties;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.bson.Document;
import org.json.JSONArray;
import org.json.JSONObject;

/**
 *
 * @author prakharagarwal
 */
public class GetMonthlyServlet extends HttpServlet {

    Map<Long, Integer> prsMap = null;
    Map<Long, Integer> psmMap = null;
    Map<Long, Integer> rentalMap = null;

    int val = 0;
    int i = 0;
    String type = null;

    /**
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code>
     * methods.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        response.setContentType("text/plain;charset=UTF-8");

        try (PrintWriter out = response.getWriter()) {
            prsMap = getMapper();
            psmMap = getMapper();
            rentalMap = getMapper();
            Properties defaultProps = new Properties();
            InputStream in = GetProcessSpeed.class.getResourceAsStream("configuration.properties");
            defaultProps.load(in);
            in.close();
            String database = (String) defaultProps.get("database_requirementLog");
            String collection = (String) defaultProps.get("collection_requirementLog");
            String serverAddress = (String) defaultProps.get("IP_mbstats");

            String initial = request.getParameter("initial");
            //if(initial.equals("yes"))
            //{
            final JSONArray counts1 = new JSONArray();

            final JSONArray counts2 = new JSONArray();
            final JSONArray counts3 = new JSONArray();

            final JSONArray labels = new JSONArray();

            JSONObject counts = new JSONObject();
            AggregateIterable<Document> countsPRS = getCounts("PRS", database, collection, serverAddress);

            Calendar labeldate = Calendar.getInstance();

            labeldate.add(Calendar.DATE, -29);
            for (i = 0; i <= 29; i++) {
                Date date = labeldate.getTime();
                SimpleDateFormat format1 = new SimpleDateFormat("dd MMMM");
                String date1 = format1.format(date);
                labels.put(date1);
                labeldate.add(Calendar.DATE, 1);
            }
            countsPRS.forEach(new Block<Document>() {
                @Override
                public void apply(final Document document) {
                    Document doc1 = (Document) document.get("_id");
                    long date = (long) doc1.get("date");
                    type = (String) doc1.get("type");
                    val = (int) document.get("count");
                    if (type.equalsIgnoreCase("PRS")) {
                        prsMap.put(date, val);
                    } else if (type.equalsIgnoreCase("PSM")) {
                        psmMap.put(date, val);
                    } else if (type.equalsIgnoreCase("RENTAL")) {
                        rentalMap.put(date, val);

                    }

                }

            });

            //   System.out.println("PRS MAP "+prsMap.toString());
            // System.out.println("Psm MAP "+psmMap.toString());
            //System.out.println("rental MAP "+rentalMap.toString());

            for (Map.Entry<Long, Integer> map : prsMap.entrySet()) {
                counts1.put(map.getValue());
            }

            for (Map.Entry<Long, Integer> map : psmMap.entrySet()) {
                counts2.put(map.getValue());
            }
            for (Map.Entry<Long, Integer> map : rentalMap.entrySet()) {
                counts3.put(map.getValue());
            }

            //---------------------------------------------------------------

            try {
                counts.put("countsPRS", counts1);

                counts.put("countsPSM", counts2);
                counts.put("countsRENTAL", counts3);

                counts.put("labels", labels);

            } catch (Exception e) {

            }
            response.setContentType("text/plain");
            response.setCharacterEncoding("UTF-8");
            response.getWriter().write(counts.toString());
        }
    }

    // }
    private Map<Long, Integer> getMapper() {
        Calendar current = Calendar.getInstance();
        /* current.set(Calendar.HOUR_OF_DAY,0);
         current.set(Calendar.MINUTE,0);
         current.set(Calendar.SECOND,0);
         current.set(Calendar.MILLISECOND,0);
         current.add(Calendar.DATE, -29);*/
        current.set(2016, 6, 1, 0, 0, 0);
        current.set(Calendar.MILLISECOND, 0);

        Map<Long, Integer> map = new LinkedHashMap<Long, Integer>();
        for (int i = 0; i <= 29; i++) {
            long millis = current.getTimeInMillis();
            map.put(millis, 0);

            current.add(Calendar.DATE, 1);
        }
        // System.out.println(map.toString());
        return map;
    }

    public AggregateIterable<Document> getCounts(String alertType, String database, String collection,
            String serverAddress) {

        Calendar current = Calendar.getInstance();
        Calendar min_13 = Calendar.getInstance();

        min_13.set(2016, 6, 1, 0, 0, 0);
        current.set(2016, 6, 1, 23, 59, 59);
        current.add(Calendar.DATE, 29);
        //  min_13.add(Calendar.DATE,-29);
        int y = min_13.get(Calendar.YEAR);
        int m = min_13.get(Calendar.MONTH);
        int d = min_13.get(Calendar.DAY_OF_MONTH);
        min_13.set(y, m, d, 0, 0, 0);
        min_13.set(Calendar.MILLISECOND, 0);
        //MongoClient mongoClient = new MongoClient(new ServerAddress(serverAddress,27017));
        MongoClient mongoClient = new MongoClient();
        AggregateIterable<Document> iterable_6_9_p = mongoClient.getDatabase(database).getCollection(collection)
                .aggregate(asList(
                        new Document("$match",
                                new Document("sentDate",
                                        new Document("$gte", min_13.getTimeInMillis()).append("$lte",
                                                current.getTimeInMillis()))
                                                        //.append("isMailSent","Y")                
                                                        .append("isMailSent", "Y")),
                        new Document("$group",
                                new Document("_id", new Document("date", "$sentDate").append("type", "$alertType"))
                                        .append("count", new Document("$sum", 1)))));

        return iterable_6_9_p;
    }

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /**
     * Handles the HTTP <code>GET</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Handles the HTTP <code>POST</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Returns a short description of the servlet.
     *
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>

}