thread_context.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include <string>
#include <chrono>
#include <optional>
#include <unordered_map>
#include <memory>
 
namespace kcenon { namespace monitoring {
 
struct context_metadata {
    std::string request_id;       
    std::string correlation_id;   
    std::string user_id;          
    std::unordered_map<std::string, std::string> tags; 
 
    explicit context_metadata(std::string req_id = "")
        : request_id(std::move(req_id)) {}
 
    bool empty() const {
        return request_id.empty() && correlation_id.empty() && user_id.empty() && tags.empty();
    }
 
    void set_tag(const std::string& key, const std::string& value) {
        tags[key] = value;
    }
 
    std::string get_tag(const std::string& key) const {
        auto it = tags.find(key);
        return it != tags.end() ? it->second : "";
    }
};
 
struct thread_context_data {
    std::string request_id;       
    std::string correlation_id;   
    std::string user_id;          
    std::string span_id;          
    std::string trace_id;         
    std::chrono::steady_clock::time_point start_time; 
    std::optional<std::string> parent_span_id; 
    std::unordered_map<std::string, std::string> tags; 
 
    thread_context_data() : start_time(std::chrono::steady_clock::now()) {}
 
    explicit thread_context_data(std::string req_id)
        : request_id(std::move(req_id))
        , start_time(std::chrono::steady_clock::now()) {}
 
    void add_tag(const std::string& key, const std::string& value) {
        tags[key] = value;
    }
 
    std::string get_tag(const std::string& key) const {
        auto it = tags.find(key);
        return it != tags.end() ? it->second : "";
    }
};
 
class thread_context {
public:
    static thread_context_data& create(const std::string& request_id = "");
 
    static thread_context_data* current();
 
    static bool has_context();
 
    static void clear();
 
    static std::string generate_request_id();
 
    static std::string generate_correlation_id();
 
    static bool copy_from(const thread_context_data& source);
 
private:
    static thread_local std::unique_ptr<thread_context_data> current_context_;
};
 
} } // namespace kcenon::monitoring