Add an option to specify how often a statement kind should be retried

mrigger · mrigger · commit 09b755839968 · 2020-04-06T10:02:14.000+02:00
diff --git a/src/sqlancer/MainOptions.java b/src/sqlancer/MainOptions.java
@@ -25,6 +25,10 @@ public class MainOptions {
 			"--num_queries" }, description = "Specifies the number of queries to be issued to a database before creating a new database")
 	private int nrQueries = 100000;
 
+	@Parameter(names = {
+			"--num_statement_kind_retries" }, description = "Specifies the number of times a specific statement kind (e.g., INSERT) should be retried when the DBMS indicates that it failed")
+	private int nrStatementRetryCount = 1000;
+
 	@Parameter(names = "--log-each-select", description = "Logs every statement issued", arity = 1)
 	private boolean logEachSelect = true;
 
@@ -55,6 +59,10 @@ public int getNrQueries() {
 	public int getMaxNumberInserts() {
 		return maxNumberInserts;
 	}
+	
+	public int getNrStatementRetryCount() {
+		return nrStatementRetryCount;
+	}
 
 	public static enum DBMS {
 		MariaDB, SQLite3, MySQL, PostgreSQL, TDEngine, CockroachDB, TiDB
diff --git a/src/sqlancer/StatementExecutor.java b/src/sqlancer/StatementExecutor.java
@@ -69,7 +69,7 @@ public void executeStatements() throws SQLException {
 						globalState.getLogger().writeCurrent(query.getQueryString());
 					}
 					success = globalState.getManager().execute(query);
-				} while (!success && nrTries++ < 1000);
+				} while (!success && nrTries++ < globalState.getOptions().getNrStatementRetryCount());
 			} catch (IgnoreMeException e) {
 
 			}

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ public void executeStatements() throws SQLException {`
`69`	`69`	`globalState.getLogger().writeCurrent(query.getQueryString());`
`70`	`70`	`}`
`71`	`71`	`success = globalState.getManager().execute(query);`
`72`		`- } while (!success && nrTries++ < 1000);`
	`72`	`+ } while (!success && nrTries++ < globalState.getOptions().getNrStatementRetryCount());`
`73`	`73`	`} catch (IgnoreMeException e) {`
`74`	`74`
`75`	`75`	`}`