FilesExpand file tree

 main

/

TiDBDQPOracle.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

80 lines (70 loc) · 3.2 KB

 main

/

TiDBDQPOracle.java

Top

File metadata and controls

• Code
• Blame

80 lines (70 loc) · 3.2 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

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

package sqlancer.tidb.oracle;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.List;

import java.util.stream.Collectors;

import sqlancer.ComparatorHelper;

import sqlancer.Randomly;

import sqlancer.common.oracle.TestOracle;

import sqlancer.common.query.ExpectedErrors;

import sqlancer.tidb.TiDBErrors;

import sqlancer.tidb.TiDBExpressionGenerator;

import sqlancer.tidb.TiDBProvider.TiDBGlobalState;

import sqlancer.tidb.TiDBSchema.TiDBTables;

import sqlancer.tidb.ast.TiDBColumnReference;

import sqlancer.tidb.ast.TiDBExpression;

import sqlancer.tidb.ast.TiDBJoin;

import sqlancer.tidb.ast.TiDBSelect;

import sqlancer.tidb.ast.TiDBTableReference;

import sqlancer.tidb.ast.TiDBText;

import sqlancer.tidb.gen.TiDBHintGenerator;

import sqlancer.tidb.visitor.TiDBVisitor;

public class TiDBDQPOracle implements TestOracle<TiDBGlobalState> {

private TiDBExpressionGenerator gen;

private final TiDBGlobalState state;

private TiDBSelect select;

private final ExpectedErrors errors = new ExpectedErrors();

public TiDBDQPOracle(TiDBGlobalState globalState) {

state = globalState;

TiDBErrors.addExpressionErrors(errors);

}

@Override

public void check() throws SQLException {

// Randomly generate a query

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

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

select = new TiDBSelect();

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

fetchColumns.addAll(Randomly.nonEmptySubset(tables.getColumns()).stream().map(c -> new TiDBColumnReference(c))

.collect(Collectors.toList()));

select.setFetchColumns(fetchColumns);

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

.collect(Collectors.toList());

List<TiDBExpression> joins = TiDBJoin.getJoins(tableList, state).stream().collect(Collectors.toList());

select.setJoinList(joins);

select.setFromList(tableList);

if (Randomly.getBoolean()) {

select.setWhereClause(gen.generateExpression());

}

if (Randomly.getBooleanWithRatherLowProbability()) {

select.setOrderByClauses(gen.generateOrderBys());

}

if (Randomly.getBoolean()) {

select.setLimitClause(gen.generateExpression());

}

if (Randomly.getBoolean()) {

select.setOffsetClause(gen.generateExpression());

}

String originalQueryString = TiDBVisitor.asString(select);

List<String> originalResult = ComparatorHelper.getResultSetFirstColumnAsString(originalQueryString, errors,

state);

List<TiDBText> hintList = TiDBHintGenerator.generateAllHints(select, tables.getTables());

for (TiDBText hint : hintList) {

select.setHint(hint);

String queryString = TiDBVisitor.asString(select);

List<String> result = ComparatorHelper.getResultSetFirstColumnAsString(queryString, errors, state);

ComparatorHelper.assumeResultSetsAreEqual(originalResult, result, originalQueryString, List.of(queryString),

state);

}

}

}