Program Listing for File context.hpp

↰ Return to documentation for file (falcon/context.hpp)

// ---------------------------------------------------------------------
// This file is part of falcon-core.
//
// Copyright (C) 2015, 2016, 2017 Neuro-Electronics Research Flanders
//
// Falcon-server 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.
//
// Falcon-server 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 falcon-core. If not, see <http://www.gnu.org/licenses/>.
// ---------------------------------------------------------------------

#pragma once

#include <atomic>
#include <string>
#include <map>
#include <zmq.hpp>
#include "utilities/string.hpp"

class StorageContext {
 public:
  StorageContext(std::string default_context = "")
      : default_context_(default_context) {}

  std::map<std::string, std::string> storage_map() const {
    return storage_map_;
  }

  std::string default_context() const { return default_context_; }
  void set_default_context(std::string c) { default_context_ = c; }

  std::string storage_context(const std::string& c) { return storage_map_[c]; }

  std::string resolve_path(const std::string &p) const {
    return resolve_server_path(p, storage_map_, default_context_);
  }
  std::string resolve_path(const std::string &p,
                           std::string default_context) const {
    return resolve_server_path(p, storage_map_, default_context);
  }

 protected:
  void add_storage_context(std::string key, std::string value) {
    storage_map_[key] = value;
  }

 protected:
  std::string default_context_;
  std::map<std::string, std::string> storage_map_;
};

class GlobalContext : public StorageContext {
 public:
  GlobalContext(bool test_flag, const std::map<std::string, std::string> &uri)
      : StorageContext(""), default_test_flag_(test_flag) {
    zmq_context_ = zmq::context_t(1);
    for (auto &it : uri) {
      add_storage_context(it.first, it.second);
    }
  }

  zmq::context_t &zmq() { return zmq_context_; }

  bool test() const { return default_test_flag_.load(); }
  void set_test(bool value) { default_test_flag_ = value; }

 private:
  zmq::context_t zmq_context_;
  std::atomic<bool> default_test_flag_;
};