diff --git a/src/main/java/demo/security/servlet/UserServlet.java b/src/main/java/demo/security/servlet/UserServlet.java index 2735381..66317ed 100644 --- a/src/main/java/demo/security/servlet/UserServlet.java +++ b/src/main/java/demo/security/servlet/UserServlet.java @@ -17,10 +17,7 @@ public class UserServlet extends HttpServlet { @Override protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { - SessionHeader sessionHeader = getSessionHeader(request); - if (sessionHeader == null) return; - String user = sessionHeader.getUsername(); - String query = "SELECT userid FROM users WHERE username = '" + user + "'"; + String user = request.getParameter("username"); try { DBUtils db = new DBUtils(); List users = db.findUsers(user); @@ -52,6 +49,20 @@ private SessionHeader getSessionHeader(HttpServletRequest request) { @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { - + SessionHeader sessionHeader = getSessionHeader(request); + if (sessionHeader == null) return; + String user = sessionHeader.getUsername(); + try { + DBUtils db = new DBUtils(); + List users = db.findUsers(user); + response.setContentType("text/html"); + PrintWriter out = response.getWriter(); + users.forEach((result) -> { + out.print("

User "+result+ "

"); + }); + out.close(); + } catch (Exception e) { + throw new RuntimeException(e); + } } }