contrib/clickhouse/src/Functions/kostikConsistentHash.cpp


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

#include "FunctionsConsistentHashing.h"
#include <Functions/FunctionFactory.h>

#include <consistent_hashing.h>

namespace DB
{

/// An O(1) time and space consistent hash algorithm by Konstantin Oblakov
struct KostikConsistentHashImpl
{
    static constexpr auto name = "kostikConsistentHash";

    using HashType = UInt64;
    /// Actually it supports UInt64, but it is efficient only if n <= 32768
    using ResultType = UInt16;
    using BucketsType = ResultType;
    static constexpr auto max_buckets = 32768;

    static inline ResultType apply(UInt64 hash, BucketsType n)
    {
        return ConsistentHashing(hash, n);
    }
};

using FunctionKostikConsistentHash = FunctionConsistentHashImpl<KostikConsistentHashImpl>;

REGISTER_FUNCTION(KostikConsistentHash)
{
    factory.registerFunction<FunctionKostikConsistentHash>();
    factory.registerAlias("yandexConsistentHash", "kostikConsistentHash");
}

}