Postgres: only set force_generic_plan for versions which support it

launchbadge · Oct 5, 2024 · 59da05d · 59da05d
1 parent 5bba30d
commit 59da05d
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/sqlx-postgres/src/connection/describe.rs b/sqlx-postgres/src/connection/describe.rs
@@ -526,8 +526,19 @@ WHERE rngtypid = $1
         let mut tx = self.begin().await?;
 
         if params_len > 0 {
-            tx.execute("set local plan_cache_mode = force_generic_plan;")
-                .await?;
+            tx.execute(
+                " DO $$
+                    BEGIN
+                      IF EXISTS (
+                        SELECT 1
+                        FROM pg_settings
+                        WHERE name = 'plan_cache_mode'
+                      ) THEN
+                        SET LOCAL plan_cache_mode = 'force_generic_plan';
+                      END IF;
+                    END $$;",
+            )
+            .await?;
 
             explain += "(";