forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathAbstractSchema.java
More file actions
91 lines (75 loc) · 2.59 KB
/
AbstractSchema.java
File metadata and controls
91 lines (75 loc) · 2.59 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
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
81
82
83
84
85
86
87
88
89
90
91
package sqlancer.schema;
import java.util.Collections;
import java.util.List;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import sqlancer.Randomly;
public class AbstractSchema<A extends AbstractTable<?, ?>> {
private final List<A> databaseTables;
public AbstractSchema(List<A> databaseTables) {
this.databaseTables = Collections.unmodifiableList(databaseTables);
}
@Override
public String toString() {
StringBuffer sb = new StringBuffer();
for (A t : getDatabaseTables()) {
sb.append(t + "\n");
}
return sb.toString();
}
public A getRandomTable() {
return Randomly.fromList(getDatabaseTables());
}
public A getRandomTable(Predicate<A> predicate) {
return Randomly.fromList(getDatabaseTables().stream().filter(predicate).collect(Collectors.toList()));
}
public List<A> getDatabaseTables() {
return databaseTables;
}
public List<A> getDatabaseTablesRandomSubsetNotEmpty() {
return Randomly.nonEmptySubset(databaseTables);
}
public String getFreeIndexName() {
int i = 0;
if (Randomly.getBooleanWithRatherLowProbability()) {
i = (int) Randomly.getNotCachedInteger(0, 100);
}
do {
String indexName = String.format("i%d", i++);
boolean indexNameFound = false;
for (A table : databaseTables) {
if (table.getIndexes().stream().anyMatch(ind -> ind.getIndexName().contentEquals(indexName))) {
indexNameFound = true;
break;
}
}
if (!indexNameFound) {
return indexName;
}
} while (true);
}
public String getFreeTableName() {
int i = 0;
if (Randomly.getBooleanWithRatherLowProbability()) {
i = (int) Randomly.getNotCachedInteger(0, 100);
}
do {
String tableName = String.format("t%d", i++);
if (databaseTables.stream().noneMatch(t -> t.getName().contentEquals(tableName))) {
return tableName;
}
} while (true);
}
public String getFreeViewName() {
int i = 0;
if (Randomly.getBooleanWithRatherLowProbability()) {
i = (int) Randomly.getNotCachedInteger(0, 100);
}
do {
String tableName = String.format("v%d", i++);
if (databaseTables.stream().noneMatch(t -> t.getName().contentEquals(tableName))) {
return tableName;
}
} while (true);
}
}