[enhancement] JCommon MongoDB UserController set default role at login

XiaoMi · Jul 2, 2024 · beeae95 · beeae95
1 parent 7ab5014
commit beeae95
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...docean-plugin/docean-plugin-mongodb/src/main/java/run/mone/controller/UserController.java b/...docean-plugin/docean-plugin-mongodb/src/main/java/run/mone/controller/UserController.java
@@ -4,6 +4,7 @@
 import com.xiaomi.youpin.docean.anno.ModelAttribute;
 import com.xiaomi.youpin.docean.anno.RequestMapping;
 import com.xiaomi.youpin.docean.anno.RequestParam;
+import com.xiaomi.youpin.docean.common.StringUtils;
 import com.xiaomi.youpin.docean.mvc.ContextHolder;
 import com.xiaomi.youpin.docean.mvc.MvcContext;
 import run.mone.bo.User;
@@ -104,6 +105,9 @@ public String login(User userReq) {
             return "error";
         }
         user.setPassword("");
+        if (StringUtils.isEmpty(user.getRole())) {
+            user.setRole("user");
+        }
         context.session().setAttribute("user", user);
         return "ok";
     }