Add get_identifier to hive source in metadata ingestion (#2667)

metadata-ingestion/setup.py

@@ -145,6 +145,7 @@ base_dev_requirements = {
             "ldap",
             "looker",
             "glue",
+            "hive",
             "datahub-kafka",
             "datahub-rest",
             # airflow is added below

metadata-ingestion/src/datahub/ingestion/source/hive.py

@@ -22,6 +22,12 @@ class HiveConfig(BasicSQLAlchemyConfig):
     # defaults
     scheme = "hive"
 
+    def get_identifier(self, schema: str, table: str) -> str:
+        regular = f"{schema}.{table}"
+        if self.database:
+            return f"{self.database}.{regular}"
+        return regular
+
 
 class HiveSource(SQLAlchemySource):
     def __init__(self, config, ctx):

metadata-ingestion/tests/unit/test_hive_source.py

@@ -0,0 +1,23 @@
+import unittest
+
+from datahub.ingestion.source.hive import HiveConfig
+
+
+class HiveSinkTest(unittest.TestCase):
+    def test_hive_configuration_get_indentifier_with_database(self):
+        test_db_name = "test_database"
+        test_schema_name = "test_schema"
+        test_table_name = "test_table"
+        config_dict = {
+            "username": "test",
+            "password": "test",
+            "host_port": "test:80",
+            "database": test_db_name,
+            "scheme": "hive+https",
+        }
+        hive_config = HiveConfig.parse_obj(config_dict)
+        expected_output = f"{test_db_name}.{test_schema_name}.{test_table_name}"
+        output = hive_config.get_identifier(
+            schema=test_schema_name, table=test_table_name
+        )
+        assert output == expected_output