mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-22 17:40:34 +00:00
c8ffffa557
Some packets are processed by a cpu other than the one that allocates it and its fragments. free_on_cpu() function should be called on a cpu that does processing and it returns a packet that is deletable by the current cpu. It is done by copying packet/packet::impl to locally allocated one and adding new deleter that runs old deleter on original cpu.
52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
/*
|
|
* Copyright (C) 2014 Cloudius Systems, Ltd.
|
|
*/
|
|
|
|
#include "core/reactor.hh"
|
|
#include "packet.hh"
|
|
|
|
namespace net {
|
|
|
|
constexpr size_t packet::internal_data_size;
|
|
constexpr size_t packet::default_nr_frags;
|
|
|
|
void packet::linearize(size_t at_frag, size_t desired_size) {
|
|
_impl->unuse_internal_data();
|
|
size_t nr_frags = 0;
|
|
size_t accum_size = 0;
|
|
while (accum_size < desired_size) {
|
|
accum_size += _impl->_frags[at_frag + nr_frags].size;
|
|
++nr_frags;
|
|
}
|
|
std::unique_ptr<char[]> new_frag{new char[accum_size]};
|
|
auto p = new_frag.get();
|
|
for (size_t i = 0; i < nr_frags; ++i) {
|
|
auto& f = _impl->_frags[at_frag + i];
|
|
p = std::copy(f.base, f.base + f.size, p);
|
|
}
|
|
// collapse nr_frags into one fragment
|
|
std::copy(_impl->_frags + at_frag + nr_frags, _impl->_frags + _impl->_nr_frags,
|
|
_impl->_frags + at_frag + 1);
|
|
_impl->_nr_frags -= nr_frags - 1;
|
|
_impl->_frags[at_frag] = fragment{new_frag.get(), accum_size};
|
|
_impl->_deleter = make_deleter(std::move(_impl->_deleter), [buf = std::move(new_frag)] {});
|
|
}
|
|
|
|
|
|
packet packet::free_on_cpu(unsigned cpu)
|
|
{
|
|
// make new deleter that runs old deleter on an origin cpu
|
|
_impl->_deleter = make_deleter(deleter(), [d = std::move(_impl->_deleter), cpu] () mutable {
|
|
smp::submit_to(cpu, [d = std::move(d)] () mutable {
|
|
// deleter needs to be moved from lambda capture to be destroyed here
|
|
// otherwise deleter destructor will be called on a cpu that called smp::submit_to()
|
|
// when work_item is destroyed.
|
|
deleter xxx(std::move(d));
|
|
});
|
|
});
|
|
|
|
return packet(impl::copy(_impl.get()));
|
|
}
|
|
|
|
}
|