node_cache.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/graph/types.h>
 
#include <cstddef>
#include <mutex>
#include <optional>
#include <set>
#include <string>
#include <unordered_map>
 
namespace neograph::graph {
 
class NEOGRAPH_API NodeCache {
public:
    NodeCache() = default;
 
    void set_enabled(const std::string& node_name, bool enabled);
 
    bool is_enabled(const std::string& node_name) const;
 
    std::optional<NodeResult> lookup(const std::string& node_name,
                                     const std::string& state_hash) const;
 
    void store(const std::string& node_name,
               const std::string& state_hash,
               NodeResult result);
 
    void clear();
 
    std::size_t size() const;
 
    std::size_t hit_count() const;
    std::size_t miss_count() const;
 
private:
    static std::string make_key(const std::string& node_name,
                                const std::string& state_hash);
 
    mutable std::mutex mu_;
    std::set<std::string> enabled_nodes_;
    std::unordered_map<std::string, NodeResult> entries_;
    mutable std::size_t hits_   = 0;
    mutable std::size_t misses_ = 0;
};
 
NEOGRAPH_API std::string hash_state_for_cache(const json& state_value);
 
} // namespace neograph::graph