forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFelderaInsertGenerator.java
More file actions
47 lines (40 loc) · 1.79 KB
/
FelderaInsertGenerator.java
File metadata and controls
47 lines (40 loc) · 1.79 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package sqlancer.feldera.gen;
import sqlancer.common.gen.AbstractInsertGenerator;
import sqlancer.common.query.ExpectedErrors;
import sqlancer.common.schema.AbstractTableColumn;
import sqlancer.feldera.FelderaGlobalState;
import sqlancer.feldera.FelderaSchema;
import sqlancer.feldera.FelderaToStringVisitor;
import sqlancer.feldera.query.FelderaOtherQuery;
import java.util.List;
import java.util.stream.Collectors;
public class FelderaInsertGenerator extends AbstractInsertGenerator<FelderaSchema.FelderaColumn> {
private final FelderaGlobalState globalState;
private final ExpectedErrors errors = new ExpectedErrors();
private FelderaInsertGenerator(FelderaGlobalState globalState) {
this.globalState = globalState;
}
public static FelderaOtherQuery getQuery(FelderaGlobalState globalState) {
return new FelderaInsertGenerator(globalState).generate();
}
private FelderaOtherQuery generate() {
sb.append("INSERT INTO ");
FelderaSchema.FelderaTable table = globalState.getSchema().getRandomTable(t -> !t.isView());
List<FelderaSchema.FelderaColumn> columns = table
.getRandomNonEmptyColumnSubsetFilter(p -> !p.getName().equals("rowid"));
sb.append(table.getName());
sb.append("(");
sb.append(columns.stream().map(AbstractTableColumn::getName).collect(Collectors.joining(", ")));
sb.append(")");
sb.append(" VALUES ");
insertColumns(columns);
sb.append(";");
String s = sb.toString();
return new FelderaOtherQuery(s, errors);
}
@Override
protected void insertValue(FelderaSchema.FelderaColumn column) {
sb.append(FelderaToStringVisitor
.asString(new FelderaExpressionGenerator(globalState).generateConstant(column.getType())));
}
}