72a2554a86
do_linearize() is an impure function as it changes state in linearization_context. Extract the pure parts into a new do_linearize_pure(). This will be used to linearize managed_bytes without a linearization_context, during the transition period where fragmented and non-fragmented values coexist.
60 lines
1.8 KiB
C++
60 lines
1.8 KiB
C++
|
|
/*
|
|
* Copyright 2015 ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* This file is part of Scylla.
|
|
*
|
|
* Scylla is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Scylla is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
|
|
#include "managed_bytes.hh"
|
|
|
|
thread_local managed_bytes::linearization_context managed_bytes::_linearization_context;
|
|
thread_local std::unordered_map<const blob_storage*, std::unique_ptr<bytes_view::value_type[]>> managed_bytes::_lc_state;
|
|
|
|
void
|
|
managed_bytes::linearization_context::forget(const blob_storage* p) noexcept {
|
|
_lc_state.erase(p);
|
|
}
|
|
|
|
std::unique_ptr<bytes_view::value_type[]>
|
|
managed_bytes::do_linearize_pure() const {
|
|
auto b = _u.ptr;
|
|
auto data = std::unique_ptr<bytes_view::value_type[]>(new bytes_view::value_type[b->size]);
|
|
auto e = data.get();
|
|
while (b) {
|
|
e = std::copy_n(b->data, b->frag_size, e);
|
|
b = b->next;
|
|
}
|
|
return data;
|
|
}
|
|
|
|
const bytes_view::value_type*
|
|
managed_bytes::do_linearize() const {
|
|
auto& lc = _linearization_context;
|
|
assert(lc._nesting);
|
|
lc._state_ptr = &_lc_state;
|
|
auto b = _u.ptr;
|
|
auto i = _lc_state.find(b);
|
|
if (i == _lc_state.end()) {
|
|
auto data = do_linearize_pure();
|
|
i = _lc_state.emplace(_u.ptr, std::move(data)).first;
|
|
}
|
|
return i->second.get();
|
|
}
|
|
|