Add boolean flag for drop db force

yaojiejia · yaojiejia · commit ebda7309b843 · 2025-06-08T20:19:33.000-04:00
diff --git a/src/sqlancer/postgres/PostgresProvider.java b/src/sqlancer/postgres/PostgresProvider.java
@@ -277,7 +277,8 @@ public SQLConnection createDatabase(PostgresGlobalState globalState) throws SQLE
         globalState.getState().logStatement(String.format("\\c %s;", entryDatabaseName));
         
         String dropCommand = "DROP DATABASE";
-        if (Randomly.getBoolean()) {
+        boolean forceDrop = Randomly.getBoolean();
+        if (forceDrop) {
             dropCommand += " FORCE";
         }
         dropCommand += " IF EXISTS " + databaseName;
@@ -287,7 +288,7 @@ public SQLConnection createDatabase(PostgresGlobalState globalState) throws SQLE
             s.execute(dropCommand);
         } catch (SQLException e) {
             // If force fails, fall back to regular drop
-            if (dropCommand.contains("FORCE")) {
+            if (forceDrop) {
                 String fallbackDrop = "DROP DATABASE IF EXISTS " + databaseName;
                 globalState.getState().logStatement(fallbackDrop + ";");
                 try (Statement s = con.createStatement()) {
