FilesExpand file tree

 main

/

TiDBRandomQuerySynthesizer.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

61 lines (54 loc) · 2.33 KB

 main

/

TiDBRandomQuerySynthesizer.java

Top

File metadata and controls

• Code
• Blame

61 lines (54 loc) · 2.33 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

56

57

58

59

60

61

package sqlancer.tidb.gen;

import java.util.ArrayList;

import java.util.List;

import java.util.stream.Collectors;

import sqlancer.Randomly;

import sqlancer.common.query.SQLQueryAdapter;

import sqlancer.tidb.TiDBExpressionGenerator;

import sqlancer.tidb.TiDBProvider.TiDBGlobalState;

import sqlancer.tidb.TiDBSchema.TiDBTables;

import sqlancer.tidb.ast.TiDBExpression;

import sqlancer.tidb.ast.TiDBSelect;

import sqlancer.tidb.ast.TiDBTableReference;

import sqlancer.tidb.visitor.TiDBVisitor;

public final class TiDBRandomQuerySynthesizer {

private TiDBRandomQuerySynthesizer() {

}

public static SQLQueryAdapter generate(TiDBGlobalState globalState, int nrColumns) {

TiDBSelect select = generateSelect(globalState, nrColumns);

return new SQLQueryAdapter(TiDBVisitor.asString(select));

}

public static TiDBSelect generateSelect(TiDBGlobalState globalState, int nrColumns) {

TiDBTables tables = globalState.getSchema().getRandomTableNonEmptyTables();

TiDBExpressionGenerator gen = new TiDBExpressionGenerator(globalState).setColumns(tables.getColumns());

TiDBSelect select = new TiDBSelect();

// select.setDistinct(Randomly.getBoolean());

List<TiDBExpression> columns = new ArrayList<>();

// TODO: also generate aggregates

columns.addAll(gen.generateExpressions(nrColumns));

select.setFetchColumns(columns);

List<TiDBExpression> tableList = tables.getTables().stream().map(t -> new TiDBTableReference(t))

.collect(Collectors.toList());

// TODO: generate joins

select.setFromList(tableList);

if (Randomly.getBoolean()) {

select.setWhereClause(gen.generateExpression());

}

if (Randomly.getBooleanWithRatherLowProbability()) {

select.setOrderByClauses(gen.generateOrderBys());

}

if (Randomly.getBoolean()) {

select.setGroupByExpressions(gen.generateExpressions(Randomly.smallNumber() + 1));

if (Randomly.getBoolean()) {

select.setHavingClause(gen.generateHavingClause());

}

}

if (Randomly.getBoolean()) {

select.setLimitClause(gen.generateExpression());

}

if (Randomly.getBoolean()) {

select.setOffsetClause(gen.generateExpression());

}

return select;

}

}