Thúy- fix bug 154 (Completed)

hardingadonis · Mar 6, 2024 · e000fc5 · e000fc5
1 parent 5483afb
commit e000fc5
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/main/java/io/hardingadonis/saledock/controller/management/customer/CustomerServlet.java b/src/main/java/io/hardingadonis/saledock/controller/management/customer/CustomerServlet.java
@@ -21,9 +21,18 @@ protected void doGet(HttpServletRequest request, HttpServletResponse response)
         int pageSize = 10;
 
         // Get the requested page number
-        String pageStr = request.getParameter("page");
-        if (pageStr != null) {
-            pageNumber = Integer.parseInt(pageStr);
+//        String pageStr = request.getParameter("page");
+//        if (pageStr != null) {
+//            pageNumber = Integer.parseInt(pageStr);
+//        }
+
+        if (request.getParameter("page") != null) {
+            try {
+                pageNumber = Integer.parseInt(request.getParameter("page"));
+            } catch (Exception e) {
+                response.sendRedirect("./error-404");
+                return;
+            }
         }
 
         List<Customer> customers = Singleton.customerDAO.pagination((pageNumber - 1) * pageSize, pageSize);
@@ -38,7 +47,6 @@ protected void doGet(HttpServletRequest request, HttpServletResponse response)
         request.setAttribute("totalPages", totalPages);
         request.setAttribute("pageSize", pageSize);
         request.setAttribute("page", "customer");
-
 
         request.getRequestDispatcher("/view/jsp/management/customer/customer.jsp").forward(request, response);
     }