From b81300d72d75d9619d2e8e4331d7d5e1b0bf5877 Mon Sep 17 00:00:00 2001
From: AbdulRehman <ar.al-faraj@outlook.com>
Date: Tue, 7 May 2024 09:33:41 -0400
Subject: [PATCH] Extended QPT to JDBC Multiplexing (#1959)

Enables QPT for Multiplexing JDBC-based connectors.
---
 .../connectors/jdbc/MultiplexingJdbcMetadataHandler.java  | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/athena-jdbc/src/main/java/com/amazonaws/athena/connectors/jdbc/MultiplexingJdbcMetadataHandler.java b/athena-jdbc/src/main/java/com/amazonaws/athena/connectors/jdbc/MultiplexingJdbcMetadataHandler.java
index a8538c2aa4..c20fa505da 100644
--- a/athena-jdbc/src/main/java/com/amazonaws/athena/connectors/jdbc/MultiplexingJdbcMetadataHandler.java
+++ b/athena-jdbc/src/main/java/com/amazonaws/athena/connectors/jdbc/MultiplexingJdbcMetadataHandler.java
@@ -126,6 +126,14 @@ public GetTableResponse doGetTable(BlockAllocator blockAllocator, GetTableReques
         return this.metadataHandlerMap.get(getTableRequest.getCatalogName()).doGetTable(blockAllocator, getTableRequest);
     }
 
+    @Override
+    public GetTableResponse doGetQueryPassthroughSchema(final BlockAllocator blockAllocator, final GetTableRequest getTableRequest)
+          throws Exception
+    {
+      validateMultiplexer(getTableRequest.getCatalogName());
+      return this.metadataHandlerMap.get(getTableRequest.getCatalogName()).doGetQueryPassthroughSchema(blockAllocator, getTableRequest);
+    }
+
     @Override
     public void getPartitions(final BlockWriter blockWriter, final GetTableLayoutRequest getTableLayoutRequest, QueryStatusChecker queryStatusChecker)
             throws Exception