tdlight/tddb/td/db/SeqKeyValue.h

106 lines
2.2 KiB
C
Raw Normal View History

//
2022-12-31 22:28:08 +01:00
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
2023-08-05 14:31:42 +02:00
#include "td/utils/common.h"
2023-07-27 12:58:20 +02:00
#include "td/utils/FlatHashMap.h"
2022-02-07 22:04:34 +01:00
#include "td/utils/Slice.h"
namespace td {
class SeqKeyValue {
public:
using SeqNo = uint64;
SeqKeyValue() = default;
SeqKeyValue(SeqKeyValue &&) = default;
SeqKeyValue &operator=(SeqKeyValue &&) = default;
SeqKeyValue(const SeqKeyValue &) = delete;
SeqKeyValue &operator=(const SeqKeyValue &) = delete;
~SeqKeyValue() = default;
SeqNo set(Slice key, Slice value) {
2023-07-27 12:58:20 +02:00
CHECK(!key.empty());
auto it_ok = map_.emplace(key.str(), value.str());
if (!it_ok.second) {
if (it_ok.first->second == value) {
return 0;
}
it_ok.first->second = value.str();
}
return next_seq_no();
}
2022-08-17 15:52:27 +02:00
SeqNo erase(const string &key) {
auto it = map_.find(key);
if (it == map_.end()) {
return 0;
}
map_.erase(it);
return next_seq_no();
}
2022-08-17 15:52:27 +02:00
2023-02-12 01:29:19 +01:00
SeqNo erase_batch(vector<string> keys) {
size_t count = 0;
for (auto &key : keys) {
auto it = map_.find(key);
if (it != map_.end()) {
map_.erase(it);
count++;
}
}
if (count == 0) {
return 0;
}
SeqNo result = current_id_ + 1;
current_id_ += count;
return result;
}
SeqNo seq_no() const {
return current_id_ + 1;
}
2022-08-17 15:52:27 +02:00
string get(const string &key) const {
auto it = map_.find(key);
if (it == map_.end()) {
return string();
}
return it->second;
}
2022-08-17 15:52:27 +02:00
bool isset(const string &key) const {
auto it = map_.find(key);
if (it == map_.end()) {
return false;
}
return true;
}
size_t size() const {
return map_.size();
}
2023-07-27 12:58:20 +02:00
FlatHashMap<string, string> get_all() const {
FlatHashMap<string, string> result;
result.reserve(map_.size());
for (auto &it : map_) {
result.emplace(it.first, it.second);
}
return result;
}
private:
2023-07-27 12:58:20 +02:00
FlatHashMap<string, string> map_;
SeqNo current_id_ = 0;
2023-07-27 12:58:20 +02:00
SeqNo next_seq_no() {
return ++current_id_;
}
};
} // namespace td