[native] Update presto_protocol

Co-authored-by: Pramod Satya <pramod.satya@ibm.com>
prestodb · Aug 9, 2024 · 9c2fb0e · 9c2fb0e
1 parent f8a7e8e
commit 9c2fb0e
Show file tree

Hide file tree

Showing 7 changed files with 363 additions and 0 deletions.
diff --git a/presto-native-execution/presto_cpp/presto_protocol/ConnectorProtocol.h b/presto-native-execution/presto_cpp/presto_protocol/ConnectorProtocol.h
@@ -272,4 +272,15 @@ using SystemConnectorProtocol = ConnectorProtocolTemplate<
     SystemTransactionHandle,
     NotImplemented>;
 
+using TpcdsConnectorProtocol = ConnectorProtocolTemplate<
+    TpcdsTableHandle,
+    TpcdsTableLayoutHandle,
+    TpcdsColumnHandle,
+    NotImplemented,
+    NotImplemented,
+    TpcdsSplit,
+    TpcdsPartitioningHandle,
+    TpcdsTransactionHandle,
+    NotImplemented>;
+
 } // namespace facebook::presto::protocol
diff --git a/presto-native-execution/presto_cpp/presto_protocol/presto_protocol.cpp b/presto-native-execution/presto_cpp/presto_protocol/presto_protocol.cpp
diff --git a/presto-native-execution/presto_cpp/presto_protocol/presto_protocol.h b/presto-native-execution/presto_cpp/presto_protocol/presto_protocol.h
diff --git a/presto-native-execution/presto_cpp/presto_protocol/presto_protocol.yml b/presto-native-execution/presto_cpp/presto_protocol/presto_protocol.yml
@@ -53,13 +53,15 @@ AbstractClasses:
         - { name: IcebergColumnHandle,      key: hive-iceberg }
         - { name: TpchColumnHandle,         key: tpch }
         - { name: SystemColumnHandle,       key: $system@system }
+        - { name: TpcdsColumnHandle,        key: tpcds }
 
     ConnectorPartitioningHandle:
       super: JsonEncodedSubclass
       subclasses:
         - { name: SystemPartitioningHandle, key: $remote }
         - { name: HivePartitioningHandle,   key: hive}
         - { name: TpchPartitioningHandle,   key: tpch}
+        - { name: TpcdsPartitioningHandle,  key: tpcds}
 
     ConnectorTableHandle:
       super: JsonEncodedSubclass
@@ -68,6 +70,7 @@ AbstractClasses:
         - { name: IcebergTableHandle,       key: hive-iceberg }
         - { name: TpchTableHandle,          key: tpch }
         - { name: SystemTableHandle,        key: $system@system }
+        - { name: TpcdsTableHandle,         key: tpcds }
 
     ConnectorOutputTableHandle:
       super: JsonEncodedSubclass
@@ -93,6 +96,7 @@ AbstractClasses:
         - { name: IcebergTableLayoutHandle, key: hive-iceberg }
         - { name: TpchTableLayoutHandle,    key: tpch }
         - { name: SystemTableLayoutHandle,  key: $system@system }
+        - { name: TpcdsTableLayoutHandle,   key: tpcds }
 
     ConnectorMetadataUpdateHandle:
       super: JsonEncodedSubclass
@@ -105,6 +109,7 @@ AbstractClasses:
         - { name: HiveSplit,            key: hive }
         - { name: IcebergSplit,         key: hive-iceberg }
         - { name: TpchSplit,            key: tpch }
+        - { name: TpcdsSplit,           key: tpcds }
         - { name: RemoteSplit,          key: $remote }
         - { name: EmptySplit,           key: $empty }
         - { name: SystemSplit,          key: $system@system }
@@ -265,6 +270,11 @@ JavaClasses:
   - presto-tpch/src/main/java/com/facebook/presto/tpch/TpchTableLayoutHandle.java
   - presto-tpch/src/main/java/com/facebook/presto/tpch/TpchColumnHandle.java
   - presto-tpch/src/main/java/com/facebook/presto/tpch/TpchPartitioningHandle.java
+  - presto-tpcds/src/main/java/com/facebook/presto/tpcds/TpcdsSplit.java
+  - presto-tpcds/src/main/java/com/facebook/presto/tpcds/TpcdsTableHandle.java
+  - presto-tpcds/src/main/java/com/facebook/presto/tpcds/TpcdsTableLayoutHandle.java
+  - presto-tpcds/src/main/java/com/facebook/presto/tpcds/TpcdsColumnHandle.java
+  - presto-tpcds/src/main/java/com/facebook/presto/tpcds/TpcdsPartitioningHandle.java
   - presto-spi/src/main/java/com/facebook/presto/spi/plan/LimitNode.java
   - presto-spi/src/main/java/com/facebook/presto/spi/function/LongVariableConstraint.java
   - presto-common/src/main/java/com/facebook/presto/common/predicate/Marker.java

diff --git a/...to-native-execution/presto_cpp/presto_protocol/special/ConnectorTransactionHandle.cpp.inc b/...to-native-execution/presto_cpp/presto_protocol/special/ConnectorTransactionHandle.cpp.inc
@@ -13,6 +13,7 @@
  */
 
 // dependency TpchTransactionHandle
+// dependency TpcdsTransactionHandle
 
 namespace facebook::presto::protocol {
 void to_json(json& j, const std::shared_ptr<ConnectorTransactionHandle>& p) {