Advertisement
Guest User

Untitled

a guest
Jul 15th, 2018
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 1.40 KB | None | 0 0
  1.     private Integer getUserRole(HttpServletRequest req) {
  2.         String token = req.getHeader("cookie-token");
  3.         String loggedUsername = req.getHeader("logged-user");
  4.         if(token == null || loggedUsername == null){
  5.             token = req.getParameter("cookie-token");
  6.             loggedUsername = req.getParameter("logged-user");
  7.             if(token == null || loggedUsername == null)
  8.                 return -1;
  9.         }
  10.        
  11.         HashMap<String, LoggedUser> loggedUsers = getLoggedUsers();
  12.         LoggedUser logged = loggedUsers.get(loggedUsername);
  13.         if(logged == null)
  14.             return -1;
  15.         else {
  16.             if(token.equals(logged.token)){
  17.                 return logged.role;
  18.             }else {
  19.                 return -1;
  20.             }
  21.         }
  22.     }
  23.    
  24.     private HashMap<String, LoggedUser> getLoggedUsers(){
  25.         ServletContext sc = getServletContext();
  26.         HashMap<String, LoggedUser> loggedUsers = (HashMap<String, LoggedUser>)sc.getAttribute("loginList");
  27.         if(loggedUsers == null){
  28.             loggedUsers = new HashMap<String, LoggedUser>();
  29.             sc.setAttribute("loginList", loggedUsers);
  30.         }
  31.         return loggedUsers;
  32.     }
  33.    
  34.     private void storeLoggedUsers(HashMap<String, LoggedUser> loggedUsers){
  35.         ServletContext sc = getServletContext();
  36.         sc.setAttribute("loginList",loggedUsers);
  37.     }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement