x

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

 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 hasRole(AppUser user, String...roles) {
+	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;
+	}
 	
 	
 }