Update KeycloakUserStore.java

atlanhq · Jan 15, 2024 · 064982e · 064982e
1 parent 75a99ca
commit 064982e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/auth-agents-common/src/main/java/org/apache/atlas/plugin/util/KeycloakUserStore.java b/auth-agents-common/src/main/java/org/apache/atlas/plugin/util/KeycloakUserStore.java
@@ -155,7 +155,7 @@ public RangerRoles loadRolesIfUpdated(long lastUpdatedTime) throws AtlasBaseExce
         List<UserRepresentation> userRetrievalResult;
 
         do {
-            userRetrievalResult = getHeraclesClient().getUsersMappings(userFrom, userSize, new String[]{"roles", "groups"});
+            userRetrievalResult = getHeraclesClient().getUsersMappings(userFrom, userSize, new String[]{"roles"});
 
             if (!CollectionUtils.isEmpty(userRetrievalResult)) {
                 userRetrievalResult.forEach(user -> {
@@ -177,7 +177,7 @@ public RangerRoles loadRolesIfUpdated(long lastUpdatedTime) throws AtlasBaseExce
         List<HeraclesRoleViewRepresentation> roleRetrievalResult;
 
         do {
-            roleRetrievalResult = getHeraclesClient().getRolesMappings(roleFrom, roleSize, new String[]{"roles", "groups"});
+            roleRetrievalResult = getHeraclesClient().getRolesMappings(roleFrom, roleSize, new String[]{"composite_roles", "groups"});
 
             if (!CollectionUtils.isEmpty(roleRetrievalResult)) {
                 roleRetrievalResult.forEach(role -> {