Set receive_timeout for anti join query (#230)

yandex · Sep 6, 2024 · b3e694d · b3e694d
1 parent 4949482
commit b3e694d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ch_tools/common/commands/clean_object_storage.py b/ch_tools/common/commands/clean_object_storage.py
@@ -159,9 +159,11 @@ def _clean_object_storage(
     deleted = 0
     total_size = 0
     with TemporaryFile() as keys_file:
+        timeout = ctx.obj["config"]["object_storage"]["clean"]["antijoin_timeout"]
         with ch_client.query(
             antijoin_query,
-            timeout=ctx.obj["config"]["object_storage"]["clean"]["antijoin_timeout"],
+            timeout=timeout,
+            settings={"receive_timeout": timeout},
             stream=True,
             format_="TabSeparated",
         ) as resp: