diff options
Diffstat (limited to 'src/database/count_query.hpp')
| -rw-r--r-- | src/database/count_query.hpp | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/src/database/count_query.hpp b/src/database/count_query.hpp index 0dde63c..e8d24ef 100644 --- a/src/database/count_query.hpp +++ b/src/database/count_query.hpp @@ -2,11 +2,10 @@ #include <database/query.hpp> #include <database/table.hpp> +#include <database/statement.hpp> #include <string> -#include <sqlite3.h> - struct CountQuery: public Query { CountQuery(std::string name): @@ -15,20 +14,17 @@ struct CountQuery: public Query this->body += std::move(name); } - int64_t execute(sqlite3* db) + int64_t execute(DatabaseEngine& db) { - auto statement = this->prepare(db); + auto statement = db.prepare(this->body); int64_t res = 0; - if (sqlite3_step(statement.get()) == SQLITE_ROW) - res = sqlite3_column_int64(statement.get(), 0); + if (statement->step() != StepResult::Error) + res = statement->get_column_int64(0); else { log_error("Count request didn’t return a result"); return 0; } - if (sqlite3_step(statement.get()) != SQLITE_DONE) - log_warning("Count request returned more than one result."); - return res; } }; |