xiaoxue_admin
/
dbx
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(core): guard BIT literal formatting behind MySQL-compatible db_type

This commit is contained in:
t8y2 2026-06-10 21:54:10 +08:00
parent 4389de386d
commit 8f92eeb638
2 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
crates/dbx-core/src/database_export.rs
View File

@ -839,7 +839,7 @@ mod tests {
assert_eq!(
statements,
vec!["INSERT INTO `flags` (`enabled`, `mask`, `label`) VALUES (b'1', b'1010', '1010'), (0, 3, 'off');"]
vec!["INSERT INTO `flags` (`enabled`, `mask`, `label`) VALUES (b'1', b'1010', '1010'), (b'0', 3, 'off');"]
);
}

33
crates/dbx-core/src/transfer.rs
View File

@ -478,24 +478,25 @@ pub fn escape_value_typed(val: &serde_json::Value, db_type: &DatabaseType, colum
}
}
},
serde_json::Value::Number(n) => {
if column_type.is_some_and(is_mysql_bit_type) {
format!("b'{}'", n.to_string())
} else {
n.to_string()
serde_json::Value::Number(n) => match db_type {
DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => {
if column_type.is_some_and(is_mysql_bit_type) {
format!("b'{}'", n.to_string())
} else {
n.to_string()
}
}
}
_ => n.to_string(),
},
serde_json::Value::String(s) => {
if column_type.is_some_and(is_mysql_bit_type) {
format!(
"b'{}'",
format_literal_string(s, db_type, column_type).replace('\\', "\\\\").replace('\'', "''")
)
} else {
format!(
"'{}'",
format_literal_string(s, db_type, column_type).replace('\\', "\\\\").replace('\'', "''")
)
let escaped = format_literal_string(s, db_type, column_type).replace('\\', "\\\\").replace('\'', "''");
match db_type {
DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks
if column_type.is_some_and(is_mysql_bit_type) =>
{
format!("b'{escaped}'")
}
_ => format!("'{escaped}'"),
}
}
serde_json::Value::Array(arr) => match db_type {