commit 47c18b9419f825fd6a15bbf9f22196b4ef49aefe
author Radhika Kundam <rkundam@cloudera.com> Thu Feb 18 16:26:41 2021 -0800
committer Sarath Subramanian <sarath@apache.org> Thu Feb 18 21:57:49 2021 -0800
tree b0bd4784fe66f9823cf6d455be557b9c739ebaca
parent 928d86864b6e4c9244cd77c18c1d3d54d6a6a836

ATLAS-4169: Add Hive Location Path entities to HMS Hook

Signed-off-by: Sarath Subramanian <sarath@apache.org>

addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/AlterDatabase.java

diff --git a/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/AlterDatabase.java b/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/AlterDatabase.java
index 6b01c4e..d2623b3 100644
--- a/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/AlterDatabase.java
+++ b/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/AlterDatabase.java
@@ -61,6 +61,8 @@
             AtlasEntity dbEntity = toDbEntity(newDb);
 
             ret.addEntity(dbEntity);
+
+            addLocationEntities(dbEntity, ret);
         } else {
             LOG.error("AlterDatabase.getEntities(): failed to retrieve db");
         }

addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/CreateDatabase.java

diff --git a/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/CreateDatabase.java b/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/CreateDatabase.java
index 8305a44..bf5f562 100644
--- a/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/CreateDatabase.java
+++ b/addons/hive-bridge/src/main/java/org/apache/atlas/hive/hook/events/CreateDatabase.java
@@ -68,6 +68,8 @@
             AtlasEntity dbEntity = toDbEntity(db);
 
             ret.addEntity(dbEntity);
+
+            addLocationEntities(dbEntity, ret);
         } else {
             LOG.error("CreateDatabase.getEntities(): failed to retrieve db");
         }
@@ -98,10 +100,7 @@
                         ret.addEntity(dbDDLEntity);
                     }
 
-                    AtlasEntity dbLocationEntity = createHiveLocationEntity(dbEntity, ret);
-                    if (dbLocationEntity != null) {
-                        ret.addEntity(dbLocationEntity);
-                    }
+                    addLocationEntities(dbEntity, ret);
                 } else {
                     LOG.error("CreateDatabase.getEntities(): failed to retrieve db");
                 }
@@ -112,4 +111,12 @@
 
         return ret;
     }
+
+    public void addLocationEntities(AtlasEntity dbEntity, AtlasEntitiesWithExtInfo ret) {
+        AtlasEntity dbLocationEntity = createHiveLocationEntity(dbEntity, ret);
+
+        if (dbLocationEntity != null) {
+            ret.addEntity(dbLocationEntity);
+        }
+    }
 }
\ No newline at end of file