Skip to content

Commit

Permalink
avoid duplicate construction for http dialer
Browse files Browse the repository at this point in the history
Signed-off-by: liulanzheng <[email protected]>
  • Loading branch information
liulanzheng committed May 29, 2024
1 parent 0db8545 commit 14afec1
Showing 1 changed file with 27 additions and 11 deletions.
38 changes: 27 additions & 11 deletions net/http/client.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -36,26 +36,41 @@ static constexpr char USERAGENT[] = "PhotonLibOS_HTTP";
class PooledDialer {
public:
net::TLSContext* tls_ctx = nullptr;
bool tls_ctx_ownership;
std::unique_ptr<ISocketClient> tcpsock;
std::unique_ptr<ISocketClient> tlssock;
std::unique_ptr<ISocketClient> udssock;
std::unique_ptr<Resolver> resolver;
photon::mutex init_mtx;
bool initialized = false;
bool tls_ctx_ownership = false;

// If there is a photon thread switch during construction, the constructor might be called
// multiple times, even for a thread_local instance. Therefore, ensure that there is no photon
// thread switch inside the constructor. Place the initialization work in init() and ensure it
// is initialized only once.
PooledDialer() {
photon::fini_hook({this, &PooledDialer::at_photon_fini});
}

//etsocket seems not support multi thread very well, use tcp_socket now. need to find out why
PooledDialer(TLSContext *_tls_ctx) :
tls_ctx(_tls_ctx ? _tls_ctx : new_tls_context(nullptr, nullptr, nullptr)),
tls_ctx_ownership(_tls_ctx == nullptr),
resolver(new_default_resolver(kDNSCacheLife)) {
int init(TLSContext *_tls_ctx) {
if (initialized)
return 0;
SCOPED_LOCK(init_mtx);
if (initialized)
return 0;
tls_ctx = _tls_ctx;
if (!tls_ctx) {
tls_ctx_ownership = true;
tls_ctx = new_tls_context(nullptr, nullptr, nullptr);
}
auto tcp_cli = new_tcp_socket_client();
auto tls_cli = new_tls_client(tls_ctx, new_tcp_socket_client(), true);
tcpsock.reset(new_tcp_socket_pool(tcp_cli, -1, true));
tlssock.reset(new_tcp_socket_pool(tls_cli, -1, true));
udssock.reset(new_uds_client());
photon::fini_hook({this, &PooledDialer::at_photon_fini});
}

~PooledDialer() {
resolver.reset(new_default_resolver(kDNSCacheLife));
initialized = true;
return 0;
}

void at_photon_fini() {
Expand Down Expand Up @@ -145,7 +160,8 @@ class ClientImpl : public Client {
}

PooledDialer& get_dialer() {
thread_local PooledDialer dialer(m_tls_ctx);
thread_local PooledDialer dialer;
dialer.init(m_tls_ctx);
return dialer;
}

Expand Down

0 comments on commit 14afec1

Please sign in to comment.