Extended QPT to JDBC Multiplexing (#1959)

Enables QPT for Multiplexing JDBC-based connectors.
awslabs · May 7, 2024 · b81300d · b81300d
1 parent 17fba43
commit b81300d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/...c/src/main/java/com/amazonaws/athena/connectors/jdbc/MultiplexingJdbcMetadataHandler.java b/...c/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