if not user model retutn UNAUTHORIZED

jkl09 · Jan 27, 2016 · 21b63af · 21b63af
1 parent 061bd9a
commit 21b63af
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/resty-security/src/main/java/cn/dreampie/security/Principal.java b/resty-security/src/main/java/cn/dreampie/security/Principal.java
@@ -2,6 +2,8 @@
 
 
 import cn.dreampie.common.entity.Entity;
+import cn.dreampie.common.http.exception.WebException;
+import cn.dreampie.common.http.result.HttpStatus;
 
 import java.io.Serializable;
 import java.util.Set;
@@ -51,6 +53,10 @@ public boolean hasCredential(String permission) {
   }
 
   public M getModel() {
-    return model;
+    if (model == null) {
+      throw new WebException(HttpStatus.UNAUTHORIZED, "Could not found any user for this session.");
+    } else {
+      return model;
+    }
   }
 }