document.h

Go to the documentation of this file.

#pragma once

#include "element.h"
#include "typeinfo"
#include <initializer_list>
#include <map>
#include <set>
#include <string>
#include <sstream>
#include <ostream>

namespace bson
{
  
  class Document
  {     
    private:
      friend class Element;
      std::map<std::string, Element> m_data;
      std::set<std::string> m_field_names;
      
    public:
      Document() = default;
      Document(std::initializer_list<std::pair<std::string, Element>> list){for (auto i: list) {m_data.emplace(i.first, i.second); m_field_names.emplace(i.first);}}
      const Element& operator[] (const std::string & index) const {return m_data.at(index);}
      Element& operator[] (const std::string & index) {return m_data.at(index);}
      
      void add (const std::string name, const Element& data) {m_data.emplace(name, data); m_field_names.insert(name);}
      template <class T>
      void add (const std::string name, const T& data, const TypeInfo ti = _UNKNOWN) {m_data[name] = Element(data, ti); m_field_names.insert(name);}
      
      void set(const std::string name, const Element& data) {m_data.at(name) = data;}
      template <class T>
      void set(const std::string & name, const T& data, const TypeInfo ti = _UNKNOWN){m_data.at(name) = Element(data, ti);}
      
      friend std::ostream& operator << (std::ostream& o, const Document& d);
      
      auto begin() -> decltype(m_data.begin()) {return m_data.begin();}
      auto end() -> decltype(m_data.end()) {return m_data.end();}
      auto rbegin() -> decltype(m_data.rbegin()) {return m_data.rbegin();}
      auto rend() -> decltype(m_data.rend()) {return m_data.rend();}
      auto cbegin() -> decltype(m_data.cbegin()) {return m_data.cbegin();}
      auto cend() -> decltype(m_data.cend()) {return m_data.cbegin();}
      auto crbegin() -> decltype(m_data.crbegin()) {return m_data.crbegin();}
      auto crend() -> decltype(m_data.crend()) {return m_data.crend();}
      
      std::set<std::string> field_names() const {return m_field_names;}
  };

  inline std::ostream& operator << (std::ostream& oss, const Document& d)
  {
    bool first = true;
    oss << "{ ";
    for (std::pair<std::string, Element> p: d.m_data)
    {
      if (!first)
      {
    oss << ", ";
      }
      first = false;
      oss << p.first << " : " << static_cast<std::string>(p.second);
    }
    oss << " }";
    return oss;
  }
}