cursor.hpp

Go to the documentation of this file.

/* LiteSQL 
 * 
 * The list of contributors at http://litesql.sf.net/ 
 * 
 * See LICENSE for copyright information. */
 
#ifndef litesql_cursor_hpp
#define litesql_cursor_hpp
 
#include "litesql/backend.hpp"
#include "litesql/except.hpp"
 
namespace litesql {
class Database;
template <class T>
class Cursor {
private:
    const Database& db;
    Backend::Cursor * cursor;
    bool done;
    bool dataReady;
    Record currentRow;
public:
    Cursor(const Database& db, Backend::Cursor * c);
    ~Cursor();
    Cursor<T> & operator++();
    Cursor<T> & operator++(int) { return operator++();}
    std::vector<T> dump();
    T operator*();
    inline bool rowsLeft() { return !done; }
};
 
template <class T>
Cursor<T>::Cursor(const Database& db_, Backend::Cursor * c)
    : db(db_), cursor(c), done(false), dataReady(false) {
    operator++();
}
template <class T>
Cursor<T>::~Cursor() {
    delete cursor;
}
template <class T>
Cursor<T> & Cursor<T>::operator++() {
    if (done)
        return *this;
    currentRow = cursor->fetchOne();
    if (currentRow.size() == 0) {
        done = true;
        dataReady = false;
    }
    else
        dataReady = true;
        
    return *this;
}
 
template <class T>
std::vector<T> Cursor<T>::dump() {
    std::vector<T> res;
    for (;!done;operator++())
        res.push_back(operator*());
    return res;
}
template <class T>
T Cursor<T>::operator*() {
    Record rec;
 
    if (!dataReady)
        throw NotFound();
 
    return T(db,currentRow);
}
 
}
 
#endif  // #ifndef litesql_cursor_hpp