From c26d409f1697ec2a9366a2769b1d32adfe110900 Mon Sep 17 00:00:00 2001
From: Soumitra Kumar <soumitra@startree.ai>
Date: Mon, 6 May 2024 18:08:19 -0700
Subject: [PATCH] Multi-stage query engine of Pinot supports subquery, join,
 aliases

---
 docs/docs/configuration/databases.mdx | 5 +++++
 superset/db_engine_specs/pinot.py     | 8 ++++----
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/docs/docs/configuration/databases.mdx b/docs/docs/configuration/databases.mdx
index 5c73a44d2d..9ecbd4fe4a 100644
--- a/docs/docs/configuration/databases.mdx
+++ b/docs/docs/configuration/databases.mdx
@@ -1035,6 +1035,11 @@ The expected connection string using username and password is formatted as follo
 pinot://<username>:<password>@<pinot-broker-host>:<pinot-broker-port>/query/sql?controller=http://<pinot-controller-host>:<pinot-controller-port>/verify_ssl=true``
 ```
 
+If you want to use explore view or joins, window functions, etc. then enable [multi-stage query engine](https://docs.pinot.apache.org/reference/multi-stage-engine).
+Add below argument while creating database connection in Advanced -> Other -> ENGINE PARAMETERS
+```
+{"connect_args":{"use_multistage_engine":"true"}}
+```
 
 #### Postgres
 
diff --git a/superset/db_engine_specs/pinot.py b/superset/db_engine_specs/pinot.py
index 0f53cfa77b..80eeb237e6 100644
--- a/superset/db_engine_specs/pinot.py
+++ b/superset/db_engine_specs/pinot.py
@@ -25,10 +25,10 @@ from superset.db_engine_specs.base import BaseEngineSpec
 class PinotEngineSpec(BaseEngineSpec):
     engine = "pinot"
     engine_name = "Apache Pinot"
-    allows_subqueries = False
-    allows_joins = False
-    allows_alias_in_select = False
-    allows_alias_in_orderby = False
+    allows_subqueries = True
+    allows_joins = True
+    allows_alias_in_select = True
+    allows_alias_in_orderby = True
 
     # https://docs.pinot.apache.org/users/user-guide-query/supported-transformations#datetime-functions
     _time_grain_expressions = {