YQ Connector: move tests from yql to ydb (OSS)

Перенос папки с тестами на Коннектор из папки yql в папку ydb (синхронизируется с github).

1 files changed, 59 insertions, 0 deletions

diff --git a/contrib/clickhouse/src/Storages/MergeTree/MergeFromLogEntryTask.h b/contrib/clickhouse/src/Storages/MergeTree/MergeFromLogEntryTask.h
new file mode 100644
index 0000000000..16e69a568b
--- /dev/null
+++ b/contrib/clickhouse/src/Storages/MergeTree/MergeFromLogEntryTask.h
@@ -0,0 +1,59 @@
+#pragma once
+
+#include <memory>
+#include <utility>
+
+#include <pcg_random.hpp>
+
+#include <Storages/MergeTree/IExecutableTask.h>
+#include <Storages/MergeTree/MergeTask.h>
+#include <Storages/MergeTree/ReplicatedMergeTreeQueue.h>
+#include <Storages/MergeTree/ReplicatedMergeTreeLogEntry.h>
+#include <Storages/MergeTree/ReplicatedMergeMutateTaskBase.h>
+#include <Storages/MergeTree/ZeroCopyLock.h>
+
+
+namespace DB
+{
+
+class MergeFromLogEntryTask : public ReplicatedMergeMutateTaskBase
+{
+public:
+    MergeFromLogEntryTask(
+        ReplicatedMergeTreeQueue::SelectedEntryPtr selected_entry_,
+        StorageReplicatedMergeTree & storage_,
+        IExecutableTask::TaskResultCallback & task_result_callback_);
+
+    Priority getPriority() const override { return priority; }
+
+protected:
+    /// Both return false if we can't execute merge.
+    ReplicatedMergeMutateTaskBase::PrepareResult prepare() override;
+    bool finalize(ReplicatedMergeMutateTaskBase::PartLogWriter write_part_log) override;
+
+    bool executeInnerTask() override
+    {
+        return merge_task->execute();
+    }
+
+private:
+    TableLockHolder table_lock_holder{nullptr};
+
+    MergeTreeData::DataPartsVector parts;
+    MergeTreeData::TransactionUniquePtr transaction_ptr{nullptr};
+    std::optional<ZeroCopyLock> zero_copy_lock;
+
+    StopwatchUniquePtr stopwatch_ptr{nullptr};
+    MergeTreeData::MutableDataPartPtr part;
+
+    Priority priority;
+
+    MergeTaskPtr merge_task;
+    pcg64 rng;
+};
+
+
+using MergeFromLogEntryTaskPtr = std::shared_ptr<MergeFromLogEntryTask>;
+
+
+}