about summary refs log tree commit diff stats
path: root/src/server/hashtable.h
blob: 0989a79ee3a6420a5223c339dc19ccc85c8abb80 (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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#pragma once

#include <algorithm>
#include <iostream>
#include <list>
#include <optional>
#include <vector>

template <typename K, typename V>
class HashTable {
public:
    HashTable(size_t size)
        : size { size }
        , table(size)
    {
    }

    bool insert(K key, V value)
    {
        std::list<std::pair<K, V>>& list = get_bucket(key);

        if (bucket_contains_key(list, key)) {
            return false;
        }

        list.push_back(std::pair(key, value));

        return true;
    }

    std::optional<V> get(K key)
    {
        std::list<std::pair<K, V>>& list = get_bucket(key);

        auto iter = bucket_find_key(list, key);
        if (iter != list.end()) {
            return std::optional<V>((*iter).second);
        }

        return std::optional<V>();
    }

    bool remove(K key)
    {
        std::list<std::pair<K, V>>& list = get_bucket(key);

        auto iter = bucket_find_key(list, key);
        if (iter != list.end()) {
            list.erase(iter);
            return true;
        }

        return false;
    }

    void print()
    {
        size_t index { 0 };
        for (auto bucket : table) {
            std::cout << "Bucket " << index++ << ": [";
            for (auto pair : bucket) {
                std::cout << "(" << pair.first << ", " << pair.second << ")";
            }
            std::cout << "]" << "\n";
        }
    }

private:
    size_t size;

    std::vector<std::list<std::pair<K, V>>> table;

    std::hash<K> hash_function;

    auto bucket_find_key(std::list<std::pair<K, V>>& list, K key)
    {
        return std::find_if(list.begin(), list.end(), [&key](const std::pair<K, V>& pair) {
            return pair.first == key;
        });
    }

    bool bucket_contains_key(std::list<std::pair<K, V>>& list, K key)
    {
        return list.begin() != list.end() && bucket_find_key(list, key) != list.end();
    }

    std::list<std::pair<K, V>>& get_bucket(K key)
    {
        size_t index = hash_function(key) % size;
        return table.at(index);
    }
};