Add 404 redirect in dashboard

lt77777 · Dec 5, 2021 · 83f2e47 · 83f2e47
1 parent 378cacb
commit 83f2e47
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/src/main/java/com/amigo/control/DashboardController.java b/app/src/main/java/com/amigo/control/DashboardController.java
@@ -17,8 +17,13 @@ public class DashboardController {
 
  @GetMapping("/dashboard")
  public ModelAndView showDashboard(@ModelAttribute User user) {
+ if (!user.isValid() && !presenter.hasUser()) {
+ return new ModelAndView("404");
+ }
+ if (user != null) {
+ presenter.setUser(user);
+ }
  ModelAndView mav = new ModelAndView("dashboard");
- presenter.setUser(user);
  presenter.populate(mav.getModel());
  return mav;
  }