From 745ff15e91fe394ebce6d2ba1a85d51652664e7d Mon Sep 17 00:00:00 2001
From: Mohit Tilala <63147650+mohittilala@users.noreply.github.com>
Date: Fri, 6 Jun 2025 11:48:52 +0530
Subject: [PATCH] Add lineage dialect for Exasol, Trino and Vertica (#21604)

(cherry picked from commit ea63db993a596a307bc688399f02da6b64935111)
---
 ingestion/src/metadata/ingestion/lineage/models.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/ingestion/src/metadata/ingestion/lineage/models.py b/ingestion/src/metadata/ingestion/lineage/models.py
index 2e5053a7339..9f37a0ccb60 100644
--- a/ingestion/src/metadata/ingestion/lineage/models.py
+++ b/ingestion/src/metadata/ingestion/lineage/models.py
@@ -37,6 +37,9 @@ from metadata.generated.schema.entity.services.connections.database.db2Connectio
 from metadata.generated.schema.entity.services.connections.database.deltaLakeConnection import (
     DeltaLakeType,
 )
+from metadata.generated.schema.entity.services.connections.database.exasolConnection import (
+    ExasolType,
+)
 from metadata.generated.schema.entity.services.connections.database.hiveConnection import (
     HiveType,
 )
@@ -73,9 +76,15 @@ from metadata.generated.schema.entity.services.connections.database.sqliteConnec
 from metadata.generated.schema.entity.services.connections.database.teradataConnection import (
     TeradataType,
 )
+from metadata.generated.schema.entity.services.connections.database.trinoConnection import (
+    TrinoType,
+)
 from metadata.generated.schema.entity.services.connections.database.unityCatalogConnection import (
     DatabricksType as UnityCatalogType,
 )
+from metadata.generated.schema.entity.services.connections.database.verticaConnection import (
+    VerticaType,
+)
 from metadata.utils.singleton import Singleton
 
 
@@ -106,6 +115,8 @@ class Dialect(Enum):
     TERADATA = "teradata"
     TSQL = "tsql"
     MARIADB = "mariadb"
+    TRINO = "trino"
+    VERTICA = "vertica"
 
 
 MAP_CONNECTION_TYPE_DIALECT: Dict[str, Dialect] = {
@@ -129,6 +140,9 @@ MAP_CONNECTION_TYPE_DIALECT: Dict[str, Dialect] = {
     str(TeradataType.Teradata.value): Dialect.TERADATA,
     str(MariaDBType.MariaDB.value): Dialect.MARIADB,
     str(SingleStoreType.SingleStore.value): Dialect.MYSQL,
+    str(ExasolType.Exasol.value): Dialect.EXASOL,
+    str(TrinoType.Trino.value): Dialect.TRINO,
+    str(VerticaType.Vertica.value): Dialect.VERTICA,
 }