Fix schema search path

SeaQL · Dec 24, 2024 · 64582d1 · 64582d1
1 parent 9deef4f
commit 64582d1
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/src/driver/sqlx_postgres.rs b/src/driver/sqlx_postgres.rs
@@ -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