From 50cadf3dac0d56ef8181d1800cc30f8dcb749141 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?louiz=E2=80=99?= <louiz@louiz.org>
Date: Tue, 13 Jun 2017 10:38:39 +0200
Subject: Implement our own database ORM, and update the whole code to use it

Entirely replace LiteSQL

fix #3271
---
 src/database/count_query.hpp | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 src/database/count_query.hpp

(limited to 'src/database/count_query.hpp')

diff --git a/src/database/count_query.hpp b/src/database/count_query.hpp
new file mode 100644
index 0000000..863fad1
--- /dev/null
+++ b/src/database/count_query.hpp
@@ -0,0 +1,35 @@
+#pragma once
+
+#include <database/query.hpp>
+#include <database/table.hpp>
+
+#include <string>
+
+#include <sqlite3.h>
+
+struct CountQuery: public Query
+{
+    CountQuery(std::string name):
+        Query("SELECT count(*) FROM ")
+    {
+      this->body += std::move(name);
+    }
+
+    std::size_t execute(sqlite3* db)
+    {
+      auto statement = this->prepare(db);
+      std::size_t res = 0;
+      if (sqlite3_step(statement) == SQLITE_ROW)
+        res = sqlite3_column_int64(statement, 0);
+      else
+        {
+          log_error("Count request didn’t return a result");
+          return 0;
+        }
+      if (sqlite3_step(statement) != SQLITE_DONE)
+        log_warning("Count request returned more than one result.");
+
+      log_debug("Returning count: ", res);
+      return res;
+    }
+};
-- 
cgit v1.2.3