stream_iterator.hxx

Go to the documentation of this file.

 
#ifndef PQXX_INTERNAL_STREAM_ITERATOR_HXX
#define PQXX_INTERNAL_STREAM_ITERATOR_HXX
 
#include <memory>
 
namespace pqxx
{
template<typename... TYPE> class stream_query;
} // namespace pqxx
 
 
namespace pqxx::internal
{
 
template<typename... TYPE> class stream_from_input_iterator final
{
  using stream_t = stream_from;
 
public:
  using value_type = std::tuple<TYPE...>;
 
  stream_from_input_iterator() = default;
 
  explicit stream_from_input_iterator(stream_t &home, sl loc) : m_home(&home)
  {
    advance(loc);
  }
  stream_from_input_iterator(stream_from_input_iterator const &) = default;
  stream_from_input_iterator(stream_from_input_iterator &&) = default;
 
  stream_from_input_iterator &operator++()
  {
    advance(sl::current());
    return *this;
  }
 
  ~stream_from_input_iterator() = default;
 
  stream_from_input_iterator &
  operator=(stream_from_input_iterator const &) = default;
  stream_from_input_iterator &
  operator=(stream_from_input_iterator &&) = default;
 
  value_type const &operator*() const noexcept { return m_value; }
 
  bool operator==(stream_from_input_iterator const &rhs) const noexcept
  {
    return m_home == rhs.m_home;
  }
  bool operator!=(stream_from_input_iterator const &rhs) const noexcept
  {
    return not(*this == rhs);
  }
 
private:
  void advance(sl loc)
  {
    if (m_home == nullptr)
      throw usage_error{"Moving stream_from iterator beyond end().", loc};
    if (not((*m_home) >> m_value))
      m_home = nullptr;
  }
 
  stream_t *m_home{nullptr};
  value_type m_value;
};
 
 
// TODO: Use separate type for end().
// TODO: Replace with generator?
template<typename... TYPE> class stream_input_iteration final
{
public:
  using stream_t = stream_from;
  using iterator = stream_from_input_iterator<TYPE...>;
  explicit stream_input_iteration(stream_t &home) : m_home{home} {}
  [[nodiscard]] iterator begin(sl loc = sl::current()) const
  {
    return iterator{m_home, loc};
  }
  [[nodiscard]] iterator end() const { return {}; }
 
private:
  // NOLINTNEXTLINE(cppcoreguidelines-avoid-const-or-ref-data-members)
  stream_t &m_home;
};
} // namespace pqxx::internal
#endif