use super::base::DatabaseBackend;
#[derive(Debug, Default, Clone, Copy)]
pub struct MySqlBackend;
#[must_use]
fn parse_optional_length(field_type: &str, base: &str) -> Option<Option<usize>> {
let field_type = field_type.trim();
if field_type == base {
return Some(None);
}
let rest = field_type.strip_prefix(base)?;
let length = rest.strip_prefix('(')?.strip_suffix(')')?.parse().ok()?;
Some(Some(length))
}
impl DatabaseBackend for MySqlBackend {
fn name(&self) -> &str {
"mysql"
}
fn quote_name(&self, name: &str) -> String {
format!("`{}`", name.replace('`', "``"))
}
fn supports_partial_indexes(&self) -> bool {
false
}
fn db_type(&self, field_type: &str) -> String {
if let Some(length) = parse_optional_length(field_type, "CharField") {
let length = length.unwrap_or(255);
return format!("varchar({length})");
}
match field_type.trim() {
"AutoField" => "integer AUTO_INCREMENT".to_string(),
"BigAutoField" => "bigint AUTO_INCREMENT".to_string(),
"IntegerField" | "PositiveIntegerField" | "PositiveSmallIntegerField" => {
"integer".to_string()
}
"BigIntegerField" | "PositiveBigIntegerField" => "bigint".to_string(),
"SmallIntegerField" | "SmallAutoField" => "smallint".to_string(),
"BooleanField" | "NullBooleanField" => "bool".to_string(),
"FloatField" => "double precision".to_string(),
"DateTimeField" => "datetime(6)".to_string(),
"TextField" => "longtext".to_string(),
"JSONField" => "json".to_string(),
"UUIDField" => "char(32)".to_string(),
"BinaryField" => "longblob".to_string(),
other => other.to_string(),
}
}
fn last_insert_id_sql(&self, _table: &str, _pk_name: &str) -> String {
"SELECT LAST_INSERT_ID()".to_string()
}
}
impl MySqlBackend {
#[must_use]
pub fn supports_json(&self) -> bool {
self.supports_json_field()
}
#[must_use]
pub fn max_query_params(&self) -> usize {
65_535
}
#[must_use]
pub fn supports_returning(&self) -> bool {
false
}
}
#[cfg(test)]
mod tests {
use super::{DatabaseBackend, MySqlBackend};
#[test]
fn mysql_backend_reports_feature_flags() {
let backend = MySqlBackend;
assert_eq!(backend.name(), "mysql");
assert!(backend.supports_transactions());
assert!(backend.supports_json());
assert_eq!(backend.max_query_params(), 65_535);
assert!(!backend.supports_returning());
assert!(!backend.supports_partial_indexes());
assert!(!backend.supports_covering_indexes());
}
#[test]
fn test_mysql_quote_name() {
let backend = MySqlBackend;
assert_eq!(backend.quote_name("table_name"), "`table_name`");
assert_eq!(backend.quote_name("table`name"), "`table``name`");
}
#[test]
fn test_mysql_db_type_mappings() {
let backend = MySqlBackend;
assert_eq!(backend.db_type("AutoField"), "integer AUTO_INCREMENT");
assert_eq!(backend.db_type("CharField"), "varchar(255)");
assert_eq!(backend.db_type("CharField(64)"), "varchar(64)");
assert_eq!(backend.db_type("BooleanField"), "bool");
assert_eq!(backend.db_type("DateTimeField"), "datetime(6)");
assert_eq!(backend.db_type("JSONField"), "json");
assert_eq!(backend.db_type("UUIDField"), "char(32)");
assert_eq!(backend.db_type("BinaryField"), "longblob");
}
#[test]
fn test_mysql_last_insert_id() {
let backend = MySqlBackend;
assert_eq!(
backend.last_insert_id_sql("widgets", "id"),
"SELECT LAST_INSERT_ID()"
);
}
#[test]
fn test_mysql_no_partial_indexes() {
let backend = MySqlBackend;
assert!(!backend.supports_partial_indexes());
}
}