Summary: This patch adds three new MemTableRep's: UnsortedRep, PrefixHashRep, and VectorRep. UnsortedRep stores keys in an std::unordered_map of std::sets. When an iterator is requested, it dumps the keys into an std::set and iterates over that. VectorRep stores keys in an std::vector. When an iterator is requested, it creates a copy of the vector and sorts it using std::sort. The iterator accesses that new vector. PrefixHashRep stores keys in an unordered_map mapping prefixes to ordered sets. I also added one API change. I added a function MemTableRep::MarkImmutable. This function is called when the rep is added to the immutable list. It doesn't do anything yet, but it seems like that could be useful. In particular, for the vectorrep, it means we could elide the extra copy and just sort in place. The only reason I haven't done that yet is because the use of the ArenaAllocator complicates things (I can elaborate on this if needed). Test Plan: make -j32 check ./db_stress --memtablerep=vector ./db_stress --memtablerep=unsorted ./db_stress --memtablerep=prefixhash --prefix_size=10 Reviewers: dhruba, haobo, emayanke Reviewed By: dhruba CC: leveldb Differential Revision: https://reviews.facebook.net/D12117
69 lines
1.4 KiB
C++
69 lines
1.4 KiB
C++
// Copyright (c) 2012 The LevelDB Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file. See the AUTHORS file for names of contributors.
|
|
|
|
#include "leveldb/slice_transform.h"
|
|
#include "leveldb/slice.h"
|
|
|
|
namespace leveldb {
|
|
|
|
namespace {
|
|
|
|
class FixedPrefixTransform : public SliceTransform {
|
|
private:
|
|
size_t prefix_len_;
|
|
|
|
public:
|
|
explicit FixedPrefixTransform(size_t prefix_len) : prefix_len_(prefix_len) { }
|
|
|
|
virtual const char* Name() const {
|
|
return "rocksdb.FixedPrefix";
|
|
}
|
|
|
|
virtual Slice Transform(const Slice& src) const {
|
|
assert(InDomain(src));
|
|
return Slice(src.data(), prefix_len_);
|
|
}
|
|
|
|
virtual bool InDomain(const Slice& src) const {
|
|
return (src.size() >= prefix_len_);
|
|
}
|
|
|
|
virtual bool InRange(const Slice& dst) const {
|
|
return (dst.size() == prefix_len_);
|
|
}
|
|
};
|
|
|
|
class NoopTransform : public SliceTransform {
|
|
public:
|
|
explicit NoopTransform() { }
|
|
|
|
virtual const char* Name() const {
|
|
return "rocksdb.Noop";
|
|
}
|
|
|
|
virtual Slice Transform(const Slice& src) const {
|
|
return src;
|
|
}
|
|
|
|
virtual bool InDomain(const Slice& src) const {
|
|
return true;
|
|
}
|
|
|
|
virtual bool InRange(const Slice& dst) const {
|
|
return true;
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
const SliceTransform* NewFixedPrefixTransform(size_t prefix_len) {
|
|
return new FixedPrefixTransform(prefix_len);
|
|
}
|
|
|
|
const SliceTransform* NewNoopTransform() {
|
|
return new NoopTransform;
|
|
}
|
|
|
|
} // namespace leveldb
|