Add error redirect and user persistence

lt77777 · Dec 5, 2021 · 65652d3 · 65652d3
1 parent f48583b
commit 65652d3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/app/src/main/java/com/amigo/control/DashboardController.java b/app/src/main/java/com/amigo/control/DashboardController.java
@@ -19,9 +19,9 @@ public class DashboardController {
  @GetMapping("/dashboard")
  public ModelAndView showDashboard(@ModelAttribute User user) {
  if (!user.isValid() && !presenter.hasUser()) {
- return new ModelAndView("404");
+ return new ModelAndView("error-page");
  }
- if (user != null) {
+ if (user.isValid()) {
  presenter.setUser(user);
  }
  ModelAndView mav = new ModelAndView("dashboard");
@@ -48,4 +48,9 @@ public ModelAndView showMatchingScreen() {
  }
  return mav;
  }
+
+ @GetMapping("/*") 
+ public String showError() {
+ return "error-page";
+ }
 }