Browse Source

feat(user):user返回前端信息增加管理员判断

Hu Jie 7 years ago
parent
commit
ec9240f5ed

+ 20 - 0
src/main/java/com/uas/platform/b2c/common/account/model/UserInfo.java

@@ -74,6 +74,16 @@ public class UserInfo {
 		this.secLevel = countSecLevel();
 		this.emailValidCode = user.getEmailValidCode();
 		this.imageUrl = user.getImageUrl();
+		user.setCurrentEnterpriseRoles();
+		Set<Role> currentRoles = user.getRoles();
+		if (!CollectionUtils.isEmpty(currentRoles)) {
+			currentRoles.forEach( current -> {
+				if ("管理员".equals(current.getDesc())) {
+					this.admin = true;
+				}
+			});
+		}
+		this.admin = this.admin == null ? false : this.admin;
 	}
 
 	private Short countSecLevel(){
@@ -109,6 +119,8 @@ public class UserInfo {
 		return new PageImpl<UserInfo>(infos);
 	}
 
+	private Boolean admin;
+
 	private Long userUU;
 
 	private String userName;
@@ -141,6 +153,14 @@ public class UserInfo {
 
 	private String imageUrl;
 
+	public Boolean isAdmin() {
+		return admin;
+	}
+
+	public void setAdmin(Boolean admin) {
+		this.admin = admin;
+	}
+
 	public Integer getEmailValidCode() {
 		return emailValidCode;
 	}