90 lines
3.2 KiB
C++
90 lines
3.2 KiB
C++
/*
|
|
* Subspace.cpp
|
|
*
|
|
* This source file is part of the FoundationDB open source project
|
|
*
|
|
* Copyright 2013-2022 Apple Inc. and the FoundationDB project authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "fdbclient/Subspace.h"
|
|
|
|
Subspace::Subspace(Tuple const& tuple, StringRef const& rawPrefix) {
|
|
StringRef packed = tuple.pack();
|
|
|
|
this->rawPrefix.reserve(this->rawPrefix.arena(), rawPrefix.size() + packed.size());
|
|
this->rawPrefix.append(this->rawPrefix.arena(), rawPrefix.begin(), rawPrefix.size());
|
|
this->rawPrefix.append(this->rawPrefix.arena(), packed.begin(), packed.size());
|
|
}
|
|
|
|
Subspace::Subspace(Tuple const& tuple, Standalone<VectorRef<uint8_t>> const& rawPrefix) {
|
|
this->rawPrefix.reserve(this->rawPrefix.arena(), rawPrefix.size() + tuple.pack().size());
|
|
this->rawPrefix.append(this->rawPrefix.arena(), rawPrefix.begin(), rawPrefix.size());
|
|
this->rawPrefix.append(this->rawPrefix.arena(), tuple.pack().begin(), tuple.pack().size());
|
|
}
|
|
|
|
Subspace::Subspace(StringRef const& rawPrefix) {
|
|
this->rawPrefix.append(this->rawPrefix.arena(), rawPrefix.begin(), rawPrefix.size());
|
|
}
|
|
|
|
Subspace::~Subspace() {}
|
|
|
|
Key Subspace::key() const {
|
|
return StringRef(rawPrefix.begin(), rawPrefix.size());
|
|
}
|
|
|
|
Key Subspace::pack(const Tuple& tuple) const {
|
|
return tuple.pack().withPrefix(StringRef(rawPrefix.begin(), rawPrefix.size()));
|
|
}
|
|
|
|
Tuple Subspace::unpack(StringRef const& key) const {
|
|
if (!contains(key)) {
|
|
throw key_not_in_subspace();
|
|
}
|
|
return Tuple::unpack(key.substr(rawPrefix.size()));
|
|
}
|
|
|
|
KeyRange Subspace::range(Tuple const& tuple) const {
|
|
VectorRef<uint8_t> begin;
|
|
VectorRef<uint8_t> end;
|
|
|
|
KeyRange keyRange;
|
|
|
|
begin.reserve(keyRange.arena(), rawPrefix.size() + tuple.pack().size() + 1);
|
|
begin.append(keyRange.arena(), rawPrefix.begin(), rawPrefix.size());
|
|
begin.append(keyRange.arena(), tuple.pack().begin(), tuple.pack().size());
|
|
begin.push_back(keyRange.arena(), uint8_t('\x00'));
|
|
|
|
end.reserve(keyRange.arena(), rawPrefix.size() + tuple.pack().size() + 1);
|
|
end.append(keyRange.arena(), rawPrefix.begin(), rawPrefix.size());
|
|
end.append(keyRange.arena(), tuple.pack().begin(), tuple.pack().size());
|
|
end.push_back(keyRange.arena(), uint8_t('\xff'));
|
|
|
|
// FIXME: test that this uses the keyRange arena and doesn't create another one
|
|
keyRange.KeyRangeRef::operator=(
|
|
KeyRangeRef(StringRef(begin.begin(), begin.size()), StringRef(end.begin(), end.size())));
|
|
return keyRange;
|
|
}
|
|
|
|
bool Subspace::contains(KeyRef const& key) const {
|
|
return key.startsWith(StringRef(rawPrefix.begin(), rawPrefix.size()));
|
|
}
|
|
|
|
Subspace Subspace::subspace(Tuple const& tuple) const {
|
|
return Subspace(tuple, rawPrefix);
|
|
}
|
|
|
|
Subspace Subspace::get(Tuple const& tuple) const {
|
|
return subspace(tuple);
|
|
} |