aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/clickhouse/src/Interpreters/JIT/CompiledExpressionCache.h
blob: f5b2aab3233918125c2fb608dbbe986eac7fef21 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#pragma once

#include "clickhouse_config.h"

#if USE_EMBEDDED_COMPILER
#    include <Common/CacheBase.h>
#    include <Common/HashTable/Hash.h>
#    include <Interpreters/JIT/CHJIT.h>

namespace DB
{

class CompiledExpressionCacheEntry
{
public:
    explicit CompiledExpressionCacheEntry(size_t compiled_expression_size_)
        : compiled_expression_size(compiled_expression_size_)
    {}

    size_t getCompiledExpressionSize() const { return compiled_expression_size; }

    virtual ~CompiledExpressionCacheEntry() = default;

private:

    size_t compiled_expression_size = 0;

};

struct CompiledFunctionWeightFunction
{
    size_t operator()(const CompiledExpressionCacheEntry & compiled_function) const
    {
        return compiled_function.getCompiledExpressionSize();
    }
};

class CompiledExpressionCache : public CacheBase<UInt128, CompiledExpressionCacheEntry, UInt128Hash, CompiledFunctionWeightFunction>
{
public:
    using Base = CacheBase<UInt128, CompiledExpressionCacheEntry, UInt128Hash, CompiledFunctionWeightFunction>;
    using Base::Base;
};

class CompiledExpressionCacheFactory
{
private:
    std::unique_ptr<CompiledExpressionCache> cache;

public:
    static CompiledExpressionCacheFactory & instance();

    void init(size_t cache_size_in_bytes, size_t cache_size_in_elements);
    CompiledExpressionCache * tryGetCache();
};

}

#endif