Cursor.h

Go to the documentation of this file.

#ifndef LINTEL_POSTGRESQL_CURSOR_H
#define LINTEL_POSTGRESQL_CURSOR_H
 
#include <libpq-fe.h>
 
#include <atomic>
#include <cstddef>
#include <string>
#include <vector>
 
#include "lintel/core/persistence/Arguments.h"
 
namespace lintel {
namespace postgresql {
  class Connection;
 
  class Cursor
  {
  private:
    PGconn *m_conn;
    bool m_active;
    bool m_serverSide;
    bool m_ownsTransaction;
    std::string m_cursorName;
    std::size_t m_fetchSize;
 
    void drain();
 
    std::vector<db::Arguments> fetchFromServer(std::size_t maxRows);
 
    void close();
 
  public:
    explicit Cursor(const Connection &connection, const std::string &query, const std::vector<std::string> &params);
 
    explicit Cursor(const Connection &connection,
      const std::string &query,
      const std::vector<std::string> &params,
      std::size_t fetchSize,
      const std::string &cursorName,
      bool noScroll);
 
    Cursor(Cursor &cursor) = delete;
 
    ~Cursor();
 
    [[nodiscard]] std::vector<db::Arguments> fetchNext(std::size_t maxRows);
  };
}// namespace postgresql
 
}// namespace lintel
#endif// LINTEL_POSTGRESQL_CURSOR_H