Merge pull request ClickHouse#65917 from azat/mysql-server-TLS-retries

Fix possible issues with MySQL server protocol TLS connections

src/Server/MySQLHandler.cpp

@@ -204,6 +204,10 @@ void MySQLHandler::run()
 
     session->setClientConnectionId(connection_id);
 
+    const Settings & settings = server.context()->getSettingsRef();
+    socket().setReceiveTimeout(settings.receive_timeout);
+    socket().setSendTimeout(settings.send_timeout);
+
     in = std::make_shared<ReadBufferFromPocoSocket>(socket(), read_event);
     out = std::make_shared<WriteBufferFromPocoSocket>(socket(), write_event);
     packet_endpoint = std::make_shared<MySQLProtocol::PacketEndpoint>(*in, *out, sequence_id);
@@ -451,6 +455,7 @@ void MySQLHandler::comQuery(ReadBuffer & payload, bool binary_protocol)
 
         // Settings replacements
         if (!should_replace)
+        {
             for (auto const & [mysql_setting, clickhouse_setting] : settings_replacements)
             {
                 const auto replacement_query_opt = setSettingReplacementQuery(query, mysql_setting, clickhouse_setting);
@@ -461,6 +466,7 @@ void MySQLHandler::comQuery(ReadBuffer & payload, bool binary_protocol)
                     break;
                 }
             }
+        }
 
         auto query_context = session->makeQueryContext();
         query_context->setCurrentQueryId(fmt::format("mysql:{}:{}", connection_id, toString(UUIDHelpers::generateV4())));
@@ -470,6 +476,10 @@ void MySQLHandler::comQuery(ReadBuffer & payload, bool binary_protocol)
         settings.prefer_column_name_to_alias = true;
         query_context->setSettings(settings);
 
+        /// Update timeouts
+        socket().setReceiveTimeout(settings.receive_timeout);
+        socket().setSendTimeout(settings.send_timeout);
+
         CurrentThread::QueryScope query_scope{query_context};
 
         std::atomic<size_t> affected_rows {0};
@@ -643,7 +653,11 @@ void MySQLHandlerSSL::finishHandshakeSSL(
     client_capabilities = ssl_request.capability_flags;
     max_packet_size = ssl_request.max_packet_size ? ssl_request.max_packet_size : MAX_PACKET_LENGTH;
     secure_connection = true;
+
     ss = std::make_shared<SecureStreamSocket>(SecureStreamSocket::attach(socket(), SSLManager::instance().defaultServerContext()));
+    ss->setReceiveTimeout(socket().getReceiveTimeout());
+    ss->setSendTimeout(socket().getSendTimeout());
+
     in = std::make_shared<ReadBufferFromPocoSocket>(*ss);
     out = std::make_shared<WriteBufferFromPocoSocket>(*ss);
     sequence_id = 2;