forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCnosDBSchema.java
More file actions
243 lines (200 loc) · 8.46 KB
/
CnosDBSchema.java
File metadata and controls
243 lines (200 loc) · 8.46 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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
package sqlancer.cnosdb;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import sqlancer.Randomly;
import sqlancer.cnosdb.ast.CnosDBConstant;
import sqlancer.cnosdb.client.CnosDBConnection;
import sqlancer.cnosdb.client.CnosDBResultSet;
import sqlancer.common.schema.AbstractRowValue;
import sqlancer.common.schema.AbstractSchema;
import sqlancer.common.schema.AbstractTable;
import sqlancer.common.schema.AbstractTableColumn;
import sqlancer.common.schema.AbstractTables;
import sqlancer.common.schema.TableIndex;
public class CnosDBSchema extends AbstractSchema<CnosDBGlobalState, CnosDBSchema.CnosDBTable> {
private final String databaseName;
public CnosDBSchema(List<CnosDBTable> databaseTables, String databaseName) {
super(databaseTables);
this.databaseName = databaseName;
}
public static CnosDBDataType getColumnType(String typeString) {
switch (typeString.toLowerCase()) {
case "bigint":
return CnosDBDataType.INT;
case "boolean":
return CnosDBDataType.BOOLEAN;
case "string":
return CnosDBDataType.STRING;
case "double":
return CnosDBDataType.DOUBLE;
case "bigint unsigned":
case "unsigned":
return CnosDBDataType.UINT;
case "timestamp(nanosecond)":
return CnosDBDataType.TIMESTAMP;
default:
throw new AssertionError(typeString);
}
}
public static CnosDBSchema fromConnection(CnosDBConnection con) throws Exception {
CnosDBResultSet tablesRes = con.getClient().executeQuery("SHOW TABLES");
List<CnosDBTable> tables = new ArrayList<>();
while (tablesRes.next()) {
String tableName = tablesRes.getString(1);
List<CnosDBColumn> columns = getTableColumns(con, tableName);
tables.add(new CnosDBTable(tableName, columns));
}
return new CnosDBSchema(tables, con.getClient().getDatabase());
}
protected static List<CnosDBColumn> getTableColumns(CnosDBConnection con, String tableName) throws Exception {
CnosDBResultSet columnsRes = con.getClient().executeQuery("DESCRIBE TABLE " + tableName);
List<CnosDBColumn> columns = new ArrayList<>();
CnosDBTable table = new CnosDBTable(tableName, columns);
while (columnsRes.next()) {
String columnName = columnsRes.getString(1);
String columnType = columnsRes.getString(3).toLowerCase();
CnosDBDataType dataType = CnosDBSchema.getColumnType(columnsRes.getString(2));
CnosDBColumn column;
if (columnType.contentEquals("time")) {
column = new CnosDBTimeColumn();
} else if (columnType.contentEquals("tag")) {
column = new CnosDBTagColumn(columnName);
} else {
column = new CnosDBFieldColumn(columnName, dataType);
}
column.setTable(table);
columns.add(column);
}
return columns;
}
public CnosDBTables getRandomTableNonEmptyTables() {
return new CnosDBTables(Randomly.nonEmptySubset(getDatabaseTables()));
}
public String getDatabaseName() {
return databaseName;
}
public enum CnosDBDataType {
INT, BOOLEAN, STRING, DOUBLE, UINT, TIMESTAMP;
public static CnosDBDataType getRandomType() {
return Randomly.fromOptions(values());
}
public static CnosDBDataType getRandomTypeWithoutTimeStamp() {
List<CnosDBDataType> dataTypes = new ArrayList<>(Arrays.asList(values()));
dataTypes.remove(TIMESTAMP);
return Randomly.fromList(dataTypes);
}
}
public static class CnosDBColumn extends AbstractTableColumn<CnosDBTable, CnosDBDataType> {
public CnosDBColumn(String name, CnosDBDataType columnType) {
super(name, null, columnType);
}
public static CnosDBColumn createDummy(String name) {
return new CnosDBColumn(name, CnosDBDataType.INT);
}
}
public static class CnosDBTagColumn extends CnosDBColumn {
public CnosDBTagColumn(String name) {
super(name, CnosDBDataType.STRING);
}
}
public static class CnosDBTimeColumn extends CnosDBColumn {
public CnosDBTimeColumn() {
super("TIME", CnosDBDataType.TIMESTAMP);
}
}
public static class CnosDBFieldColumn extends CnosDBColumn {
public CnosDBFieldColumn(String name, CnosDBDataType columnType) {
super(name, columnType);
assert columnType != CnosDBDataType.TIMESTAMP;
}
}
public static class CnosDBTables extends AbstractTables<CnosDBTable, CnosDBColumn> {
public CnosDBTables(List<CnosDBTable> tables) {
super(tables);
}
public CnosDBRowValue getRandomRowValue(CnosDBConnection con) {
return null;
}
public List<CnosDBColumn> getRandomColumnsWithOnlyOneField() {
ArrayList<CnosDBColumn> res = new ArrayList<>();
this.getTables().forEach(table -> res.addAll(table.getRandomColumnsWithOnlyOneField()));
return res;
}
}
public static class CnosDBRowValue extends AbstractRowValue<CnosDBTables, CnosDBColumn, CnosDBConstant> {
protected CnosDBRowValue(CnosDBTables tables, Map<CnosDBColumn, CnosDBConstant> values) {
super(tables, values);
}
}
public static class CnosDBTable extends AbstractTable<CnosDBColumn, TableIndex, CnosDBGlobalState> {
public CnosDBTable(String tableName, List<CnosDBColumn> columns) {
super(tableName, columns, null, false);
}
@Override
public List<CnosDBColumn> getColumns() {
List<CnosDBColumn> res = super.getColumns();
boolean hasTime = false;
for (CnosDBColumn column : res) {
if (column instanceof CnosDBTimeColumn) {
hasTime = true;
break;
}
}
assert hasTime;
return res;
}
public List<CnosDBColumn> getRandomColumnsWithOnlyOneField() {
ArrayList<CnosDBColumn> res = new ArrayList<>();
boolean hasField = false;
for (CnosDBColumn column : getColumns()) {
if (column instanceof CnosDBTagColumn && Randomly.getBoolean()) {
res.add(column);
} else if (column instanceof CnosDBFieldColumn && !hasField) {
res.add(column);
hasField = true;
}
}
return res;
}
// SELECT COUNT(*) FROM table;
@Override
public long getNrRows(CnosDBGlobalState globalState) {
long res;
try {
CnosDBResultSet tableCountRes = globalState.getConnection().getClient()
.executeQuery("SELECT COUNT(time) FROM " + this.name);
tableCountRes.next();
res = tableCountRes.getLong(1);
} catch (Exception e) {
res = 0;
}
return res;
}
@Override
public List<CnosDBColumn> getRandomNonEmptyColumnSubset() {
List<CnosDBColumn> selectedColumns = new ArrayList<>();
ArrayList<CnosDBColumn> remainingColumns = new ArrayList<>(this.getColumns());
remainingColumns.removeIf(column -> column instanceof CnosDBTimeColumn);
CnosDBTimeColumn timeColumn = new CnosDBTimeColumn();
timeColumn.setTable(this);
selectedColumns.add(timeColumn);
remainingColumns.stream().filter(column -> column instanceof CnosDBTagColumn).findFirst().ifPresent(tag -> {
selectedColumns.add(tag);
remainingColumns.remove(tag);
});
remainingColumns.stream().filter(column -> column instanceof CnosDBFieldColumn).findFirst()
.ifPresent(field -> {
selectedColumns.add(field);
remainingColumns.remove(field);
});
int nr = Math.min(Randomly.smallNumber() + 1, remainingColumns.size());
for (int i = 0; i < nr; i++) {
selectedColumns
.add(remainingColumns.remove((int) Randomly.getNotCachedInteger(0, remainingColumns.size())));
}
return selectedColumns;
}
}
}