diff --git a/benchmark/check_proxy.cpp b/benchmark/check_proxy.cpp index a493b1c46..e1ff5523e 100644 --- a/benchmark/check_proxy.cpp +++ b/benchmark/check_proxy.cpp @@ -160,20 +160,22 @@ int main(int argc, char **argv) { SET_VERBOSITY_LEVEL(new_verbosity_level); - td::Client client; + td::ClientManager client_manager; + auto client_id = client_manager.create_client(); for (size_t i = 0; i < requests.size(); i++) { auto &request = requests[i].second; request->dc_id_ = dc_id; request->timeout_ = timeout; - client.send({i + 1, std::move(request)}); + client_manager.send(client_id, i + 1, std::move(request)); } size_t successful_requests = 0; size_t failed_requests = 0; while (successful_requests + failed_requests != requests.size()) { - auto response = client.receive(100.0); - if (1 <= response.id && response.id <= requests.size()) { - auto &proxy = requests[static_cast(response.id - 1)].first; + auto response = client_manager.receive(100.0); + CHECK(client_id == response.client_id); + if (1 <= response.request_id && response.request_id <= requests.size()) { + auto &proxy = requests[static_cast(response.request_id - 1)].first; if (response.object->get_id() == td::td_api::error::ID) { LOG(ERROR) << proxy << ": " << to_string(response.object); failed_requests++;