Disable S3 requests processing during context shutdown to speed up termination process.

src/Disks/IDisk.h

@@ -183,6 +183,9 @@ public:
     /// Return disk type - "local", "s3", etc.
     virtual const String getType() const = 0;
 
+    /// Invoked when Global Context is shutdown.
+    virtual void shutdown() { }
+
 private:
     /// Returns executor to perform asynchronous operations.
     Executor & getExecutor() { return *executor; }

src/Disks/S3/DiskS3.cpp

@@ -746,4 +746,9 @@ void DiskS3::setReadOnly(const String & path)
     Poco::File(metadata_path + path).setReadOnly(true);
 }
 
+void DiskS3::shutdown()
+{
+    client->DisableRequestProcessing();
+}
+
 }

src/Disks/S3/DiskS3.h

@@ -102,6 +102,8 @@ public:
 
     const String getType() const override { return "s3"; }
 
+    void shutdown() override;
+
 private:
     bool tryReserve(UInt64 bytes);
 

src/Interpreters/Context.cpp

@@ -1973,6 +1973,13 @@ void Context::reloadConfig() const
 
 void Context::shutdown()
 {
+    auto disks = getDisksMap();
+    for (auto & [disk_name, disk] : disks)
+    {
+        LOG_INFO(shared->log, "Shutdown disk {}", disk_name);
+        disk->shutdown();
+    }
+
     shared->shutdown();
 }