test

1
package servlets;
2

3
import java.io.IOException;
4
import java.io.PrintWriter;
5

6
import javax.servlet.RequestDispatcher;
7
import javax.servlet.ServletException;
8
import javax.servlet.http.HttpServlet;
9
import javax.servlet.http.HttpServletRequest;
10
import javax.servlet.http.HttpServletResponse;
11

12
import com.bittercode.constant.BookStoreConstants;
13
import com.bittercode.constant.db.UsersDBConstants;
14
import com.bittercode.model.User;
15
import com.bittercode.model.UserRole;
16
import com.bittercode.service.UserService;
17
import com.bittercode.service.impl.UserServiceImpl;
18

19
public class CustomerLoginServlet extends HttpServlet {
20

21
    UserService authService = new UserServiceImpl();
22

23
    public void doPost(HttpServletRequest req, HttpServletResponse res) throws IOException, ServletException {
24
        PrintWriter pw = res.getWriter();
25
        res.setContentType(BookStoreConstants.CONTENT_TYPE_TEXT_HTML);
26
        String uName = req.getParameter(UsersDBConstants.COLUMN_USERNAME);
27
        String pWord = req.getParameter(UsersDBConstants.COLUMN_PASSWORD);
28
        User user = authService.login(UserRole.CUSTOMER, uName, pWord, req.getSession());
29

30
        try {
31

32
            if (user != null) {
33

34
                RequestDispatcher rd = req.getRequestDispatcher("CustomerHome.html");
35
                rd.include(req, res);
36
                pw.println("    <div id=\"topmid\"><h1>Welcome to Online <br>Book Store</h1></div>\r\n"
37
                        + "    <br>\r\n"
38
                        + "    <table class=\"tab\">\r\n"
39
                        + "        <tr>\r\n"
40
                        + "            <td><p>Welcome "+user.getFirstName()+", Happy Learning !!</p></td>\r\n"
41
                        + "        </tr>\r\n"
42
                        + "    </table>");
43

44
            } else {
45

46
                RequestDispatcher rd = req.getRequestDispatcher("CustomerLogin.html");
47
                rd.include(req, res);
48
                pw.println("<table class=\"tab\"><tr><td>Incorrect UserName or PassWord</td></tr></table>");
49
            }
50

51
        } catch (Exception e) {
52
            e.printStackTrace();
53
        }
54
    }
55

56
}