Backport #60373 to 24.1: Detect io_uring in tests

robot-clickhouse · robot-clickhouse · commit a225dfcdc6ae · 2024-03-01T13:04:44.000Z
diff --git a/src/Storages/StorageFile.cpp b/src/Storages/StorageFile.cpp
@@ -24,7 +24,9 @@
 #include <IO/WriteBufferFromFile.h>
 #include <IO/WriteHelpers.h>
 #include <IO/Archives/createArchiveReader.h>
-#include <IO/Archives/IArchiveReader.h>
+#include <IO/AsynchronousReadBufferFromFile.h>
+#include <Disks/IO/IOUringReader.h>
+
 
 #include <Formats/FormatFactory.h>
 #include <Formats/ReadSchemaUtils.h>
@@ -90,6 +92,7 @@ namespace ErrorCodes
     extern const int CANNOT_APPEND_TO_FILE;
     extern const int CANNOT_EXTRACT_TABLE_STRUCTURE;
     extern const int CANNOT_COMPILE_REGEXP;
+    extern const int UNSUPPORTED_METHOD;
 }
 
 namespace
@@ -274,6 +277,22 @@ std::unique_ptr<ReadBuffer> selectReadBuffer(
 
         ProfileEvents::increment(ProfileEvents::CreatedReadBufferOrdinary);
     }
+    else if (read_method == LocalFSReadMethod::io_uring && !use_table_fd)
+    {
+#if USE_LIBURING
+        auto & reader = context->getIOURingReader();
+        if (!reader.isSupported())
+            throw Exception(ErrorCodes::UNSUPPORTED_METHOD, "io_uring is not supported by this system");
+
+        res = std::make_unique<AsynchronousReadBufferFromFileWithDescriptorsCache>(
+            reader,
+            Priority{},
+            current_path,
+            context->getSettingsRef().max_read_buffer_size);
+#else
+        throw Exception(ErrorCodes::UNSUPPORTED_METHOD, "Read method io_uring is only supported in Linux");
+#endif
+    }
     else
     {
         if (use_table_fd)
diff --git a/tests/clickhouse-test b/tests/clickhouse-test
@@ -12,7 +12,6 @@ import itertools
 import sys
 import os
 import os.path
-import platform
 import signal
 import re
 import copy
@@ -563,6 +562,27 @@ def get_localzone():
     return os.getenv("TZ", "/".join(os.readlink("/etc/localtime").split("/")[-2:]))
 
 
+def supports_io_uring():
+    return not subprocess.call(
+        [
+            args.binary,
+            "-q",
+            "select * from file('/dev/null', 'LineAsString')",
+            "--storage_file_read_method",
+            "io_uring",
+        ],
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+    )
+
+
+def get_local_filesystem_methods():
+    methods = ["read", "pread", "mmap", "pread_threadpool"]
+    if supports_io_uring():
+        methods.append("io_uring")
+    return methods
+
+
 class SettingsRandomizer:
     settings = {
         "max_insert_threads": lambda: 0
@@ -603,10 +623,7 @@ class SettingsRandomizer:
             0.2, 0.5, 1, 10 * 1024 * 1024 * 1024
         ),
         "local_filesystem_read_method": lambda: random.choice(
-            # Allow to use uring only when running on Linux
-            ["read", "pread", "mmap", "pread_threadpool", "io_uring"]
-            if platform.system().lower() == "linux"
-            else ["read", "pread", "mmap", "pread_threadpool"]
+            get_local_filesystem_methods()
         ),
         "remote_filesystem_read_method": lambda: random.choice(["read", "threadpool"]),
         "local_filesystem_read_prefetch": lambda: random.randint(0, 1),
