Add a method to specify whether a db creation action should be retried

mrigger · mrigger · commit 6e85452c567d · 2020-07-14T13:39:45.000+02:00
diff --git a/src/sqlancer/AbstractAction.java b/src/sqlancer/AbstractAction.java
@@ -6,4 +6,14 @@ public interface AbstractAction<G> {
 
     Query getQuery(G globalState) throws SQLException;
 
+    /**
+     * Specifies whether it makes sense to request a {@link Query}, when the previous call to {@link #getQuery(Object)}
+     * returned a query that failed executing.
+     *
+     * @return whether retrying getting queries makes sense, if the first query failed executing.
+     */
+    default boolean canBeRetried() {
+        return true;
+    }
+
 }
diff --git a/src/sqlancer/StatementExecutor.java b/src/sqlancer/StatementExecutor.java
@@ -65,7 +65,8 @@ public void executeStatements() throws SQLException {
                 do {
                     query = nextAction.getQuery(globalState);
                     success = globalState.executeStatement(query);
-                } while (!success && nrTries++ < globalState.getOptions().getNrStatementRetryCount());
+                } while (nextAction.canBeRetried() && !success
+                        && nrTries++ < globalState.getOptions().getNrStatementRetryCount());
             } catch (IgnoreMeException e) {
 
             }

Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,8 @@ public void executeStatements() throws SQLException {`
`65`	`65`	`do {`
`66`	`66`	`query = nextAction.getQuery(globalState);`
`67`	`67`	`success = globalState.executeStatement(query);`
`68`		`- } while (!success && nrTries++ < globalState.getOptions().getNrStatementRetryCount());`
	`68`	`+ } while (nextAction.canBeRetried() && !success`
	`69`	`+ && nrTries++ < globalState.getOptions().getNrStatementRetryCount());`
`69`	`70`	`} catch (IgnoreMeException e) {`
`70`	`71`
`71`	`72`	`}`