rebase main

Author: joeydewaal

sqlx-core/src/executor.rs

@@ -182,3 +182,76 @@ pub trait Executor<'c>: Send + Debug + Sized {
     where
         'c: 'e;
 }
+
+/// A type that may be executed against a database connection.
+///
+/// Implemented for the following:
+///
+///  * [`&str`](std::str)
+///  * [`Query`](super::query::Query)
+///
+pub trait Execute<'q, DB: Database>: Send + Sized {
+    /// Gets the SQL that will be executed.
+    fn sql(&self) -> &'q str;
+
+    /// Gets the previously cached statement, if available.
+    fn statement(&self) -> Option<&DB::Statement<'q>>;
+
+    /// Returns the arguments to be bound against the query string.
+    ///
+    /// Returning `Ok(None)` for `Arguments` indicates to use a "simple" query protocol and to not
+    /// prepare the query. Returning `Ok(Some(Default::default()))` is an empty arguments object that
+    /// will be prepared (and cached) before execution.
+    ///
+    /// Returns `Err` if encoding any of the arguments failed.
+    fn take_arguments(&mut self) -> Result<Option<<DB as Database>::Arguments<'q>>, BoxDynError>;
+
+    /// Returns `true` if the statement should be cached.
+    fn persistent(&self) -> bool;
+}
+
+// NOTE: `Execute` is explicitly not implemented for String and &String to make it slightly more
+//       involved to write `conn.execute(format!("SELECT {val}"))`
+impl<'q, DB: Database> Execute<'q, DB> for &'q str {
+    #[inline]
+    fn sql(&self) -> &'q str {
+        self
+    }
+
+    #[inline]
+    fn statement(&self) -> Option<&DB::Statement<'q>> {
+        None
+    }
+
+    #[inline]
+    fn take_arguments(&mut self) -> Result<Option<<DB as Database>::Arguments<'q>>, BoxDynError> {
+        Ok(None)
+    }
+
+    #[inline]
+    fn persistent(&self) -> bool {
+        true
+    }
+}
+
+impl<'q, DB: Database> Execute<'q, DB> for (&'q str, Option<<DB as Database>::Arguments<'q>>) {
+    #[inline]
+    fn sql(&self) -> &'q str {
+        self.0
+    }
+
+    #[inline]
+    fn statement(&self) -> Option<&DB::Statement<'q>> {
+        None
+    }
+
+    #[inline]
+    fn take_arguments(&mut self) -> Result<Option<<DB as Database>::Arguments<'q>>, BoxDynError> {
+        Ok(self.1.take())
+    }
+
+    #[inline]
+    fn persistent(&self) -> bool {
+        true
+    }
+}