tdlight/benchmark/bench_http.cpp

78 lines
2.2 KiB
C++
Raw Normal View History

//
2022-01-01 01:35:39 +01:00
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/net/HttpOutboundConnection.h"
#include "td/net/HttpQuery.h"
#include "td/net/SslStream.h"
2021-09-18 23:47:05 +02:00
#include "td/actor/actor.h"
#include "td/actor/ConcurrentScheduler.h"
#include "td/utils/buffer.h"
2021-11-04 10:46:08 +01:00
#include "td/utils/BufferedFd.h"
#include "td/utils/logging.h"
#include "td/utils/port/IPAddress.h"
#include "td/utils/port/SocketFd.h"
#include "td/utils/Status.h"
#include <atomic>
#include <limits>
std::atomic<int> counter;
2021-11-06 21:45:02 +01:00
class HttpClient final : public td::HttpOutboundConnection::Callback {
2021-07-03 22:51:36 +02:00
void start_up() final {
2021-11-06 21:45:02 +01:00
td::IPAddress addr;
addr.init_ipv4_port("127.0.0.1", 8082).ensure();
2021-11-06 21:45:02 +01:00
auto fd = td::SocketFd::open(addr);
LOG_CHECK(fd.is_ok()) << fd.error();
2021-11-06 21:45:02 +01:00
connection_ = td::create_actor<td::HttpOutboundConnection>(
"Connect", td::BufferedFd<td::SocketFd>(fd.move_as_ok()), td::SslStream{}, std::numeric_limits<size_t>::max(),
0, 0, td::ActorOwn<td::HttpOutboundConnection::Callback>(actor_id(this)));
yield();
cnt_ = 100000;
counter++;
}
2021-11-06 21:45:02 +01:00
2021-07-03 22:51:36 +02:00
void tear_down() final {
if (--counter == 0) {
2021-11-06 21:45:02 +01:00
td::Scheduler::instance()->finish();
}
}
2021-11-06 21:45:02 +01:00
2021-07-03 22:51:36 +02:00
void loop() final {
if (cnt_-- < 0) {
return stop();
}
2021-11-06 21:45:02 +01:00
send_closure(connection_, &td::HttpOutboundConnection::write_next, td::BufferSlice("GET / HTTP/1.1\r\n\r\n"));
send_closure(connection_, &td::HttpOutboundConnection::write_ok);
LOG(INFO) << "SEND";
}
2021-11-06 21:45:02 +01:00
void handle(td::unique_ptr<td::HttpQuery> result) final {
loop();
}
2021-11-06 21:45:02 +01:00
void on_connection_error(td::Status error) final {
LOG(ERROR) << "ERROR: " << error;
}
2021-11-06 21:45:02 +01:00
td::ActorOwn<td::HttpOutboundConnection> connection_;
2021-11-11 15:39:09 +01:00
int cnt_ = 0;
};
int main() {
SET_VERBOSITY_LEVEL(VERBOSITY_NAME(ERROR));
auto scheduler = td::make_unique<td::ConcurrentScheduler>(0, 0);
scheduler->create_actor_unsafe<HttpClient>(0, "Client1").release();
scheduler->create_actor_unsafe<HttpClient>(0, "Client2").release();
scheduler->start();
while (scheduler->run_main(10)) {
// empty
}
scheduler->finish();
}