feat: add appendType StoneDBTableGenerator.java

zhenglin-charlie-li · Jun 23, 2023 · 46f45e9 · 46f45e9
1 parent 5203913
commit 46f45e9
Show file tree

Hide file tree

Showing 3 changed files with 238 additions and 138 deletions.
diff --git a/src/sqlancer/Randomly.java b/src/sqlancer/Randomly.java
@@ -1,11 +1,8 @@
 package sqlancer;
 
 import java.math.BigDecimal;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Collections;
-import java.util.List;
-import java.util.Random;
+import java.math.BigInteger;
+import java.util.*;
 import java.util.function.Supplier;
 
 public final class Randomly {
@@ -215,7 +212,6 @@ public String getString(Randomly r) {
 
         },
         ALPHANUMERIC {
-
             @Override
             public String getString(Randomly r) {
                 return getStringOfAlphabet(r, ALPHANUMERIC_ALPHABET);
@@ -224,7 +220,6 @@ public String getString(Randomly r) {
 
         },
         ALPHANUMERIC_SPECIALCHAR {
-
             @Override
             public String getString(Randomly r) {
                 return getStringOfAlphabet(r, ALPHANUMERIC_SPECIALCHAR_ALPHABET);
@@ -452,6 +447,18 @@ public long getLong(long left, long right) {
         return getNextLong(left, right);
     }
 
+    public BigInteger getBigInteger(BigInteger left, BigInteger right) {
+        if (left.equals(right)) {
+            return left;
+        }
+        while (true) {
+            BigInteger result = new BigInteger(63, new Random());
+            if (result.compareTo(left) >= 0 && result.compareTo(right) <= 0) {
+                return result;
+            }
+        }
+    }
+
     public BigDecimal getRandomBigDecimal() {
         return BigDecimal.valueOf(getThreadRandom().get().nextDouble());
     }

diff --git a/src/sqlancer/stonedb/StoneDBProvider.java b/src/sqlancer/stonedb/StoneDBProvider.java
@@ -28,6 +28,7 @@ protected StoneDBSchema readSchema() throws Exception {
 
     enum Action implements AbstractAction<StoneDBGlobalState> {
         SHOW_TABLES((g) -> new SQLQueryAdapter("SHOW TABLES"));
+
         private final SQLQueryProvider<StoneDBGlobalState> sqlQueryProvider;
 
         Action(SQLQueryProvider<StoneDBGlobalState> sqlQueryProvider) {
@@ -43,10 +44,10 @@ public Query<?> getQuery(StoneDBGlobalState globalState) throws Exception {
     private static int mapActions(StoneDBGlobalState globalState, Action a) {
         Randomly r = globalState.getRandomly();
         switch (a) {
-            case SHOW_TABLES:
-                return 1;
-            default:
-                throw new AssertionError(a);
+        case SHOW_TABLES:
+            return 1;
+        default:
+            throw new AssertionError(a);
         }
     }
 
@@ -59,10 +60,10 @@ public void generateDatabase(StoneDBGlobalState globalState) throws Exception {
         }
         StatementExecutor<StoneDBGlobalState, Action> se = new StatementExecutor<>(globalState, Action.values(),
                 StoneDBProvider::mapActions, (q) -> {
-            if (globalState.getSchema().getDatabaseTables().isEmpty()) {
-                throw new IgnoreMeException();
-            }
-        });
+                    if (globalState.getSchema().getDatabaseTables().isEmpty()) {
+                        throw new IgnoreMeException();
+                    }
+                });
         se.executeStatements();
     }