summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/dbRecordSet.cpp39
-rw-r--r--lib/dbRecordSet.h8
2 files changed, 47 insertions, 0 deletions
diff --git a/lib/dbRecordSet.cpp b/lib/dbRecordSet.cpp
index 3754130..d5e55f0 100644
--- a/lib/dbRecordSet.cpp
+++ b/lib/dbRecordSet.cpp
@@ -1,4 +1,5 @@
#include "dbRecordSet.h"
+#include "helpers.h"
namespace MyGrate {
RowView::RowView(const RecordSet * rs, std::size_t rw) : recordSet {rs}, row {rw} { }
@@ -9,6 +10,32 @@ namespace MyGrate {
return recordSet->at(row, col);
}
+ bool
+ RowView::operator==(const RowView & that) const
+ {
+ verify<std::logic_error>(this->recordSet == that.recordSet, "Cannot iterator between RecordSets");
+ return this->row == that.row;
+ }
+
+ RowView &
+ RowView::operator++()
+ {
+ row++;
+ return *this;
+ }
+
+ const RowView &
+ RowView::operator*() const
+ {
+ return *this;
+ }
+
+ std::size_t
+ RowView::currentRow() const
+ {
+ return row;
+ }
+
RowView
RecordSet::operator[](std::size_t row) const
{
@@ -20,4 +47,16 @@ namespace MyGrate {
{
return at(0, 0);
}
+
+ RowView
+ RecordSet::begin() const
+ {
+ return {this, 0};
+ }
+
+ RowView
+ RecordSet::end() const
+ {
+ return {this, rows()};
+ }
}
diff --git a/lib/dbRecordSet.h b/lib/dbRecordSet.h
index e430e58..a9da882 100644
--- a/lib/dbRecordSet.h
+++ b/lib/dbRecordSet.h
@@ -25,6 +25,11 @@ namespace MyGrate {
return std::make_unique<S>((*this)[I + O]...);
}
+ bool operator==(const RowView &) const;
+ RowView & operator++();
+ const RowView & operator*() const;
+ std::size_t currentRow() const;
+
private:
const RecordSet * recordSet;
std::size_t row;
@@ -39,6 +44,9 @@ namespace MyGrate {
virtual DbValue at(std::size_t, std::size_t) const = 0;
RowView operator[](std::size_t row) const;
DbValue operator*() const;
+
+ RowView begin() const;
+ RowView end() const;
};
using RecordSetPtr = std::unique_ptr<RecordSet>;
}