feat: add AS support when inserting

zhenglin-charlie-li · Aug 27, 2023 · 0124532 · 0124532
1 parent bb4ce75
commit 0124532
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/sqlancer/stonedb/gen/StoneDBTableInsertGenerator.java b/src/sqlancer/stonedb/gen/StoneDBTableInsertGenerator.java
@@ -48,6 +48,7 @@ private SQLQueryAdapter getQuery() {
         sb.append(table.getName());
         appendPartition();
         appendColumnsAndValues(columns);
+        appendAS();
         appendOnDuplicateUpdate();
         addExpectedErrors();
         return new SQLQueryAdapter(sb.toString(), errors);
@@ -84,6 +85,14 @@ private void appendColumnsAndValues(List<StoneDBColumn> columns) {
         appendValues();
     }
 
+    private void appendAS() {
+        if (!globalState.getDbmsSpecificOptions().test80Version) {
+            return;
+        }
+        sb.append(" AS ");
+        sb.append("r").append(table.getNrRows(globalState));
+    }
+
     private void appendOnDuplicateUpdate() {
         sb.append("on duplicate key update ");
         StoneDBColumn randomColumn = table.getRandomColumn();