From ebd6020329da2676ba0dd36b94c5023e8e618b47 Mon Sep 17 00:00:00 2001
From: piby180 <piby180@gmail.com>
Date: Mon, 6 Jan 2025 10:14:02 +0100
Subject: [PATCH] Specify Length function for PinotDB (#19189)

---
 ingestion/src/metadata/profiler/orm/functions/length.py | 1 +
 ingestion/src/metadata/profiler/orm/registry.py         | 1 +
 2 files changed, 2 insertions(+)

diff --git a/ingestion/src/metadata/profiler/orm/functions/length.py b/ingestion/src/metadata/profiler/orm/functions/length.py
index 88aa32b3799d..19b32879be21 100644
--- a/ingestion/src/metadata/profiler/orm/functions/length.py
+++ b/ingestion/src/metadata/profiler/orm/functions/length.py
@@ -43,6 +43,7 @@ def _(element, compiler, **kw):
 @compiles(LenFn, Dialects.MariaDB)
 @compiles(LenFn, Dialects.Athena)
 @compiles(LenFn, Dialects.Trino)
+@compiles(LenFn, Dialects.PinotDB)
 @compiles(LenFn, Dialects.Presto)
 @compiles(LenFn, Dialects.BigQuery)
 @compiles(LenFn, Dialects.Oracle)
diff --git a/ingestion/src/metadata/profiler/orm/registry.py b/ingestion/src/metadata/profiler/orm/registry.py
index c1e42c5cdae5..efef50c4b103 100644
--- a/ingestion/src/metadata/profiler/orm/registry.py
+++ b/ingestion/src/metadata/profiler/orm/registry.py
@@ -76,6 +76,7 @@ class PythonDialects(Enum):
     MSSQL = "mssql"
     MySQL = "mysql"
     Oracle = "oracle"
+    PinotDB = "pinotdb"
     Postgres = "postgresql"
     Presto = "presto"
     Redshift = "redshift"