ClickHouse/dbms/include/DB/Storages/MarkCache.h
2016-05-28 13:35:44 +03:00

66 lines
1.6 KiB
C++

#pragma once
#include <memory>
#include <DB/Common/LRUCache.h>
#include <DB/Common/ProfileEvents.h>
#include <DB/Common/SipHash.h>
#include <DB/Interpreters/AggregationCommon.h>
#include <DB/DataStreams/MarkInCompressedFile.h>
namespace DB
{
/// Оценка количества байтов, занимаемых засечками в кеше.
struct MarksWeightFunction
{
size_t operator()(const MarksInCompressedFile & marks) const
{
/// Можно еще добавить порядка 100 байт на накладные расходы вектора и кеша.
return marks.size() * sizeof(MarkInCompressedFile);
}
};
/** Кэш засечек в столбце из StorageMergeTree.
*/
class MarkCache : public LRUCache<UInt128, MarksInCompressedFile, UInt128TrivialHash, MarksWeightFunction>
{
private:
using Base = LRUCache<UInt128, MarksInCompressedFile, UInt128TrivialHash, MarksWeightFunction>;
public:
MarkCache(size_t max_size_in_bytes, const Delay & expiration_delay)
: Base(max_size_in_bytes, expiration_delay) {}
/// Посчитать ключ от пути к файлу и смещения.
static UInt128 hash(const String & path_to_file)
{
UInt128 key;
SipHash hash;
hash.update(path_to_file.data(), path_to_file.size() + 1);
hash.get128(key.first, key.second);
return key;
}
MappedPtr get(const Key & key)
{
MappedPtr res = Base::get(key);
if (res)
ProfileEvents::increment(ProfileEvents::MarkCacheHits);
else
ProfileEvents::increment(ProfileEvents::MarkCacheMisses);
return res;
}
};
using MarkCachePtr = std::shared_ptr<MarkCache>;
}