Merge pull request #144 from RenathoAzevedo/master

Function to return the count of database changes.

Author: zauguin

hdr/sqlite_modern_cpp.h

@@ -80,7 +80,7 @@ namespace sqlite {
 			_inx(other._inx), execution_started(other.execution_started) { }
 
 		void execute();
-		
+
 		std::string sql() {
 #if SQLITE_VERSION_NUMBER >= 3014000
 			auto sqlite_deleter = [](void *ptr) {sqlite3_free(ptr);};
@@ -101,7 +101,7 @@ namespace sqlite {
 				_next_index();
 				--_inx;
 			}
-			execution_started = state; 
+			execution_started = state;
 		}
 		bool used() const { return execution_started; }
 		row_iterator begin();
@@ -445,7 +445,7 @@ namespace sqlite {
 				Values&&...      values
 		);
 	}
-	
+
 	enum class OpenFlags {
 		READONLY = SQLITE_OPEN_READONLY,
 		READWRITE = SQLITE_OPEN_READWRITE,
@@ -521,6 +521,10 @@ namespace sqlite {
 			return sqlite3_last_insert_rowid(_db.get());
 		}
 
+		int rows_modified() const {
+			return sqlite3_changes(_db.get());
+		}
+
 		template <typename Function>
 		void define(const std::string &name, Function&& func) {
 			typedef utility::function_traits<Function> traits;

tests/simple_examples.cc

@@ -3,7 +3,7 @@
 #include <cstdlib>
 #include <sqlite_modern_cpp.h>
 #include <catch.hpp>
-using namespace  sqlite;
+using namespace sqlite;
 using namespace std;
 
 TEST_CASE("simple examples", "[examples]") {
@@ -23,5 +23,14 @@ TEST_CASE("simple examples", "[examples]") {
     db << sql >> test;
 
     REQUIRE(test == 2);
+    
+    db << "UPDATE foo SET b=? WHERE a=?;" << "hi" << 1L;
+    db << "SELECT b FROM foo WHERE a=?;" << 1L >> str;
 
+    REQUIRE(str == "hi");
+    REQUIRE(db.rows_modified() == 1);
+    
+    db << "UPDATE foo SET b=?;" << "hello world";
+
+    REQUIRE(db.rows_modified() == 2);
 }