1 files changed, 123 insertions, 0 deletions

@@ -0,0 +1,123 @@
+#ifndef QUERY_H_7CC5284C
+#define QUERY_H_7CC5284C
+
+#include <vector>
+#include <stdexcept>
+#include <string>
+#include <list>
+#include <sqlite3.h>
+#include <iostream>
+#include "statement.h"
+#include "binding.h"
+
+namespace verbly {
+  
+  class database_error : public std::logic_error {
+  public:
+    
+    database_error(std::string msg, std::string sqlMsg) : std::logic_error(msg + " (" + sqlMsg + ")")
+    {
+    }
+  };
+  
+  template <typename Object>
+  class query {
+  public:
+    
+    query(const database& db, sqlite3* ppdb, filter queryFilter, bool random, int limit) : db_(&db)
+    {
+      statement stmt(Object::objectType, std::move(queryFilter));
+      
+      std::string queryString = stmt.getQueryString(Object::select, random, limit);
+      std::list<binding> bindings = stmt.getBindings();
+      
+      std::cout << queryString << std::endl;
+      
+      if (sqlite3_prepare_v2(ppdb, queryString.c_str(), queryString.length(), &ppstmt_, NULL) != SQLITE_OK)
+      {
+        std::string errorMsg = sqlite3_errmsg(ppdb);
+        sqlite3_finalize(ppstmt_);
+        
+        throw database_error("Error preparing query", errorMsg);
+      }
+      
+      int i = 1;
+      for (const binding& value : bindings)
+      {
+        switch (value.getType())
+        {
+          case binding::type::integer:
+          {
+            if (sqlite3_bind_int(ppstmt_, i, value.getInteger()) != SQLITE_OK)
+            {
+              std::string errorMsg = sqlite3_errmsg(ppdb);
+              sqlite3_finalize(ppstmt_);
+              
+              throw database_error("Error binding value to query", errorMsg);
+            }
+          
+            break;
+          }
+        
+          case binding::type::string:
+          {
+            if (sqlite3_bind_text(ppstmt_, i, value.getString().c_str(), value.getString().length(), SQLITE_TRANSIENT) != SQLITE_OK)
+            {
+              std::string errorMsg = sqlite3_errmsg(ppdb);
+              sqlite3_finalize(ppstmt_);
+              
+              throw database_error("Error binding value to query", errorMsg);
+            }
+          
+            break;
+          }
+          
+          case binding::type::invalid:
+          {
+            throw std::logic_error("Cannot use invalid bindings");
+          }
+        }
+      
+        i++;
+      }
+    }
+    
+    ~query()
+    {
+      sqlite3_finalize(ppstmt_);
+    }
+    
+    std::vector<Object> all() const
+    {
+      std::vector<Object> result;
+      
+      while (sqlite3_step(ppstmt_) == SQLITE_ROW)
+      {
+        result.emplace_back(*db_, ppstmt_);
+      }
+      
+      sqlite3_reset(ppstmt_);
+      
+      return result;
+    }
+
+    Object first() const
+    {
+      std::vector<Object> results = all();
+      if (!results.empty())
+      {
+        return results.front();
+      } else {
+        throw std::logic_error("query returned empty dataset");
+      }
+    }
+    
+  private:
+    const database* db_;
+    sqlite3_stmt* ppstmt_;
+    
+  };
+  
+};
+
+#endif /* end of include guard: QUERY_H_7CC5284C */