summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/database/async_result.hpp77
-rw-r--r--src/database/select_query.hpp10
2 files changed, 87 insertions, 0 deletions
diff --git a/src/database/async_result.hpp b/src/database/async_result.hpp
new file mode 100644
index 0000000..f4109c0
--- /dev/null
+++ b/src/database/async_result.hpp
@@ -0,0 +1,77 @@
+#include <database/row.hpp>
+
+template <typename... T>
+class AsyncResult
+{
+ std::unique_ptr<Statement> statement{};
+ std::string table_name;
+
+public:
+ AsyncResult(std::unique_ptr<Statement> s, const std::string& table_name):
+ statement{std::move(s)},
+ table_name{table_name}
+ {}
+
+ class iterator
+ {
+ using iterator_category = std::input_iterator_tag;
+ using value_type = Row<T...>;
+ using difference_type = std::ptrdiff_t;
+ using pointer = Row<T...>*;
+ using reference = Row<T...>&;
+
+ Row<T...> row{};
+ Statement* statement;
+ bool is_end;
+
+ public:
+ iterator(Statement* s, const std::string& table_name, bool end=false):
+ row{table_name},
+ statement{s},
+ is_end{end}
+ {}
+
+ reference operator*()
+ {
+ extract_row_values(this->row, *statement);
+ return this->row;
+ }
+
+ bool operator==(const iterator& o) const
+ {
+ if (this->is_end && o.is_end)
+ return true;
+ return false;
+ }
+ bool operator!=(const iterator& o) const
+ {
+ return !(*this == o);
+ }
+
+ iterator& operator++()
+ {
+ if (statement->step() != StepResult::Row)
+ this->is_end = true;
+ return *this;
+ }
+
+ iterator& operator++(int)
+ {
+ iterator old = *this;
+ if (statement->step() != StepResult::Row)
+ this->is_end = true;
+ return old;
+ }
+ };
+
+ iterator begin() const
+ {
+ iterator it{this->statement.get(), this->table_name};
+ ++it;
+ return it;
+ }
+ iterator end() const
+ {
+ return {this->statement.get(), this->table_name, true};
+ }
+};
diff --git a/src/database/select_query.hpp b/src/database/select_query.hpp
index e372f2e..1399417 100644
--- a/src/database/select_query.hpp
+++ b/src/database/select_query.hpp
@@ -8,6 +8,7 @@
#include <database/query.hpp>
#include <logger/logger.hpp>
#include <database/row.hpp>
+#include <database/async_result.hpp>
#include <utils/optional_bool.hpp>
@@ -131,6 +132,15 @@ struct SelectQuery: public Query
return rows;
}
+ AsyncResult<T...> execute_async(DatabaseEngine& db)
+ {
+ auto statement = db.prepare(this->body);
+ if (!statement)
+ return {{}, {}};
+ statement->bind(std::move(this->params));
+ return {std::move(statement), this->table_name};
+ }
+
const std::string table_name;
};