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 <stdio.h>
#include "litesql/types.hpp"
#include "litesql/backend.hpp"
namespace litesql {
using namespace std;
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