diff --git a/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.cpp b/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.cpp index ae0d9c985f8d..e5bb302fc196 100644 --- a/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.cpp +++ b/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.cpp @@ -11,11 +11,7 @@ * See the License for the specific language governing permissions and * limitations under the License. */ -/* - * Copyright (c) Facebook, Inc. and its affiliates. - */ #include "presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.h" - #include "presto_cpp/main/operators/PartitionAndSerialize.h" #include "velox/tool/trace/TraceReplayTaskRunner.h" @@ -32,6 +28,7 @@ PartitionAndSerializeReplayer::PartitionAndSerializeReplayer( const std::string& taskId, const std::string& nodeId, const std::string& nodeName, + const std::string& spillBaseDir, const std::string& driverIds, uint64_t queryCapacity, folly::Executor* executor) @@ -41,6 +38,7 @@ PartitionAndSerializeReplayer::PartitionAndSerializeReplayer( taskId, nodeId, nodeName, + spillBaseDir, driverIds, queryCapacity, executor) {} diff --git a/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.h b/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.h index f1cfd1c1d5a7..51882d2ff202 100644 --- a/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.h +++ b/presto-native-execution/presto_cpp/main/tool/trace/PartitionAndSerializeReplayer.h @@ -11,16 +11,12 @@ * See the License for the specific language governing permissions and * limitations under the License. */ -/* - * Copyright (c) Facebook, Inc. and its affiliates. - */ #pragma once #include "velox/core/PlanNode.h" #include "velox/tool/trace/OperatorReplayerBase.h" namespace facebook::velox::tool::trace { - /// The replayer to replay the traced 'PartitionAndSerialize' operators. class PartitionAndSerializeReplayer final : public OperatorReplayerBase { public: @@ -30,6 +26,7 @@ class PartitionAndSerializeReplayer final : public OperatorReplayerBase { const std::string& taskId, const std::string& nodeId, const std::string& nodeName, + const std::string& spillBaseDir, const std::string& driverIds, uint64_t queryCapacity, folly::Executor* executor); diff --git a/presto-native-execution/presto_cpp/main/tool/trace/TraceReplayerMain.cpp b/presto-native-execution/presto_cpp/main/tool/trace/TraceReplayerMain.cpp index 6d3a2486d86b..fecf70ddd18c 100644 --- a/presto-native-execution/presto_cpp/main/tool/trace/TraceReplayerMain.cpp +++ b/presto-native-execution/presto_cpp/main/tool/trace/TraceReplayerMain.cpp @@ -56,6 +56,7 @@ class PrestoTraceReplayRunner FLAGS_task_id, FLAGS_node_id, nodeName, + "", FLAGS_driver_ids, queryCapacityBytes, cpuExecutor_.get()); diff --git a/presto-native-execution/presto_cpp/main/tool/trace/tests/PartitionAndSerializeReplayerTest.cpp b/presto-native-execution/presto_cpp/main/tool/trace/tests/PartitionAndSerializeReplayerTest.cpp index ec2fef74eb6f..7847a4bedfeb 100644 --- a/presto-native-execution/presto_cpp/main/tool/trace/tests/PartitionAndSerializeReplayerTest.cpp +++ b/presto-native-execution/presto_cpp/main/tool/trace/tests/PartitionAndSerializeReplayerTest.cpp @@ -202,6 +202,7 @@ TEST_F(PartitionAndSerializeReplayerTest, basicReplay) { task->taskId(), partitionNodeId, "PartitionAndSerialize", + "", driverIds, queryCapacity, executor_.get());