Checkpoint before follow-up message

Co-authored-by: contact <[email protected]>

Author: cursoragent

Cargo.lock

@@ -57,7 +57,7 @@ offline = ["sqlx-macros/offline", "sqlx-core/offline"]
 
 # intended mainly for CI and docs
 all = ["tls", "all-databases", "all-types"]
-all-databases = ["mysql", "sqlite", "postgres", "mssql", "any"]
+all-databases = ["mysql", "sqlite", "postgres", "mssql", "snowflake", "any"]
 all-types = [
     "bigdecimal",
     "decimal",
@@ -131,6 +131,7 @@ postgres = ["sqlx-core/postgres", "sqlx-macros/postgres"]
 mysql = ["sqlx-core/mysql", "sqlx-macros/mysql"]
 sqlite = ["sqlx-core/sqlite", "sqlx-macros/sqlite"]
 mssql = ["sqlx-core/mssql", "sqlx-macros/mssql"]
+snowflake = ["sqlx-core/snowflake", "sqlx-macros/snowflake"]
 
 # types
 bigdecimal = ["sqlx-core/bigdecimal", "sqlx-macros/bigdecimal"]

Cargo.toml

@@ -20,7 +20,7 @@ default = ["migrate"]
 migrate = ["sha2", "crc"]
 
 # databases
-all-databases = ["postgres", "mysql", "sqlite", "mssql", "any"]
+all-databases = ["postgres", "mysql", "sqlite", "mssql", "snowflake", "any"]
 postgres = [
     "md-5",
     "sha2",
@@ -45,6 +45,17 @@ mysql = [
 ]
 sqlite = ["libsqlite3-sys", "futures-executor", "flume"]
 mssql = ["uuid", "encoding_rs", "regex"]
+snowflake = [
+    "reqwest", 
+    "jsonwebtoken", 
+    "rsa", 
+    "sha2", 
+    "base64", 
+    "serde", 
+    "serde_json",
+    "uuid",
+    "chrono"
+]
 any = []
 
 # types
@@ -170,6 +181,8 @@ hashlink = "0.10.0"
 indexmap = "2.0.0"
 hkdf = { version = "0.12.0", optional = true }
 event-listener = "5.4.0"
+reqwest = { version = "0.12", optional = true, default-features = false, features = ["json", "rustls-tls"] }
+jsonwebtoken = { version = "9.3", optional = true }
 
 dotenvy = "0.15"
 

sqlx-core/Cargo.toml

@@ -83,7 +83,8 @@ pub mod migrate;
         feature = "postgres",
         feature = "mysql",
         feature = "mssql",
-        feature = "sqlite"
+        feature = "sqlite",
+        feature = "snowflake"
     ),
     feature = "any"
 ))]
@@ -105,6 +106,10 @@ pub mod mysql;
 #[cfg_attr(docsrs, doc(cfg(feature = "mssql")))]
 pub mod mssql;
 
+#[cfg(feature = "snowflake")]
+#[cfg_attr(docsrs, doc(cfg(feature = "snowflake")))]
+pub mod snowflake;
+
 // Implements test support with automatic DB management.
 #[cfg(feature = "migrate")]
 pub mod testing;

sqlx-core/src/lib.rs

@@ -0,0 +1,75 @@
+use crate::arguments::Arguments;
+use crate::encode::{Encode, IsNull};
+use crate::snowflake::{Snowflake, SnowflakeTypeInfo};
+use crate::types::Type;
+use std::borrow::Cow;
+
+/// Implementation of [`Arguments`] for Snowflake.
+#[derive(Debug, Default, Clone)]
+pub struct SnowflakeArguments {
+    // Store arguments as strings since Snowflake SQL API uses text protocol
+    pub(crate) bindings: Vec<String>,
+}
+
+/// Implementation of [`ArgumentBuffer`] for Snowflake.
+#[derive(Debug)]
+pub struct SnowflakeArgumentBuffer {
+    pub(crate) buffer: Vec<u8>,
+}
+
+impl SnowflakeArguments {
+    pub fn new() -> Self {
+        Self {
+            bindings: Vec::new(),
+        }
+    }
+
+    pub(crate) fn len(&self) -> usize {
+        self.bindings.len()
+    }
+
+    pub(crate) fn get(&self, index: usize) -> Option<&String> {
+        self.bindings.get(index)
+    }
+}
+
+impl<'q> Arguments<'q> for SnowflakeArguments {
+    type Database = Snowflake;
+
+    fn reserve(&mut self, additional: usize, _size: usize) {
+        self.bindings.reserve(additional);
+    }
+
+    fn add<T>(&mut self, value: T)
+    where
+        T: 'q + Encode<'q, Self::Database> + Type<Self::Database>,
+    {
+        let mut buffer = SnowflakeArgumentBuffer::new();
+        let is_null = value.encode_by_ref(&mut buffer);
+
+        match is_null {
+            IsNull::No => {
+                // Convert the encoded bytes to a string
+                let binding = String::from_utf8_lossy(&buffer.buffer).into_owned();
+                self.bindings.push(binding);
+            }
+            IsNull::Yes => {
+                self.bindings.push("NULL".to_string());
+            }
+        }
+    }
+}
+
+impl Default for SnowflakeArgumentBuffer {
+    fn default() -> Self {
+        Self {
+            buffer: Vec::new(),
+        }
+    }
+}
+
+impl SnowflakeArgumentBuffer {
+    pub fn new() -> Self {
+        Self::default()
+    }
+}

sqlx-core/src/snowflake/arguments.rs

@@ -0,0 +1,40 @@
+use crate::column::{Column, ColumnIndex};
+use crate::error::Error;
+use crate::snowflake::{Snowflake, SnowflakeTypeInfo};
+use std::borrow::Cow;
+
+/// Implementation of [`Column`] for Snowflake.
+#[derive(Debug, Clone)]
+pub struct SnowflakeColumn {
+    pub(crate) name: String,
+    pub(crate) type_info: SnowflakeTypeInfo,
+    pub(crate) ordinal: usize,
+}
+
+impl SnowflakeColumn {
+    pub(crate) fn new(name: String, type_info: SnowflakeTypeInfo, ordinal: usize) -> Self {
+        Self {
+            name,
+            type_info,
+            ordinal,
+        }
+    }
+}
+
+impl crate::column::private_column::Sealed for SnowflakeColumn {}
+
+impl Column for SnowflakeColumn {
+    type Database = Snowflake;
+
+    fn ordinal(&self) -> usize {
+        self.ordinal
+    }
+
+    fn name(&self) -> &str {
+        &self.name
+    }
+
+    fn type_info(&self) -> &SnowflakeTypeInfo {
+        &self.type_info
+    }
+}