x

src/main/java/com/jh/boot/security/SecurityHelper.java

-package com.jh.boot.security;
-
-import com.jh.boot.security.model.AppUser;
-
-public class SecurityHelper {
-
-	private SecurityHelper() {
-	}
-
-	public boolean hasRole(AppUser user, String...roles) {
-		return true;
-	}
-	
-	
-}
+package com.jh.boot.security;
+
+import java.util.Set;
+import java.util.stream.Collectors;
+
+import com.jh.boot.security.model.AppRole;
+import com.jh.boot.security.model.AppUser;
+
+public class SecurityHelper {
+
+	public static final String ADMIN_ROLE = "ROLE_ADMIM";
+	
+	private SecurityHelper() {
+	}
+
+	public boolean hasAnyRole(AppUser user, String...roles) {
+		if (user == null || user.getRoles() == null || roles == null || roles.length == 0) {
+			return false;
+		}
+		
+		Set<String> set = user.getRoles().stream().map(AppRole::getName).collect(Collectors.toSet());
+		if (set.contains(ADMIN_ROLE)) {
+			return true;
+		}
+		
+		for (String role : roles) { 
+			if (set.contains(role)) {
+				return true;
+			}
+		}
+		
+		return false;
+	}
+	
+	
+}