Files
scylladb/sstables/prepended_input_stream.cc
Avi Kivity fcb8d040e8 treewide: use Software Package Data Exchange (SPDX) license identifiers
Instead of lengthy blurbs, switch to single-line, machine-readable
standardized (https://spdx.dev) license identifiers. The Linux kernel
switched long ago, so there is strong precedent.

Three cases are handled: AGPL-only, Apache-only, and dual licensed.
For the latter case, I chose (AGPL-3.0-or-later and Apache-2.0),
reasoning that our changes are extensive enough to apply our license.

The changes we applied mechanically with a script, except to
licenses/README.md.

Closes #9937
2022-01-18 12:15:18 +01:00

54 lines
1.5 KiB
C++

/*
* Copyright (C) 2017-present ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#include "prepended_input_stream.hh"
using namespace seastar;
class prepended_data_source_impl : public data_source_impl {
private:
temporary_buffer<char> _buf;
data_source _ds;
public:
prepended_data_source_impl(temporary_buffer<char>&& buf, data_source&& ds)
: _buf(std::move(buf)), _ds(std::move(ds))
{}
prepended_data_source_impl(prepended_data_source_impl&&) = default;
prepended_data_source_impl& operator=(prepended_data_source_impl&&) = default;
virtual future<temporary_buffer<char>> get() override {
if (_buf) {
return make_ready_future<temporary_buffer<char>>(std::move(_buf));
}
return _ds.get();
}
virtual future<temporary_buffer<char>> skip(uint64_t n) override {
if (n < _buf.size()) {
_buf.trim_front(n);
return make_ready_future<temporary_buffer<char>>(std::move(_buf));
} else {
n -= _buf.size();
_buf = {};
if (n > 0) {
return _ds.skip(n);
} else {
return _ds.get();
}
}
}
virtual future<> close() override {
return _ds.close();
}
};
input_stream<char> make_prepended_input_stream(temporary_buffer<char>&& buf, data_source&& ds) {
auto impl = std::make_unique<prepended_data_source_impl>(std::move(buf), std::move(ds));
return input_stream<char>{data_source{std::move(impl)}};
}