mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-28 20:50:49 +00:00
67 lines
2.5 KiB
C++
67 lines
2.5 KiB
C++
#pragma once
|
||
|
||
#include <DB/Common/HashTable/HashMap.h>
|
||
|
||
|
||
/** Хеш-таблица, позволяющая очищать таблицу за O(1).
|
||
* Еще более простая, чем HashMap: Key и Mapped должны быть POD-типами.
|
||
*
|
||
* Вместо этого класса можно было бы просто использовать в HashMap в качестве ключа пару <версия, ключ>,
|
||
* но тогда таблица накапливала бы все ключи, которые в нее когда-либо складывали, и неоправданно росла.
|
||
* Этот класс идет на шаг дальше и считает ключи со старой версией пустыми местами в хеш-таблице.
|
||
*/
|
||
|
||
struct ClearableHashMapState
|
||
{
|
||
UInt32 version = 1;
|
||
|
||
/// Сериализация, в бинарном и текстовом виде.
|
||
void write(DB::WriteBuffer & wb) const { DB::writeBinary(version, wb); }
|
||
void writeText(DB::WriteBuffer & wb) const { DB::writeText(version, wb); }
|
||
|
||
/// Десериализация, в бинарном и текстовом виде.
|
||
void read(DB::ReadBuffer & rb) { DB::readBinary(version, rb); }
|
||
void readText(DB::ReadBuffer & rb) { DB::readText(version, rb); }
|
||
};
|
||
|
||
|
||
template <typename Key, typename Mapped, typename Hash>
|
||
struct ClearableHashMapCell : public HashMapCell
|
||
{
|
||
typedef ClearableHashMapState State;
|
||
|
||
UInt32 version;
|
||
|
||
ClearableHashMapCell() {}
|
||
ClearableHashMapCell(const Key & key_, const State & state) : value(key_, Mapped()), version(state.version) {}
|
||
ClearableHashMapCell(const value_type & value_, const State & state) : value(value_), version(state.version) {}
|
||
|
||
bool isZero(const State & state) const { return version != state.version; }
|
||
static bool isZero(const Key & key, const State & state) { return false; }
|
||
|
||
/// Установить значение ключа в ноль.
|
||
void setZero() { version = 0; }
|
||
|
||
/// Нужно ли хранить нулевой ключ отдельно (то есть, могут ли в хэш-таблицу вставить нулевой ключ).
|
||
static constexpr bool need_zero_value_storage = false;
|
||
};
|
||
|
||
|
||
template
|
||
<
|
||
typename Key,
|
||
typename Mapped,
|
||
typename Hash = DefaultHash<Key>,
|
||
typename Grower = HashTableGrower,
|
||
typename Allocator = HashTableAllocator
|
||
>
|
||
class ClearableHashMap : public HashMapTable<Key, ClearableHashMapCell<Key, Mapped, Hash>, Hash, Grower, Allocator>
|
||
{
|
||
public:
|
||
void clear()
|
||
{
|
||
++this->version;
|
||
this->m_size = 0;
|
||
}
|
||
};
|