FilesExpand file tree

 feldera

/

HSQLDBUpdateGenerator.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

55 lines (46 loc) · 2.1 KB

 feldera

/

HSQLDBUpdateGenerator.java

Top

File metadata and controls

• Code
• Blame

55 lines (46 loc) · 2.1 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

48

49

50

51

52

53

54

55

package sqlancer.hsqldb.gen;

import java.util.List;

import sqlancer.Randomly;

import sqlancer.common.gen.AbstractUpdateGenerator;

import sqlancer.common.query.SQLQueryAdapter;

import sqlancer.hsqldb.HSQLDBErrors;

import sqlancer.hsqldb.HSQLDBProvider;

import sqlancer.hsqldb.HSQLDBSchema;

import sqlancer.hsqldb.HSQLDBSchema.HSQLDBColumn;

import sqlancer.hsqldb.HSQLDBSchema.HSQLDBCompositeDataType;

import sqlancer.hsqldb.HSQLDBSchema.HSQLDBDataType;

import sqlancer.hsqldb.HSQLDBToStringVisitor;

import sqlancer.hsqldb.ast.HSQLDBExpression;

public final class HSQLDBUpdateGenerator extends AbstractUpdateGenerator<HSQLDBColumn> {

private final HSQLDBProvider.HSQLDBGlobalState globalState;

private HSQLDBExpressionGenerator gen;

private HSQLDBUpdateGenerator(HSQLDBProvider.HSQLDBGlobalState globalState) {

this.globalState = globalState;

}

public static SQLQueryAdapter getQuery(HSQLDBProvider.HSQLDBGlobalState globalState) {

return new HSQLDBUpdateGenerator(globalState).generate();

}

private SQLQueryAdapter generate() {

HSQLDBSchema.HSQLDBTable table = globalState.getSchema().getRandomTable(t -> !t.isView());

List<HSQLDBSchema.HSQLDBColumn> columns = table.getRandomNonEmptyColumnSubset();

gen = new HSQLDBExpressionGenerator(globalState).setColumns(table.getColumns());

sb.append("UPDATE ");

sb.append(table.getName());

sb.append(" SET ");

updateColumns(columns);

if (Randomly.getBooleanWithSmallProbability()) {

sb.append(" WHERE ");

sb.append(HSQLDBToStringVisitor.asString(

gen.generateExpression(HSQLDBCompositeDataType.getRandomWithType(HSQLDBDataType.BOOLEAN))));

errors.add("data type of expression is not boolean");

HSQLDBErrors.addExpressionErrors(errors);

}

return new SQLQueryAdapter(sb.toString(), errors);

}

@Override

protected void updateValue(HSQLDBColumn column) {

HSQLDBExpression expr;

expr = gen.generateConstant(column.getType());

sb.append(HSQLDBToStringVisitor.asString(expr));

}

}