ConcurrentParam.h

#pragma once
 
#include "GlobalParam.h"
 
namespace iguana {

  using concurrent_key_t = std::size_t;
 
  // ==================================================================================
  // ConcurrentParam
  // ==================================================================================

  template <typename T>
  class ConcurrentParam {
 
    public:

      ConcurrentParam(std::string const& model);
      virtual ~ConcurrentParam() = default;

      virtual T const Load(concurrent_key_t const key) const = 0;

      virtual void Save(T const& value, concurrent_key_t const key) = 0;

      virtual bool HasKey(concurrent_key_t const key) const = 0;

      bool NeedsHashing() const { return m_needs_hashing; }

      virtual std::size_t GetSize() const = 0;

      bool IsEmpty() const { return m_empty; }
 
    protected:

      bool m_needs_hashing;

      std::mutex m_mutex;

      bool m_empty{true};
 
  };
 
  // ==================================================================================
  // SingleThreadParam
  // ==================================================================================

  template <typename T>
  class SingleThreadParam : public ConcurrentParam<T> {
 
    public:
      SingleThreadParam();
      ~SingleThreadParam() override = default;
      T const Load(concurrent_key_t const key) const override;
      void Save(T const& value, concurrent_key_t const key) override;
      bool HasKey(concurrent_key_t const key) const override;
      std::size_t GetSize() const override;
 
    private:

      T m_value;
 
  };
 
  // ==================================================================================
  // MemoizedParam
  // ==================================================================================

  template <typename T>
  class MemoizedParam : public ConcurrentParam<T> {

    using container_t = std::unordered_map<concurrent_key_t, T>;
 
    public:
      MemoizedParam();
      ~MemoizedParam() override = default;
      T const Load(concurrent_key_t const key) const override;
      void Save(T const& value, concurrent_key_t const key) override;
      bool HasKey(concurrent_key_t const key) const override;
      std::size_t GetSize() const override;
 
    private:

      container_t m_container;
 
  };
 
  // ==================================================================================
  // ConcurrentParamFactory
  // ==================================================================================

  class ConcurrentParamFactory {
 
    public:
      ConcurrentParamFactory() = delete;

      template <typename T>
      static std::unique_ptr<ConcurrentParam<T>> Create() {
 
        if(GlobalConcurrencyModel() == "none")
          GlobalConcurrencyModel = "memoize"; // the safest default, but not the fastest for single-threaded users
 
        if(GlobalConcurrencyModel() == "single")
          return std::make_unique<SingleThreadParam<T>>();
        else if(GlobalConcurrencyModel() == "memoize")
          return std::make_unique<MemoizedParam<T>>();
 
        throw std::runtime_error("unknown GlobalConcurrencyModel '" + GlobalConcurrencyModel() + "'; valid options are 'single' or 'memoize'");
      }
 
  };
 
}