WebSVN – SmartDukaan – Diff – /trunk/Support/src/main/java/in/shop2020/support/services/RegisteredUsersGenerator.java

 import in.shop2020.model.v1.user.ShoppingCartException;
 import in.shop2020.model.v1.user.User;
 import in.shop2020.model.v1.user.UserContextException;
 import in.shop2020.model.v1.user.UserState;
 import in.shop2020.model.v1.user.UserType;
+import in.shop2020.thrift.clients.CatalogClient;
 import in.shop2020.thrift.clients.TransactionClient;
 import in.shop2020.thrift.clients.UserClient;
 import java.io.ByteArrayOutputStream;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.text.DateFormat;
+import java.text.ParseException;
 import java.text.SimpleDateFormat;
 import java.util.Calendar;
 import java.util.Date;
 import java.util.List;
+import java.util.TimeZone;
 import org.apache.poi.hssf.usermodel.HSSFWorkbook;
 import org.apache.poi.ss.usermodel.Cell;
 import org.apache.poi.ss.usermodel.CellStyle;
 import org.apache.poi.ss.usermodel.Font;
      * otherwise returns null
      * @param startDate -- inclusive
      * @param endDate -- inclusive
      * @return
      */
-    public ByteArrayOutputStream generateRegisteredUsersReport() {
+    public ByteArrayOutputStream generateRegisteredUsersReport(long startDate, long endDate) {
         List<User> users = null;
         try {
-            users = uClient.getAllUsers(UserType.USER, -1, -1);
+            users = uClient.getAllUsers(UserType.USER, startDate, endDate);
         } catch (TException e) {
             logger.error("Unable to get all users", e);
+        }
         if(users == null || users.isEmpty()) {
             return null;
         style.setFont(font);
         CellStyle styleWT = wb.createCellStyle();
         styleWT.setWrapText(true);
+        createRegisteredUserSheet(users, wb, style, styleWT);
+        // Write the workbook to the output stream
+        try {
+            wb.write(baosXLS);
+            baosXLS.close();
+        } catch (IOException e) {
+            logger.error("Unable to write the registered orders report to the byte array", e);
+        }
+        return baosXLS;
+    }
+    private void createRegisteredUserSheet(List<User> users, Workbook wb,
+            CellStyle style, CellStyle styleWT) {
-        Sheet userSheet = wb.createSheet("User");
+        Sheet registeredUserSheet = wb.createSheet("User");
         short userSerialNo = 0;
-        Row titleRow = userSheet.createRow(userSerialNo++);
+        Row titleRow = registeredUserSheet.createRow(userSerialNo++);
         Cell titleCell = titleRow.createCell(0);
         titleCell.setCellValue("Registered Users");
         titleCell.setCellStyle(style);
-        userSheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 6));
+        registeredUserSheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 6));
-        Row headerRow = userSheet.createRow(userSerialNo++);
+        Row headerRow = registeredUserSheet.createRow(userSerialNo++);
         headerRow.createCell(0).setCellValue("User Id");
         headerRow.createCell(1).setCellValue("Email");
         headerRow.createCell(2).setCellValue("Name");
         headerRow.createCell(3).setCellValue("Communication Email");
         headerRow.createCell(4).setCellValue("Date of Birth");
         headerRow.createCell(7).setCellValue("Registered On");
         headerRow.createCell(8).setCellValue("Last Login");
         headerRow.createCell(9).setCellValue("Successful Orders");
         headerRow.createCell(10).setCellValue("Pending/Failed Orders");
         headerRow.createCell(11).setCellValue("Items in Cart");
-        userSheet.setColumnWidth(11, 20 * 256); // set width of items in cart column to 20 chars
+        registeredUserSheet.setColumnWidth(11, 20 * 256); // set width of items in cart column to 20 chars
         Calendar calendar = Calendar.getInstance();
         Row contentRow;
-        float defaultRowHeight = userSheet.getDefaultRowHeightInPoints();
+        float defaultRowHeight = registeredUserSheet.getDefaultRowHeightInPoints();
         UserState uState;
         DateFormat formatter = new SimpleDateFormat("EEE, dd-MMM-yyyy hh:mm a");
         Date currentDate = new Date();
         for (User u : users) {
             userSerialNo++;
-            contentRow = userSheet.createRow(userSerialNo);
+            contentRow = registeredUserSheet.createRow(userSerialNo);
             contentRow.createCell(0).setCellValue(u.getUserId());
             contentRow.createCell(1).setCellValue(u.getEmail());
             contentRow.createCell(2).setCellValue(u.getName());
             contentRow.createCell(3).setCellValue(u.getCommunicationEmail());
             contentRow.createCell(4).setCellValue(u.getDateOfBirth());
+                }
                 contentRow.createCell(9).setCellValue(noOfValidOrders);
                 contentRow.createCell(10).setCellValue(noOfFailedOrders);
             } catch (TransactionServiceException e) {
                 logger.error("Error while getting orders for the customer", e);
+                e.printStackTrace();
             } catch (TException e) {
                 logger.error("Unable to get orders for the customer", e);
+                e.printStackTrace();
+            }
             String itemList = getCartItems(u.getUserId());
             contentRow.createCell(11).setCellValue(itemList);
             contentRow.getCell(11).setCellStyle(styleWT);
                 } else {
                     i++;
                     col = defaultAddrCol + i;
                     headerRow.createCell(col).setCellValue("Address " + i);
+                }
-                userSheet.setColumnWidth(col, addrColWidth * 256);  // set width of address column to 40 characters
+                registeredUserSheet.setColumnWidth(col, addrColWidth * 256);  // set width of address column to 40 characters
                 addressString = generateAddressString(a);
                 contentRow.createCell(col).setCellValue(addressString);
                 contentRow.getCell(col).setCellStyle(styleWT);
                 len = addressString.length();
                 if(len > maxAddrLength)
                 maxAddrLength = len;
+            }
             contentRow.setHeightInPoints((maxAddrLength / addrColWidth + 1) * defaultRowHeight); // Setting Row Height
+        }
         for (int i = 0; i <= 10; i++) {
-            userSheet.autoSizeColumn(i);
+            registeredUserSheet.autoSizeColumn(i);
+        }
-        // Write the workbook to the output stream
-        try {
-            wb.write(baosXLS);
-            baosXLS.close();
-        } catch (IOException e) {
-            logger.error("Unable to write the registered orders report to the byte array", e);
-        }
-        return baosXLS;
+    }
     private String getCartItems(long userId) {
         String itemList = "";
         try {
     public static void main(String[] args) {
         RegisteredUsersGenerator rug = new RegisteredUsersGenerator();
         try {
             String userHome = System.getProperty("user.home");
             FileOutputStream f = new FileOutputStream(userHome + "/registered-users.xls");
+            String startDateStr = "01/11/2011";
+            String endDateStr = "05/11/2011";
+            DateFormat df = new SimpleDateFormat("MM/dd/yyyy");
+            df.setTimeZone(TimeZone.getTimeZone("IST"));
+            Date startDate = null;
+            Date endDate = null;
+            try {
+                startDate = df.parse(startDateStr);
+                endDate = df.parse(endDateStr);
+                //Calendar cal = Calendar.getInstance();
+                //cal.setTime(endDate);
+                //endDate.setTime(cal.getTimeInMillis());
+            } catch (ParseException pe) {
+               System.out.println("Please enter start and end dates in format MM/dd/yyyy");
+            }
-            ByteArrayOutputStream baosXLS = rug.generateRegisteredUsersReport();
+            ByteArrayOutputStream baosXLS = rug.generateRegisteredUsersReport(startDate.getTime(), endDate.getTime());
             if(baosXLS == null) {
                 System.out.println("No data");
                 return;
+            }
             baosXLS.writeTo(f);

Subversion Repositories SmartDukaan

(root)/trunk/Support/src/main/java/in/shop2020/support/services/RegisteredUsersGenerator.java – Rev 3213 → 4817