RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-25 13:27:35 +00:00
Code Issues Activity Actions

RequestServer: Keep the EnsureConnection job alive until it finishes

Browse source 
Fixes #12906.
This commit is contained in:
Ali Mohammad Pur 2022-03-17 19:33:13 +03:30 • committed by Andreas Kling
parent adff9a96a6
commit dff8344336
1 changed files with 36 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

50
Userland/Services/RequestServer/ConnectionFromClient.cpp
View file

@ -5,6 +5,7 @@
*/ */
#include <AK/Badge.h> #include <AK/Badge.h>
#include <AK/NonnullOwnPtr.h>
#include <RequestServer/ConnectionFromClient.h> #include <RequestServer/ConnectionFromClient.h>
#include <RequestServer/Protocol.h> #include <RequestServer/Protocol.h>
#include <RequestServer/Request.h> #include <RequestServer/Request.h>
@ -111,6 +112,40 @@ Messages::RequestServer::SetCertificateResponse ConnectionFromClient::set_certif
return success; return success;
} }
struct Job {
explicit Job(URL url)
: m_url(move(url))
{
}
static Job& ensure(URL const& url)
{
if (auto it = s_jobs.find(url); it == s_jobs.end()) {
auto job = make<Job>(url);
s_jobs.set(url, move(job));
}
return *s_jobs.find(url)->value;
}
void start(Core::Stream::Socket& socket)
{
auto is_connected = socket.is_open();
VERIFY(is_connected);
ConnectionCache::request_did_finish(m_url, &socket);
s_jobs.remove(m_url);
}
void fail(Core::NetworkJob::Error error)
{
dbgln("Pre-connect to {} failed: {}", m_url, Core::to_string(error));
s_jobs.remove(m_url);
}
URL m_url;
private:
static HashMap<URL, NonnullOwnPtr<Job>> s_jobs;
};
HashMap<URL, NonnullOwnPtr<Job>> Job::s_jobs {};
void ConnectionFromClient::ensure_connection(URL const& url, ::RequestServer::CacheLevel const& cache_level) void ConnectionFromClient::ensure_connection(URL const& url, ::RequestServer::CacheLevel const& cache_level)
{ {
if (!url.is_valid()) { if (!url.is_valid()) {
@ -125,20 +160,7 @@ void ConnectionFromClient::ensure_connection(URL const& url, ::RequestServer::Ca
}); });
} }
struct { auto& job = Job::ensure(url);
URL m_url;
void start(Core::Stream::Socket& socket)
{
auto is_connected = socket.is_open();
VERIFY(is_connected);
ConnectionCache::request_did_finish(m_url, &socket);
}
void fail(Core::NetworkJob::Error error)
{
dbgln("Pre-connect to {} failed: {}", m_url, Core::to_string(error));
}
} job { url };
dbgln("EnsureConnection: Pre-connect to {}", url); dbgln("EnsureConnection: Pre-connect to {}", url);
auto do_preconnect = [&](auto& cache) { auto do_preconnect = [&](auto& cache) {
auto it = cache.find({ url.host(), url.port_or_default() }); auto it = cache.find({ url.host(), url.port_or_default() });