RANGER-5309: add RangerBasePlugin.getServiceDefHelper()

Author: mneethiraj

agents-common/src/main/java/org/apache/ranger/plugin/model/RangerServiceDef.java

@@ -1417,7 +1417,7 @@ public static class RangerResourceDef implements java.io.Serializable {
         private String              rbKeyValidationMessage;
         private Set<String>         accessTypeRestrictions;
         private Boolean             isValidLeaf;
-        private String              rrnTemplate;
+        private String              rrnTemplate; // resource-name template. Examples: {database}.{table}.{column}, {container}@{storageaccount}/{relativepath}
 
         public RangerResourceDef() {
             this(null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null);

agents-common/src/main/java/org/apache/ranger/plugin/model/validation/RangerServiceDefHelper.java

@@ -363,6 +363,10 @@ public boolean isResourceGraphValid() {
         return delegate.isResourceGraphValid();
     }
 
+    public Set<String> getAllResourceNames() {
+        return delegate.rrnTemplates.keySet();
+    }
+
     public List<String> getOrderedResourceNames(Collection<String> resourceNames) {
         final List<String> ret;
 
@@ -488,10 +492,8 @@ public Delegate(RangerServiceDef serviceDef, boolean checkForCycles) {
                 orderedResourceNames = buildSortedResourceNames();
 
                 for (RangerResourceDef resourceDef : serviceDef.getResources()) {
-                    String rrnTemplate = resourceDef.getRrnTemplate();
-
                     // when rrnTemplate is not specified, create a default one using the full path from root to this resource
-                    if (StringUtils.isBlank(rrnTemplate)) {
+                    if (StringUtils.isBlank(resourceDef.getRrnTemplate())) {
                         List<String> path = new ArrayList<>();
 
                         for (RangerResourceDef resource = resourceDef; resource != null; resource = getResourceDef(resource.getParent(), RangerPolicy.POLICY_TYPE_ACCESS)) {
@@ -500,10 +502,14 @@ public Delegate(RangerServiceDef serviceDef, boolean checkForCycles) {
 
                         Collections.reverse(path);
 
-                        rrnTemplate = "{" + StringUtils.join(path, "}.{") + "}";
+                        String rrnTemplate = "{" + StringUtils.join(path, "}.{") + "}";
+
+                        LOG.debug("Setting rrnTemplate for resource {}.{} to: {}", serviceName, resourceDef.getName(), rrnTemplate);
+
+                        resourceDef.setRrnTemplate(rrnTemplate);
                     }
 
-                    this.rrnTemplates.put(resourceDef.getName(), rrnTemplate);
+                    this.rrnTemplates.put(resourceDef.getName(), resourceDef.getRrnTemplate());
                 }
             } else {
                 orderedResourceNames = new ArrayList<>();

agents-common/src/main/java/org/apache/ranger/plugin/service/RangerBasePlugin.java

@@ -42,6 +42,7 @@
 import org.apache.ranger.plugin.model.RangerPolicy;
 import org.apache.ranger.plugin.model.RangerRole;
 import org.apache.ranger.plugin.model.RangerServiceDef;
+import org.apache.ranger.plugin.model.validation.RangerServiceDefHelper;
 import org.apache.ranger.plugin.policyengine.RangerAccessRequest;
 import org.apache.ranger.plugin.policyengine.RangerAccessRequestImpl;
 import org.apache.ranger.plugin.policyengine.RangerAccessResourceImpl;
@@ -363,6 +364,12 @@ public void setServiceAdmins(Set<String> users) {
         pluginConfig.setServiceAdmins(users);
     }
 
+    public RangerServiceDefHelper getServiceDefHelper() {
+        RangerPolicyEngine policyEngine = this.policyEngine;
+
+        return policyEngine != null ? policyEngine.getServiceDefHelper() : null;
+    }
+
     public RangerServiceDef getServiceDef() {
         RangerPolicyEngine policyEngine = this.policyEngine;