boostorg/mqtt5
1
0
Fork 1
mirror of https://github.com/boostorg/mqtt5.git synced 2025-10-15 09:35:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

use associated executors for intermediate handlers

Browse Source 
Summary: * per-operation cancellation changed: total/partial signals only prevent further resending, terminal signal cancels the whole client

Reviewers: ivica

Reviewed By: ivica

Subscribers: korina

Differential Revision: https://repo.mireo.local/D27246
This commit is contained in:
Bruno Iljazovic
2024-01-09 15:18:58 +01:00
parent 1996251253
commit d78fdd3208
21 changed files with 311 additions and 414 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
include/async_mqtt5/impl/async_sender.hpp
View File

@@ -2,6 +2,8 @@
#define ASYNC_MQTT5_ASYNC_SENDER_HPP
#include <boost/asio/any_completion_handler.hpp>
#include <boost/asio/bind_allocator.hpp>
#include <boost/asio/bind_executor.hpp>
#include <boost/asio/buffer.hpp>
#include <boost/asio/prepend.hpp>
#include <boost/asio/ip/tcp.hpp>
@@ -44,6 +46,14 @@ public:
std::move(_handler)(ec);
}
auto get_executor() {
return asio::get_associated_executor(_handler);
}
auto get_allocator() {
return asio::get_associated_allocator(_handler);
}
bool throttled() const {
return _flags & send_flag::throttled;
}
@@ -94,11 +104,6 @@ class async_sender {
public:
explicit async_sender(ClientService& svc) : _svc(svc) {}
using executor_type = typename client_service::executor_type;
executor_type get_executor() const noexcept {
return _svc.get_executor();
}
using allocator_type = queue_allocator_type;
allocator_type get_allocator() const noexcept {
return allocator_type {};
@@ -223,7 +228,9 @@ private:
_write_queue.begin(), _write_queue.end(),
[](const auto& op) { return !op.throttled(); }
);
uint16_t dist = static_cast<uint16_t>(std::distance(throttled_ptr, _write_queue.end()));
uint16_t dist = static_cast<uint16_t>(
std::distance(throttled_ptr, _write_queue.end())
);
uint16_t throttled_num = std::min(dist, _quota);
_quota -= throttled_num;
throttled_ptr += throttled_num;
@@ -249,8 +256,17 @@ private:
_svc._replies.clear_fast_replies();
auto ex = write_queue.front().get_executor();
auto alloc = write_queue.front().get_allocator();
_svc._stream.async_write(
buffers, asio::prepend(std::ref(*this), std::move(write_queue))
buffers,
asio::bind_executor(
ex,
asio::bind_allocator(
alloc,
asio::prepend(std::ref(*this), std::move(write_queue))
)
)
);
}