Skip to content

Commit 50d37fc

Browse files
committed
add if in policy
1 parent fb106bb commit 50d37fc

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

auth-agents-common/src/main/java/org/apache/atlas/policytransformer/CachePolicyTransformerImpl.java

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -334,10 +334,27 @@ private void setPolicyItems(RangerPolicy rangerPolicy, AtlasEntityHeader atlasPo
334334

335335
List<String> users = (List<String>) atlasPolicy.getAttribute("policyUsers");
336336
List<String> groups = (List<String>) atlasPolicy.getAttribute("policyGroups");
337-
List<String> roles = (List<String>) atlasPolicy.getAttribute("policyRoles");
337+
338+
List<String> roles = null;
339+
if (atlasPolicy.getAttribute("policyRoles") instanceof String) {
340+
String policyRoles = (String) atlasPolicy.getAttribute("policyRoles");
341+
LOG.info("policyRoles", policyRoles);
342+
roles = new ArrayList<>(Arrays.asList(policyRoles));
343+
} else {
344+
roles = (List<String>) atlasPolicy.getAttribute("policyRoles");
345+
}
346+
338347

339348
List<RangerPolicyItemAccess> accesses = new ArrayList<>();
340-
List<String> actions = (List<String>) atlasPolicy.getAttribute("policyActions");
349+
List<String> actions = null;
350+
351+
if (atlasPolicy.getAttribute("policyActions") instanceof String) {
352+
String policyActions = (String) atlasPolicy.getAttribute("policyActions");
353+
LOG.info("policyActions", policyActions);
354+
actions = new ArrayList<>(Arrays.asList(policyActions));
355+
} else {
356+
actions = (List<String>) atlasPolicy.getAttribute("policyActions");
357+
}
341358

342359
actions.forEach(action -> accesses.add(new RangerPolicyItemAccess(action)));
343360

0 commit comments

Comments
 (0)