解决登录问题

notes-pas/pom.xml

@@ -52,7 +52,10 @@
             <artifactId>spring-boot-starter-security</artifactId>
         </dependency>
 
-
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-thymeleaf</artifactId>
+        </dependency>
     </dependencies>
 
 </project>
\ No newline at end of file

notes-pas/src/main/java/com/zjty/tynotes/pas/config/SecurityConfig.java

@@ -3,6 +3,7 @@ package com.zjty.tynotes.pas.config;
 import com.zjty.tynotes.pas.config.handler.MyAccessHandler;
 import com.zjty.tynotes.pas.config.handler.MyFailHandler;
 import com.zjty.tynotes.pas.config.handler.MyLogoutHandler;
+import com.zjty.tynotes.pas.config.handler.MySuccessHandler;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -36,6 +37,8 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter {
     private MyAccessHandler accessHandler;
     @Autowired
     private MyEntryPoint myEntryPoint;
+    @Autowired
+    private MySuccessHandler successHandler;
 
     @Autowired
     private MyUserDetailsServiceImpl myUserDetailsServiceImpl;
@@ -78,6 +81,7 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter {
                 .addFilterAt(myFilter(), UsernamePasswordAuthenticationFilter.class)
                 .addFilterBefore(corsFilter(), ChannelProcessingFilter.class)
                 .csrf().csrfTokenRepository(CookieCsrfTokenRepository.withHttpOnlyFalse()).disable().sessionManagement().maximumSessions(1).expiredUrl("/userLogout").sessionRegistry(sessionRegistry())
+
         ;
     }
 
@@ -90,6 +94,7 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter {
     @Bean
     public MyFilter myFilter() throws Exception {
         MyFilter filter = new MyFilter();
+        filter.setAuthenticationSuccessHandler(successHandler);
         filter.setAuthenticationFailureHandler(failHandler);
         filter.setFilterProcessesUrl("/userLogin");
         filter.setAuthenticationManager(this.authenticationManager());

notes-pas/src/main/java/com/zjty/tynotes/pas/config/handler/MySuccessHandler.java

 package com.zjty.tynotes.pas.config.handler;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
-import com.zjty.tynotes.pas.entity.Authority;
-import com.zjty.tynotes.pas.entity.Role;
-import com.zjty.tynotes.pas.entity.User;
+import com.zjty.tynotes.pas.dao.AuthorityDao;
+import com.zjty.tynotes.pas.dao.RoleAuthorityDao;
+import com.zjty.tynotes.pas.dao.RoleDao;
+import com.zjty.tynotes.pas.dao.UserRoleDao;
+import com.zjty.tynotes.pas.entity.*;
 import com.zjty.tynotes.pas.service.IAuthorityService;
 import com.zjty.tynotes.pas.service.IRoleService;
 import com.zjty.tynotes.pas.service.IUserService;
+import com.zjty.tynotes.pas.task.Init;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
@@ -26,8 +29,11 @@ import javax.servlet.ServletException;
 import javax.servlet.http.Cookie;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import javax.validation.constraints.NotNull;
 import java.io.IOException;
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
 
 /**
  * @author mcj
@@ -42,7 +48,13 @@ public class MySuccessHandler implements AuthenticationSuccessHandler {
     private IAuthorityService iAuthorityService;
 
     @Autowired
-    IRoleService iRoleService;
+    RoleDao roleDao;
+    @Autowired
+    private UserRoleDao userRoleDao;
+    @Autowired
+    private RoleAuthorityDao roleAuthorityDao;
+    @Autowired
+    private AuthorityDao authorityDao;
 
     private BCryptPasswordEncoder bCryptPasswordEncoder = new BCryptPasswordEncoder();
 
@@ -51,6 +63,8 @@ public class MySuccessHandler implements AuthenticationSuccessHandler {
 
     @Autowired
     RedisTemplate redisTemplate;
+    @Autowired
+    private Init init;
 
     @Override
     public void onAuthenticationSuccess(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Authentication authentication) throws IOException, ServletException {
@@ -74,13 +88,32 @@ public class MySuccessHandler implements AuthenticationSuccessHandler {
             }
         }
 
-        redisTemplate.opsForValue().set(user.getUsername(),0);
-
+//        redisTemplate.opsForValue().set(user.getUsername(),0);
+        if(("root").equals(user.getUsername())){
+            user = init.root;
+        }else{
+            List<UserRole> userRoles = userRoleDao.findAllByUserId(user.getId());
+            List<Role> roles = new ArrayList<>();
+            for (UserRole userRole : userRoles) {
+                String roleId = userRole.getRoleId();
+                Optional<Role> op = roleDao.findById(userRole.getRoleId());
+                if(op.isPresent()){
+                    Role role = op.get();
+                    List<RoleAuthority> roleAuthorities = roleAuthorityDao.findAllByRoleId(roleId);
+                    List<String> authorityIds = new ArrayList<>();
+                    for (RoleAuthority roleAuthority : roleAuthorities) {
+                        authorityIds.add(roleAuthority.getAuthorityId());
+                    }
+                    List<Authority> authorities = authorityDao.findAllByIdIn(authorityIds);
+                    role.setAuthorities(authorities);
+                    roles.add(role);
+                }
+            }
+            user.setRoles(roles);
+        }
         sessionRegistry.registerNewSession(value, authentication.getPrincipal());
         httpServletResponse.setStatus(200);
         httpServletResponse.setContentType("application/json; charset=utf-8");
-        httpServletResponse.getWriter().println(new ObjectMapper().writeValueAsString("登陆成功"));
-
-
+        httpServletResponse.getWriter().println(new ObjectMapper().writeValueAsString(user));
     }
 }

notes-pas/src/main/java/com/zjty/tynotes/pas/controller/LoginController.java

@@ -7,6 +7,7 @@ import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.HashMap;
@@ -17,9 +18,9 @@ import java.util.HashMap;
  * @Version 1.0
  */
 @Api(tags = "用户管理权限管理模块",protocols = "http")
-@RestController
 @RequestMapping("/pas/count")
 @AutoDocument
+@Controller
 public class LoginController {
 
     @Autowired
@@ -27,10 +28,16 @@ public class LoginController {
 
     @ApiOperation(value = "登陆失败计数",response = String.class)
     @PostMapping
+    @ResponseBody
     public ResponseEntity login(@RequestBody String username){
         String s = iUserService.addUserCount(username);
         HashMap<Object, Object> map = new HashMap<>(16);
         map.put("msg",s);
         return ResponseEntity.ok(map);
     }
+
+    @GetMapping("/login")
+    public String login2(){
+        return "login";
+    }
 }

notes-union/src/main/resources/application-prod.properties

@@ -18,7 +18,7 @@ spring.data.mongodb.uri=mongodb://localhost:27017/notes
 spring.servlet.multipart.max-file-size=100MB
 spring.servlet.multipart.max-request-size=1000MB
 
-spring.redis.host=localhost
+spring.redis.host=127.0.0.1
 spring.redis.port=6379
 spring.redis.timeout=5000ms
 

notes-union/src/main/resources/application.properties

@@ -13,6 +13,9 @@ server.port=8289
 ## 证书别名
 #server.ssl.keyAlias= alias
 
+
+#mongodb configuration
+spring.data.mongodb.uri=mongodb://localhost:27017/note
 # servlet configuration
 spring.servlet.multipart.max-file-size=100MB
 spring.servlet.multipart.max-request-size=1000MB