fork of https://github.com/oxigraph/rocksdb and https://github.com/facebook/rocksdb for nextgraph and oxigraph
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
310 lines
10 KiB
310 lines
10 KiB
12 years ago
|
// Copyright (c) 2011 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.
|
||
|
|
||
11 years ago
|
#pragma once
|
||
11 years ago
|
|
||
|
#ifndef ROCKSDB_LITE
|
||
11 years ago
|
#include <deque>
|
||
|
#include <string>
|
||
11 years ago
|
#include <vector>
|
||
11 years ago
|
|
||
12 years ago
|
#include "rocksdb/db.h"
|
||
|
#include "rocksdb/env.h"
|
||
|
#include "rocksdb/compaction_filter.h"
|
||
|
#include "rocksdb/merge_operator.h"
|
||
12 years ago
|
#include "utilities/utility_db.h"
|
||
11 years ago
|
#include "utilities/db_ttl.h"
|
||
12 years ago
|
#include "db/db_impl.h"
|
||
|
|
||
11 years ago
|
namespace rocksdb {
|
||
12 years ago
|
|
||
11 years ago
|
class DBWithTTLImpl : public DBWithTTL {
|
||
12 years ago
|
public:
|
||
11 years ago
|
static void SanitizeOptions(int32_t ttl, ColumnFamilyOptions* options);
|
||
11 years ago
|
|
||
11 years ago
|
explicit DBWithTTLImpl(DB* db);
|
||
12 years ago
|
|
||
11 years ago
|
virtual ~DBWithTTLImpl();
|
||
|
|
||
|
Status CreateColumnFamilyWithTtl(const ColumnFamilyOptions& options,
|
||
|
const std::string& column_family_name,
|
||
|
ColumnFamilyHandle** handle,
|
||
|
int ttl) override;
|
||
|
|
||
|
Status CreateColumnFamily(const ColumnFamilyOptions& options,
|
||
|
const std::string& column_family_name,
|
||
|
ColumnFamilyHandle** handle) override;
|
||
12 years ago
|
|
||
11 years ago
|
using StackableDB::Put;
|
||
|
virtual Status Put(const WriteOptions& options,
|
||
11 years ago
|
ColumnFamilyHandle* column_family, const Slice& key,
|
||
11 years ago
|
const Slice& val) override;
|
||
12 years ago
|
|
||
11 years ago
|
using StackableDB::Get;
|
||
|
virtual Status Get(const ReadOptions& options,
|
||
11 years ago
|
ColumnFamilyHandle* column_family, const Slice& key,
|
||
11 years ago
|
std::string* value) override;
|
||
12 years ago
|
|
||
11 years ago
|
using StackableDB::MultiGet;
|
||
11 years ago
|
virtual std::vector<Status> MultiGet(
|
||
11 years ago
|
const ReadOptions& options,
|
||
11 years ago
|
const std::vector<ColumnFamilyHandle*>& column_family,
|
||
11 years ago
|
const std::vector<Slice>& keys,
|
||
11 years ago
|
std::vector<std::string>* values) override;
|
||
12 years ago
|
|
||
11 years ago
|
using StackableDB::KeyMayExist;
|
||
12 years ago
|
virtual bool KeyMayExist(const ReadOptions& options,
|
||
11 years ago
|
ColumnFamilyHandle* column_family, const Slice& key,
|
||
|
std::string* value,
|
||
12 years ago
|
bool* value_found = nullptr) override;
|
||
12 years ago
|
|
||
11 years ago
|
using StackableDB::Merge;
|
||
|
virtual Status Merge(const WriteOptions& options,
|
||
11 years ago
|
ColumnFamilyHandle* column_family, const Slice& key,
|
||
|
const Slice& value) override;
|
||
12 years ago
|
|
||
11 years ago
|
virtual Status Write(const WriteOptions& opts, WriteBatch* updates) override;
|
||
12 years ago
|
|
||
11 years ago
|
using StackableDB::NewIterator;
|
||
|
virtual Iterator* NewIterator(const ReadOptions& opts,
|
||
11 years ago
|
ColumnFamilyHandle* column_family) override;
|
||
12 years ago
|
|
||
11 years ago
|
virtual DB* GetBaseDB() { return db_; }
|
||
12 years ago
|
|
||
12 years ago
|
static bool IsStale(const Slice& value, int32_t ttl);
|
||
|
|
||
11 years ago
|
static Status AppendTS(const Slice& val, std::string* val_with_ts);
|
||
12 years ago
|
|
||
12 years ago
|
static Status SanityCheckTimestamp(const Slice& str);
|
||
12 years ago
|
|
||
12 years ago
|
static Status StripTS(std::string* str);
|
||
|
|
||
11 years ago
|
static Status GetCurrentTime(int64_t* curtime);
|
||
12 years ago
|
|
||
11 years ago
|
static const uint32_t kTSLength = sizeof(int32_t); // size of timestamp
|
||
12 years ago
|
|
||
11 years ago
|
static const int32_t kMinTimestamp = 1368146402; // 05/09/2013:5:40PM GMT-8
|
||
12 years ago
|
|
||
11 years ago
|
static const int32_t kMaxTimestamp = 2147483647; // 01/18/2038:7:14PM GMT-8
|
||
12 years ago
|
};
|
||
|
|
||
12 years ago
|
class TtlIterator : public Iterator {
|
||
|
|
||
|
public:
|
||
11 years ago
|
explicit TtlIterator(Iterator* iter) : iter_(iter) { assert(iter_); }
|
||
12 years ago
|
|
||
11 years ago
|
~TtlIterator() { delete iter_; }
|
||
12 years ago
|
|
||
11 years ago
|
bool Valid() const { return iter_->Valid(); }
|
||
12 years ago
|
|
||
11 years ago
|
void SeekToFirst() { iter_->SeekToFirst(); }
|
||
12 years ago
|
|
||
11 years ago
|
void SeekToLast() { iter_->SeekToLast(); }
|
||
12 years ago
|
|
||
11 years ago
|
void Seek(const Slice& target) { iter_->Seek(target); }
|
||
12 years ago
|
|
||
11 years ago
|
void Next() { iter_->Next(); }
|
||
12 years ago
|
|
||
11 years ago
|
void Prev() { iter_->Prev(); }
|
||
12 years ago
|
|
||
11 years ago
|
Slice key() const { return iter_->key(); }
|
||
12 years ago
|
|
||
12 years ago
|
int32_t timestamp() const {
|
||
11 years ago
|
return DecodeFixed32(iter_->value().data() + iter_->value().size() -
|
||
|
DBWithTTLImpl::kTSLength);
|
||
12 years ago
|
}
|
||
|
|
||
|
Slice value() const {
|
||
11 years ago
|
// TODO: handle timestamp corruption like in general iterator semantics
|
||
|
assert(DBWithTTLImpl::SanityCheckTimestamp(iter_->value()).ok());
|
||
12 years ago
|
Slice trimmed_value = iter_->value();
|
||
11 years ago
|
trimmed_value.size_ -= DBWithTTLImpl::kTSLength;
|
||
12 years ago
|
return trimmed_value;
|
||
|
}
|
||
|
|
||
11 years ago
|
Status status() const { return iter_->status(); }
|
||
12 years ago
|
|
||
|
private:
|
||
|
Iterator* iter_;
|
||
|
};
|
||
|
|
||
12 years ago
|
class TtlCompactionFilter : public CompactionFilter {
|
||
|
|
||
|
public:
|
||
12 years ago
|
TtlCompactionFilter(
|
||
11 years ago
|
int32_t ttl, const CompactionFilter* user_comp_filter,
|
||
|
std::unique_ptr<const CompactionFilter> user_comp_filter_from_factory =
|
||
|
nullptr)
|
||
|
: ttl_(ttl),
|
||
|
user_comp_filter_(user_comp_filter),
|
||
|
user_comp_filter_from_factory_(
|
||
|
std::move(user_comp_filter_from_factory)) {
|
||
12 years ago
|
// Unlike the merge operator, compaction filter is necessary for TTL, hence
|
||
|
// this would be called even if user doesn't specify any compaction-filter
|
||
12 years ago
|
if (!user_comp_filter_) {
|
||
|
user_comp_filter_ = user_comp_filter_from_factory_.get();
|
||
|
}
|
||
12 years ago
|
}
|
||
|
|
||
11 years ago
|
virtual bool Filter(int level, const Slice& key, const Slice& old_val,
|
||
|
std::string* new_val, bool* value_changed) const
|
||
|
override {
|
||
|
if (DBWithTTLImpl::IsStale(old_val, ttl_)) {
|
||
12 years ago
|
return true;
|
||
|
}
|
||
|
if (user_comp_filter_ == nullptr) {
|
||
|
return false;
|
||
|
}
|
||
11 years ago
|
assert(old_val.size() >= DBWithTTLImpl::kTSLength);
|
||
12 years ago
|
Slice old_val_without_ts(old_val.data(),
|
||
11 years ago
|
old_val.size() - DBWithTTLImpl::kTSLength);
|
||
12 years ago
|
if (user_comp_filter_->Filter(level, key, old_val_without_ts, new_val,
|
||
|
value_changed)) {
|
||
|
return true;
|
||
|
}
|
||
|
if (*value_changed) {
|
||
11 years ago
|
new_val->append(
|
||
|
old_val.data() + old_val.size() - DBWithTTLImpl::kTSLength,
|
||
|
DBWithTTLImpl::kTSLength);
|
||
12 years ago
|
}
|
||
|
return false;
|
||
|
}
|
||
|
|
||
11 years ago
|
virtual const char* Name() const override { return "Delete By TTL"; }
|
||
12 years ago
|
|
||
|
private:
|
||
|
int32_t ttl_;
|
||
|
const CompactionFilter* user_comp_filter_;
|
||
12 years ago
|
std::unique_ptr<const CompactionFilter> user_comp_filter_from_factory_;
|
||
|
};
|
||
|
|
||
|
class TtlCompactionFilterFactory : public CompactionFilterFactory {
|
||
11 years ago
|
public:
|
||
|
TtlCompactionFilterFactory(
|
||
|
int32_t ttl, std::shared_ptr<CompactionFilterFactory> comp_filter_factory)
|
||
|
: ttl_(ttl), user_comp_filter_factory_(comp_filter_factory) {}
|
||
|
|
||
|
virtual std::unique_ptr<CompactionFilter> CreateCompactionFilter(
|
||
|
const CompactionFilter::Context& context) {
|
||
|
return std::unique_ptr<TtlCompactionFilter>(new TtlCompactionFilter(
|
||
|
ttl_, nullptr,
|
||
|
std::move(user_comp_filter_factory_->CreateCompactionFilter(context))));
|
||
|
}
|
||
12 years ago
|
|
||
11 years ago
|
virtual const char* Name() const override {
|
||
|
return "TtlCompactionFilterFactory";
|
||
|
}
|
||
12 years ago
|
|
||
11 years ago
|
private:
|
||
|
int32_t ttl_;
|
||
|
std::shared_ptr<CompactionFilterFactory> user_comp_filter_factory_;
|
||
12 years ago
|
};
|
||
|
|
||
12 years ago
|
class TtlMergeOperator : public MergeOperator {
|
||
|
|
||
|
public:
|
||
12 years ago
|
explicit TtlMergeOperator(const std::shared_ptr<MergeOperator> merge_op)
|
||
11 years ago
|
: user_merge_op_(merge_op) {
|
||
12 years ago
|
assert(merge_op);
|
||
|
}
|
||
|
|
||
11 years ago
|
virtual bool FullMerge(const Slice& key, const Slice* existing_value,
|
||
12 years ago
|
const std::deque<std::string>& operands,
|
||
11 years ago
|
std::string* new_value, Logger* logger) const
|
||
|
override {
|
||
|
const uint32_t ts_len = DBWithTTLImpl::kTSLength;
|
||
12 years ago
|
if (existing_value && existing_value->size() < ts_len) {
|
||
|
Log(logger, "Error: Could not remove timestamp from existing value.");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
// Extract time-stamp from each operand to be passed to user_merge_op_
|
||
|
std::deque<std::string> operands_without_ts;
|
||
11 years ago
|
for (const auto& operand : operands) {
|
||
12 years ago
|
if (operand.size() < ts_len) {
|
||
12 years ago
|
Log(logger, "Error: Could not remove timestamp from operand value.");
|
||
|
return false;
|
||
|
}
|
||
12 years ago
|
operands_without_ts.push_back(operand.substr(0, operand.size() - ts_len));
|
||
12 years ago
|
}
|
||
12 years ago
|
|
||
|
// Apply the user merge operator (store result in *new_value)
|
||
12 years ago
|
bool good = true;
|
||
12 years ago
|
if (existing_value) {
|
||
|
Slice existing_value_without_ts(existing_value->data(),
|
||
|
existing_value->size() - ts_len);
|
||
12 years ago
|
good = user_merge_op_->FullMerge(key, &existing_value_without_ts,
|
||
|
operands_without_ts, new_value, logger);
|
||
12 years ago
|
} else {
|
||
12 years ago
|
good = user_merge_op_->FullMerge(key, nullptr, operands_without_ts,
|
||
|
new_value, logger);
|
||
|
}
|
||
|
|
||
|
// Return false if the user merge operator returned false
|
||
|
if (!good) {
|
||
|
return false;
|
||
12 years ago
|
}
|
||
|
|
||
|
// Augment the *new_value with the ttl time-stamp
|
||
11 years ago
|
int64_t curtime;
|
||
11 years ago
|
if (!DBWithTTLImpl::GetCurrentTime(&curtime).ok()) {
|
||
|
Log(logger,
|
||
|
"Error: Could not get current time to be attached internally "
|
||
|
"to the new value.");
|
||
12 years ago
|
return false;
|
||
12 years ago
|
} else {
|
||
12 years ago
|
char ts_string[ts_len];
|
||
11 years ago
|
EncodeFixed32(ts_string, (int32_t)curtime);
|
||
12 years ago
|
new_value->append(ts_string, ts_len);
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
|
||
11 years ago
|
virtual bool PartialMergeMulti(const Slice& key,
|
||
|
const std::deque<Slice>& operand_list,
|
||
|
std::string* new_value, Logger* logger) const
|
||
|
override {
|
||
11 years ago
|
const uint32_t ts_len = DBWithTTLImpl::kTSLength;
|
||
11 years ago
|
std::deque<Slice> operands_without_ts;
|
||
12 years ago
|
|
||
11 years ago
|
for (const auto& operand : operand_list) {
|
||
|
if (operand.size() < ts_len) {
|
||
|
Log(logger, "Error: Could not remove timestamp from value.");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
operands_without_ts.push_back(
|
||
|
Slice(operand.data(), operand.size() - ts_len));
|
||
12 years ago
|
}
|
||
12 years ago
|
|
||
|
// Apply the user partial-merge operator (store result in *new_value)
|
||
|
assert(new_value);
|
||
11 years ago
|
if (!user_merge_op_->PartialMergeMulti(key, operands_without_ts, new_value,
|
||
|
logger)) {
|
||
12 years ago
|
return false;
|
||
|
}
|
||
12 years ago
|
|
||
|
// Augment the *new_value with the ttl time-stamp
|
||
11 years ago
|
int64_t curtime;
|
||
11 years ago
|
if (!DBWithTTLImpl::GetCurrentTime(&curtime).ok()) {
|
||
|
Log(logger,
|
||
|
"Error: Could not get current time to be attached internally "
|
||
|
"to the new value.");
|
||
12 years ago
|
return false;
|
||
12 years ago
|
} else {
|
||
|
char ts_string[ts_len];
|
||
11 years ago
|
EncodeFixed32(ts_string, (int32_t)curtime);
|
||
12 years ago
|
new_value->append(ts_string, ts_len);
|
||
12 years ago
|
return true;
|
||
12 years ago
|
}
|
||
|
}
|
||
|
|
||
11 years ago
|
virtual const char* Name() const override { return "Merge By TTL"; }
|
||
12 years ago
|
|
||
|
private:
|
||
12 years ago
|
std::shared_ptr<MergeOperator> user_merge_op_;
|
||
12 years ago
|
};
|
||
12 years ago
|
}
|
||
11 years ago
|
#endif // ROCKSDB_LITE
|