From ec879337af01a1fa79514fdf6478decc3f42de42 Mon Sep 17 00:00:00 2001
From: Thom Chiovoloni <tchiovoloni@mozilla.com>
Date: Thu, 24 Jan 2019 21:43:39 -0800
Subject: [PATCH] Add support for querying sqlite_stmt_status

---
 src/lib.rs           |  2 +-
 src/raw_statement.rs |  6 ++++++
 src/statement.rs     | 37 +++++++++++++++++++++++++++++++++++++
 3 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index 40434f9..5611637 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -90,7 +90,7 @@ use crate::error::{error_from_handle, error_from_sqlite_code};
 use crate::raw_statement::RawStatement;
 use crate::types::{ToSql, ValueRef};
 
-pub use crate::statement::Statement;
+pub use crate::statement::{Statement, StatementStatus};
 
 pub use crate::row::{AndThenRows, MappedRows, Row, RowIndex, Rows};
 
diff --git a/src/raw_statement.rs b/src/raw_statement.rs
index b9d405d..8e3b5b4 100644
--- a/src/raw_statement.rs
+++ b/src/raw_statement.rs
@@ -1,5 +1,6 @@
 use super::ffi;
 use super::unlock_notify;
+use super::StatementStatus;
 use std::ffi::CStr;
 use std::os::raw::c_int;
 use std::ptr;
@@ -100,6 +101,11 @@ impl RawStatement {
             }
         }
     }
+
+    pub fn get_status(&self, status: StatementStatus, reset: bool) -> i32 {
+        assert!(!self.0.is_null());
+        unsafe { ffi::sqlite3_stmt_status(self.0, status as i32, reset as i32) }
+    }
 }
 
 impl Drop for RawStatement {
diff --git a/src/statement.rs b/src/statement.rs
index e2445a9..efda864 100644
--- a/src/statement.rs
+++ b/src/statement.rs
@@ -569,6 +569,17 @@ impl<'conn> Statement<'conn> {
                 .map(|s| str::from_utf8_unchecked(s.to_bytes()))
         }
     }
+
+    /// Get the value for one of the status counters for this statement.
+    pub fn get_status(&self, status: StatementStatus) -> i32 {
+        self.stmt.get_status(status, false)
+    }
+
+    /// Reset the value of one of the status counters for this statement,
+    /// returning the value it had before resetting.
+    pub fn reset_status(&self, status: StatementStatus) -> i32 {
+        self.stmt.get_status(status, true)
+    }
 }
 
 impl<'conn> Into<RawStatement> for Statement<'conn> {
@@ -670,6 +681,32 @@ impl<'conn> Statement<'conn> {
     }
 }
 
+/// Prepared statement status counters.
+///
+/// See https://www.sqlite.org/c3ref/c_stmtstatus_counter.html
+/// for explanations of each.
+///
+/// Note that depending on your version of SQLite, all of these
+/// may not be available.
+#[repr(i32)]
+#[derive(Clone, Copy, PartialEq, Eq)]
+pub enum StatementStatus {
+    /// Equivalent to SQLITE_STMTSTATUS_FULLSCAN_STEP
+    FullscanStep = 1,
+    /// Equivalent to SQLITE_STMTSTATUS_SORT
+    Sort = 2,
+    /// Equivalent to SQLITE_STMTSTATUS_AUTOINDEX
+    AutoIndex = 3,
+    /// Equivalent to SQLITE_STMTSTATUS_VM_STEP
+    VmStep = 4,
+    /// Equivalent to SQLITE_STMTSTATUS_REPREPARE
+    RePrepare = 5,
+    /// Equivalent to SQLITE_STMTSTATUS_RUN
+    Run = 6,
+    /// Equivalent to SQLITE_STMTSTATUS_MEMUSED
+    MemUsed = 99,
+}
+
 #[cfg(test)]
 mod test {
     use crate::{Connection, Error, Result, NO_PARAMS};