expose sqlite_changes function

Author: codekeyz

src/lib.rs

@@ -89,6 +89,7 @@ pub unsafe extern "C" fn sqlite3_open_v2(
         connection: connection.unwrap(),
         transaction_baton: Mutex::new(None),
         last_insert_rowid: Mutex::new(None),
+        rows_written: Mutex::new(None),
         transaction_has_began: Mutex::new(false),
         delete_hook: Mutex::new(None),
         insert_hook: Mutex::new(None),
@@ -383,6 +384,22 @@ pub unsafe extern "C" fn sqlite3_extended_errcode(_: *mut SQLite3) -> c_int {
     SQLITE_OK
 }
 
+#[no_mangle]
+pub unsafe extern "C" fn sqlite3_changes(db: *mut SQLite3) -> c_int {
+    if !is_aligned(db) {
+        return SQLITE_OK;
+    }
+    let db = &mut *db;
+
+    if let Ok(rows_written) = db.rows_written.lock() {
+        if rows_written.is_some() {
+            return rows_written.unwrap() as c_int;
+        }
+    }
+
+    0
+}
+
 #[no_mangle]
 pub unsafe extern "C" fn sqlite3_errmsg(_: *mut SQLite3) -> *const c_char {
     if let Some(error_entry) = sqlite::get_latest_error() {

src/sqlite.rs

@@ -86,6 +86,7 @@ lazy_static! {
 pub struct SQLite3 {
     pub connection: transport::DatabaseConnection, // Connection to the database
     pub last_insert_rowid: Mutex<Option<i64>>,     // Last inserted row ID
+    pub rows_written: Mutex<Option<u64>>,          // Number of rows written
     pub transaction_baton: Mutex<Option<String>>,  // Baton for transaction management
     pub transaction_has_began: Mutex<bool>,        // Flag to check if a transaction has started
     pub update_hook: Mutex<Option<(SqliteHook, *mut c_void)>>, // Update hook callback

src/transport/mod.rs

@@ -51,6 +51,8 @@ pub struct RemoteRow {
 pub struct QueryResult {
     pub cols: Vec<RemoteCol>,
     pub rows: Vec<Vec<RemoteRow>>,
+    pub rows_read: Option<u64>,
+    pub rows_written: Option<u64>,
     pub last_insert_rowid: Option<String>,
 }
 

src/utils.rs

@@ -144,5 +144,10 @@ pub fn get_execution_result<'a>(
         *last_insert_rowid_lock = Some(last_insert_rowid.parse::<i64>().unwrap_or(0));
     }
 
+    if let Some(rows_written) = &first_execution_result.rows_written {
+        let mut rows_written_lock = db.rows_written.lock().unwrap();
+        *rows_written_lock = Some(*rows_written);
+    }
+
     Ok(first_execution_result)
 }