aboutsummaryrefslogtreecommitdiffstats
path: root/src/core/function/CacheManager.h
blob: 7a1755861f50f34a589dc73c2eaa68746032c05b (plain)
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/**
 * Copyright (C) 2021 Saturneric
 *
 * This file is part of GpgFrontend.
 *
 * GpgFrontend is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * GpgFrontend is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GpgFrontend. If not, see <https://www.gnu.org/licenses/>.
 *
 * The initial version of the source code is inherited from
 * the gpg4usb project, which is under GPL-3.0-or-later.
 *
 * All the source code of GpgFrontend was modified and released by
 * Saturneric<[email protected]> starting on May 12, 2021.
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 */

#ifndef GPGFRONTEND_CACHEMANAGER_H
#define GPGFRONTEND_CACHEMANAGER_H

#include <string>

#include "core/GpgFunctionObject.h"

namespace GpgFrontend {

template <typename Key, typename Value>
class ThreadSafeMap {
 public:
  using MapType = std::map<Key, Value>;
  using IteratorType = typename MapType::iterator;

  void insert(const Key& key, const Value& value) {
    std::unique_lock lock(mutex_);
    map_[key] = value;
  }

  std::optional<Value> get(const Key& key) {
    std::shared_lock lock(mutex_);
    auto it = map_.find(key);
    if (it != map_.end()) {
      return it->second;
    }
    return std::nullopt;
  }

  bool exists(const Key& key) {
    std::shared_lock lock(mutex_);
    return map_.count(key) > 0;
  }

  IteratorType begin() { return map_mirror_.begin(); }

  IteratorType end() { return map_mirror_.end(); }

  ThreadSafeMap& mirror() {
    std::shared_lock lock(mutex_);
    map_mirror_ = map_;
    return *this;
  }

 private:
  MapType map_mirror_;
  MapType map_;
  mutable std::shared_mutex mutex_;
};

class GPGFRONTEND_CORE_EXPORT CacheManager
    : public QObject,
      public SingletonFunctionObject<CacheManager> {
  Q_OBJECT
 public:
  CacheManager(int channel = SingletonFunctionObject::GetDefaultChannel());

  void SaveCache(std::string key, const nlohmann::json& value);

  nlohmann::json LoadCache(std::string key);

  nlohmann::json LoadCache(std::string key, nlohmann::json default_value);

 private:
  std::string get_data_object_key(std::string key);

  nlohmann::json load_cache_storage(std::string key,
                                    nlohmann::json default_value);

  void load_all_cache_storage();

  void flush_cache_storage();

  void register_cache_key(std::string key);

  ThreadSafeMap<std::string, nlohmann::json> cache_storage_;
  nlohmann::json key_storage_;
  QTimer* m_timer_;
  const std::string drk_key_ = "__cache_manage_data_register_key_list";
};

}  // namespace GpgFrontend

#endif