caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix schema search path

Browse Source
This commit is contained in:
Chris Tsang 2024-12-24 00:23:05 +00:00
parent 9deef4f9af
commit 64582d1f4e
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/driver/sqlx_postgres.rs
View File

@ -78,14 +78,18 @@ impl SqlxPostgresConnector {
}
let set_search_path_sql = options.schema_search_path.as_ref().map(|schema| {
let mut string = "SET search_path = ".to_owned();
for (i, schema) in schema.split(',').enumerate() {
if i > 0 {
write!(&mut string, ",").unwrap();
}
if schema.starts_with('"') {
write!(&mut string, "{schema}").unwrap();
} else {
write!(&mut string, "\"{schema}\"").unwrap();
if schema.starts_with('"') {
write!(&mut string, "{schema}").unwrap();
} else {
for (i, schema) in schema.split(',').enumerate() {
if i > 0 {
write!(&mut string, ",").unwrap();
}
if schema.starts_with('"') {
write!(&mut string, "{schema}").unwrap();
} else {
write!(&mut string, "\"{schema}\"").unwrap();
}
}
}
string