ATLAS-2906: Allow transforms to be applied when entity-level transforms are present.

intg/src/main/java/org/apache/atlas/entitytransform/HiveStorageDescriptorEntityHandler.java

@@ -26,7 +26,7 @@ import java.util.List;
 import static org.apache.atlas.entitytransform.TransformationConstants.*;
 
 public class HiveStorageDescriptorEntityHandler extends BaseEntityHandler {
-    static final List<String> CUSTOM_TRANSFORM_ATTRIBUTES = Arrays.asList(HIVE_DB_NAME_ATTRIBUTE, HIVE_TABLE_NAME_ATTRIBUTE, HIVE_DB_CLUSTER_NAME_ATTRIBUTE);
+    static final List<String> CUSTOM_TRANSFORM_ATTRIBUTES = Arrays.asList(HIVE_DB_NAME_ATTRIBUTE, HIVE_TABLE_NAME_ATTRIBUTE, HIVE_DB_CLUSTER_NAME_ATTRIBUTE, HIVE_STORAGE_DESC_LOCATION_ATTRIBUTE);
 
 
     public HiveStorageDescriptorEntityHandler(List<AtlasEntityTransformer> transformers) {
@@ -90,6 +90,9 @@ public class HiveStorageDescriptorEntityHandler extends BaseEntityHandler {
 
                 case HIVE_DB_CLUSTER_NAME_ATTRIBUTE:
                     return clusterName;
+
+                case HIVE_STORAGE_DESC_LOCATION_ATTRIBUTE:
+                    return location;
             }
 
             return super.getAttribute(attribute);
@@ -116,6 +119,10 @@ public class HiveStorageDescriptorEntityHandler extends BaseEntityHandler {
                     isCustomAttributeUpdated = true;
                 break;
 
+                case HIVE_STORAGE_DESC_LOCATION_ATTRIBUTE:
+                    location = attributeValue;
+                break;
+
                 default:
                     super.setAttribute(attribute, attributeValue);
                 break;

intg/src/main/java/org/apache/atlas/entitytransform/TransformationConstants.java

@@ -37,6 +37,7 @@ public final class TransformationConstants {
     public static final String HDFS_PATH_NAME_ATTRIBUTE       = "hdfs_path.name";
     public static final String HDFS_PATH_PATH_ATTRIBUTE       = "hdfs_path.path";
     public static final String HDFS_CLUSTER_NAME_ATTRIBUTE    = "hdfs_path.clusterName";
+    public static final String HIVE_STORAGE_DESC_LOCATION_ATTRIBUTE = "hive_storagedesc.location";
 
     public static final char   TYPE_NAME_ATTRIBUTE_NAME_SEP = '.';
     public static final char   CLUSTER_DELIMITER            = '@';