diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/database/async_result.hpp | 77 | ||||
-rw-r--r-- | src/database/select_query.hpp | 10 |
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; }; |