Make RecordSet foreach ready

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>;
 }