Import upstream v0.16.22, stripped

Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
2026-09-18 10:21:56 -07:00
commit 7dae9b29fd
1650 changed files with 485521 additions and 0 deletions
@@ -0,0 +1,136 @@
+/*
+ * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
+ */
+
+use mysql_async::{Params, Row, prelude::Queryable};
+
+use crate::{IntoRows, QueryResult, QueryType, Value};
+
+use super::{MysqlStore, into_error};
+
+impl MysqlStore {
+    pub(crate) async fn sql_query<T: QueryResult>(
+        &self,
+        query: &str,
+        params: &[Value<'_>],
+    ) -> trc::Result<T> {
+        let mut conn = self.conn_pool.get_conn().await.map_err(into_error)?;
+        let s = conn.prep(query).await.map_err(into_error)?;
+        let params = Params::Positional(params.iter().map(Into::into).collect());
+
+        match T::query_type() {
+            QueryType::Execute => conn.exec_drop(s, params).await.map_or_else(
+                |e| Err(into_error(e)),
+                |_| Ok(T::from_exec(conn.affected_rows() as usize)),
+            ),
+            QueryType::Exists => conn
+                .exec_first::<Row, _, _>(s, params)
+                .await
+                .map_or_else(|e| Err(into_error(e)), |r| Ok(T::from_exists(r.is_some()))),
+            QueryType::QueryOne => conn
+                .exec_first::<Row, _, _>(s, params)
+                .await
+                .map_or_else(|e| Err(into_error(e)), |r| Ok(T::from_query_one(r))),
+            QueryType::QueryAll => conn
+                .exec::<Row, _, _>(s, params)
+                .await
+                .map_or_else(|e| Err(into_error(e)), |r| Ok(T::from_query_all(r))),
+        }
+    }
+}
+
+impl From<crate::Value<'_>> for mysql_async::Value {
+    fn from(value: crate::Value) -> Self {
+        match value {
+            crate::Value::Integer(i) => mysql_async::Value::Int(i),
+            crate::Value::Bool(b) => mysql_async::Value::Int(b as i64),
+            crate::Value::Float(f) => mysql_async::Value::Double(f),
+            crate::Value::Text(t) => mysql_async::Value::Bytes(t.into_owned().into_bytes()),
+            crate::Value::Blob(b) => mysql_async::Value::Bytes(b.into_owned()),
+            crate::Value::Null => mysql_async::Value::NULL,
+        }
+    }
+}
+
+impl From<mysql_async::Value> for crate::Value<'static> {
+    fn from(value: mysql_async::Value) -> Self {
+        match value {
+            mysql_async::Value::Int(i) => Self::Integer(i),
+            mysql_async::Value::UInt(i) => Self::Integer(i as i64),
+            mysql_async::Value::Double(f) => Self::Float(f),
+            mysql_async::Value::Bytes(b) => String::from_utf8(b).map_or_else(
+                |e| Self::Blob(e.into_bytes().into()),
+                |s| Self::Text(s.into()),
+            ),
+            mysql_async::Value::NULL => Self::Null,
+            mysql_async::Value::Float(f) => Self::Float(f as f64),
+            mysql_async::Value::Date(_, _, _, _, _, _, _)
+            | mysql_async::Value::Time(_, _, _, _, _, _) => Self::Text(value.as_sql(true).into()),
+        }
+    }
+}
+
+impl IntoRows for Vec<mysql_async::Row> {
+    fn into_rows(self) -> crate::Rows {
+        crate::Rows {
+            rows: self
+                .into_iter()
+                .map(|r| crate::Row {
+                    values: r
+                        .unwrap_raw()
+                        .into_iter()
+                        .flatten()
+                        .map(Into::into)
+                        .collect(),
+                })
+                .collect(),
+        }
+    }
+
+    fn into_named_rows(self) -> crate::NamedRows {
+        crate::NamedRows {
+            names: self
+                .first()
+                .map(|r| r.columns().iter().map(|c| c.name_str().into()).collect())
+                .unwrap_or_default(),
+            rows: self
+                .into_iter()
+                .map(|r| crate::Row {
+                    values: r
+                        .unwrap_raw()
+                        .into_iter()
+                        .flatten()
+                        .map(Into::into)
+                        .collect(),
+                })
+                .collect(),
+        }
+    }
+
+    fn into_row(self) -> Option<crate::Row> {
+        unreachable!()
+    }
+}
+
+impl IntoRows for Option<mysql_async::Row> {
+    fn into_row(self) -> Option<crate::Row> {
+        self.map(|row| crate::Row {
+            values: row
+                .unwrap_raw()
+                .into_iter()
+                .flatten()
+                .map(Into::into)
+                .collect(),
+        })
+    }
+
+    fn into_rows(self) -> crate::Rows {
+        unreachable!()
+    }
+
+    fn into_named_rows(self) -> crate::NamedRows {
+        unreachable!()
+    }
+}