FilesExpand file tree

 documentation

/

TiDBTLPBase.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

68 lines (58 loc) · 2.32 KB

 documentation

/

TiDBTLPBase.java

Top

File metadata and controls

• Code
• Blame

68 lines (58 loc) · 2.32 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

package sqlancer.tidb.oracle;

import java.sql.SQLException;

import java.util.Arrays;

import java.util.List;

import java.util.stream.Collectors;

import sqlancer.Randomly;

import sqlancer.common.gen.ExpressionGenerator;

import sqlancer.common.oracle.TernaryLogicPartitioningOracleBase;

import sqlancer.common.oracle.TestOracle;

import sqlancer.tidb.TiDBErrors;

import sqlancer.tidb.TiDBExpressionGenerator;

import sqlancer.tidb.TiDBProvider.TiDBGlobalState;

import sqlancer.tidb.TiDBSchema;

import sqlancer.tidb.TiDBSchema.TiDBTable;

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.gen.TiDBHintGenerator;

public abstract class TiDBTLPBase extends TernaryLogicPartitioningOracleBase<TiDBExpression, TiDBGlobalState>

implements TestOracle<TiDBGlobalState> {

TiDBSchema s;

TiDBTables targetTables;

TiDBExpressionGenerator gen;

TiDBSelect select;

public TiDBTLPBase(TiDBGlobalState state) {

super(state);

TiDBErrors.addExpressionErrors(errors);

}

@Override

public void check() throws SQLException {

s = state.getSchema();

targetTables = s.getRandomTableNonEmptyTables();

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

initializeTernaryPredicateVariants();

select = new TiDBSelect();

select.setFetchColumns(generateFetchColumns());

List<TiDBTable> tables = targetTables.getTables();

if (Randomly.getBoolean()) {

TiDBHintGenerator.generateHints(select, tables);

}

List<TiDBExpression> tableList = tables.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);

select.setWhereClause(null);

}

List<TiDBExpression> generateFetchColumns() {

return Arrays.asList(new TiDBColumnReference(targetTables.getColumns().get(0)));

}

@Override

protected ExpressionGenerator<TiDBExpression> getGen() {

return gen;

}

}