Fix mutations after turning on zero-copy

Author: ianton-ru

src/Storages/StorageReplicatedMergeTree.cpp

@@ -10300,6 +10300,8 @@ void StorageReplicatedMergeTree::getZeroCopyLockNodeCreateOps(
         requests.emplace_back(zkutil::makeCreateRequest(zookeeper_node, "", mode));
         if (!path_to_set_hardlinked_files.empty() && !hardlinked_files.empty())
         {
+            /// Node may be missing in rare cases when zero-copy is turned on for an existing table.
+            zookeeper->checkExistsAndGetCreateAncestorsOps(path_to_set_hardlinked_files + "/", requests);
             std::string data = boost::algorithm::join(hardlinked_files, "\n");
             /// List of files used to detect hardlinks. path_to_set_hardlinked_files --
             /// is a path to source part zero copy node. During part removal hardlinked
@@ -10312,6 +10314,8 @@ void StorageReplicatedMergeTree::getZeroCopyLockNodeCreateOps(
         Coordination::Requests ops;
         if (!path_to_set_hardlinked_files.empty() && !hardlinked_files.empty())
         {
+            /// Node may be missing in rare cases when zero-copy is turned on for an existing table.
+            zookeeper->checkExistsAndGetCreateAncestorsOps(path_to_set_hardlinked_files + "/", requests);
             std::string data = boost::algorithm::join(hardlinked_files, "\n");
             /// List of files used to detect hardlinks. path_to_set_hardlinked_files --
             /// is a path to source part zero copy node. During part removal hardlinked

tests/queries/0_stateless/03391_mutations_after_zero_copy_turned_on.reference

@@ -0,0 +1 @@
+1

tests/queries/0_stateless/03391_mutations_after_zero_copy_turned_on.sh

@@ -0,0 +1,33 @@
+#!/usr/bin/env bash
+# Tags: no-parallel, no-fasttest
+
+set -e
+
+CUR_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
+# shellcheck source=../shell_config.sh
+. "$CUR_DIR"/../shell_config.sh
+
+$CLICKHOUSE_CLIENT -q "DROP TABLE IF EXISTS test03391_zero_copy_mutations"
+$CLICKHOUSE_CLIENT -m -q "
+CREATE TABLE test03391_zero_copy_mutations
+    (key UInt32, value UInt32) 
+    ENGINE=ReplicatedMergeTree('/clickhouse/tables/$CLICKHOUSE_TEST_ZOOKEEPER_PREFIX/test03391_zero_copy_mutations', 'r1')
+    ORDER BY key
+    SETTINGS storage_policy='s3_cache', allow_remote_fs_zero_copy_replication=0
+"
+
+$CLICKHOUSE_CLIENT -q "INSERT INTO test03391_zero_copy_mutations VALUES (1,1)"
+$CLICKHOUSE_CLIENT -q "ALTER TABLE test03391_zero_copy_mutations MODIFY SETTING allow_remote_fs_zero_copy_replication=1"
+$CLICKHOUSE_CLIENT -q "ALTER TABLE test03391_zero_copy_mutations UPDATE value=value WHERE 0"
+
+for i in {1..10}
+do
+    if [ "$(${CLICKHOUSE_CLIENT} -q "SELECT count() FROM system.mutations WHERE table='test03391_zero_copy_mutations' AND is_done=1")" -eq 1 ]; then
+        break
+    fi
+    sleep 1
+done
+    
+echo '1' | $CLICKHOUSE_CLIENT -q "SELECT count() FROM system.mutations WHERE table='test03391_zero_copy_mutations' AND is_done=1"
+
+$CLICKHOUSE_CLIENT -q "DROP TABLE IF EXISTS test03391_zero_copy_mutations"