postgres/typesense
1
0
Fork 0
mirror of https://github.com/typesense/typesense.git synced 2025-05-19 05:08:43 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add guards around metrics access.

Browse Source
This commit is contained in:
kishorenc 2021-02-19 13:11:57 +05:30
parent e5c3cddfcc
commit 206b555646
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
include/app_metrics.h
View File

@ -4,9 +4,12 @@
#include "json.hpp"
#include "logger.h"
#include <string>
#include <shared_mutex>
class AppMetrics {
private:
mutable std::shared_mutex mutex;
// stores last complete window
spp::sparse_hash_map<std::string, uint64_t>* counts;
spp::sparse_hash_map<std::string, uint64_t>* durations;
@ -44,14 +47,18 @@ public:
void operator=(AppMetrics const&) = delete;
void increment_count(const std::string& identifier, uint64_t count) {
std::unique_lock lock(mutex);
(*current_counts)[identifier] += count;
}
void increment_duration(const std::string& identifier, uint64_t duration) {
std::unique_lock lock(mutex);
(*current_durations)[identifier] += duration;
}
void window_reset() {
std::unique_lock lock(mutex);
delete counts;
counts = current_counts;
current_counts = new spp::sparse_hash_map<std::string, uint64_t>();
@ -61,7 +68,9 @@ public:
current_durations = new spp::sparse_hash_map<std::string, uint64_t>();
}
void get(const std::string& count_key, const std::string& latency_key, nlohmann::json &result) {
void get(const std::string& count_key, const std::string& latency_key, nlohmann::json &result) const {
std::shared_lock lock(mutex);
result[count_key] = nlohmann::json::object();
for(const auto& kv: *counts) {
result[count_key][kv.first] = (double(kv.second) / (METRICS_REFRESH_INTERVAL_MS / 1000));