[native] Relevant changes of presto protocol for distributed procedure

Author: hantangwangd

presto-main-base/src/main/java/com/facebook/presto/execution/scheduler/ExecutionWriterTarget.java

@@ -38,7 +38,7 @@
         @JsonSubTypes.Type(value = ExecutionWriterTarget.DeleteHandle.class, name = "DeleteHandle"),
         @JsonSubTypes.Type(value = ExecutionWriterTarget.RefreshMaterializedViewHandle.class, name = "RefreshMaterializedViewHandle"),
         @JsonSubTypes.Type(value = ExecutionWriterTarget.UpdateHandle.class, name = "UpdateHandle"),
-        @JsonSubTypes.Type(value = ExecutionWriterTarget.ExecuteProcedureHandle.class, name = "TableExecuteHandle")
+        @JsonSubTypes.Type(value = ExecutionWriterTarget.ExecuteProcedureHandle.class, name = "ExecuteProcedureHandle")
 })
 @SuppressWarnings({"EmptyClass", "ClassMayBeInterface"})
 public abstract class ExecutionWriterTarget

presto-native-execution/presto_cpp/presto_protocol/connector/hive/HiveConnectorProtocol.h

@@ -26,5 +26,6 @@ using HiveConnectorProtocol = ConnectorProtocolTemplate<
     HivePartitioningHandle,
     HiveTransactionHandle,
     NotImplemented,
+    NotImplemented,
     NotImplemented>;
 } // namespace facebook::presto::protocol::hive

presto-native-execution/presto_cpp/presto_protocol/connector/iceberg/IcebergConnectorProtocol.h

@@ -28,6 +28,7 @@ using IcebergConnectorProtocol = ConnectorProtocolTemplate<
     NotImplemented,
     hive::HiveTransactionHandle,
     NotImplemented,
+    NotImplemented,
     NotImplemented>;
 
 } // namespace facebook::presto::protocol::iceberg

presto-native-execution/presto_cpp/presto_protocol/connector/tpch/TpchConnectorProtocol.h

@@ -29,6 +29,7 @@ using TpchConnectorProtocol = ConnectorProtocolTemplate<
     TpchPartitioningHandle,
     TpchTransactionHandle,
     NotImplemented,
+    NotImplemented,
     NotImplemented>;
 
 } // namespace facebook::presto::protocol::tpch

presto-native-execution/presto_cpp/presto_protocol/core/ConnectorProtocol.h

@@ -75,6 +75,13 @@ class ConnectorProtocol {
     const std::string& thrift,
     std::shared_ptr<ConnectorInsertTableHandle>& proto) const = 0;
 
+  virtual void to_json(
+        json& j,
+        const std::shared_ptr<ConnectorDistributedProcedureHandle>& p) const = 0;
+  virtual void from_json(
+      const json& j,
+      std::shared_ptr<ConnectorDistributedProcedureHandle>& p) const = 0;
+
   virtual void to_json(
       json& j,
       const std::shared_ptr<ConnectorOutputTableHandle>& p) const = 0;
@@ -117,7 +124,7 @@ class ConnectorProtocol {
       std::string& thrift) const = 0;
   virtual void deserialize(
     const std::string& thrift,
-    std::shared_ptr<ConnectorTransactionHandle>& proto) const = 0;    
+    std::shared_ptr<ConnectorTransactionHandle>& proto) const = 0;
 
   virtual void to_json(
       json& j,
@@ -153,6 +160,7 @@ template <
     typename ConnectorSplitType = NotImplemented,
     typename ConnectorPartitioningHandleType = NotImplemented,
     typename ConnectorTransactionHandleType = NotImplemented,
+    typename ConnectorDistributedProcedureHandleType = NotImplemented,
     typename ConnectorDeleteTableHandleType = NotImplemented,
     typename ConnectorIndexHandleType = NotImplemented>
 class ConnectorProtocolTemplate final : public ConnectorProtocol {
@@ -221,6 +229,15 @@ class ConnectorProtocolTemplate final : public ConnectorProtocol {
     deserializeTemplate<ConnectorInsertTableHandleType>(thrift, proto);
   }
 
+  void to_json(json& j, const std::shared_ptr<ConnectorDistributedProcedureHandle>& p)
+    const final {
+    to_json_template<ConnectorDistributedProcedureHandleType>(j, p);
+  }
+  void from_json(const json& j, std::shared_ptr<ConnectorDistributedProcedureHandle>& p)
+      const final {
+    from_json_template<ConnectorDistributedProcedureHandleType>(j, p);
+  }
+
   void to_json(json& j, const std::shared_ptr<ConnectorOutputTableHandle>& p)
       const final {
     to_json_template<ConnectorOutputTableHandleType>(j, p);
@@ -406,6 +423,7 @@ using SystemConnectorProtocol = ConnectorProtocolTemplate<
     SystemPartitioningHandle,
     SystemTransactionHandle,
     NotImplemented,
+    NotImplemented,
     NotImplemented>;
 
 } // namespace facebook::presto::protocol