forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSQLLoggableFactory.java
More file actions
52 lines (44 loc) · 1.7 KB
/
SQLLoggableFactory.java
File metadata and controls
52 lines (44 loc) · 1.7 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
package sqlancer.common.log;
import java.io.PrintWriter;
import java.io.StringWriter;
import sqlancer.common.query.Query;
import sqlancer.common.query.SQLQueryAdapter;
public class SQLLoggableFactory extends LoggableFactory {
@Override
protected Loggable createLoggable(String input, String suffix) {
String completeString = input;
if (!input.endsWith(";")) {
completeString += ";";
}
if (suffix != null && suffix.length() != 0) {
completeString += suffix;
}
return new LoggedString(completeString);
}
@Override
public SQLQueryAdapter getQueryForStateToReproduce(String queryString) {
return new SQLQueryAdapter(queryString);
}
@Override
public SQLQueryAdapter commentOutQuery(Query<?> query) {
String queryString = query.getLogString();
String newQueryString = "-- " + queryString;
return new SQLQueryAdapter(newQueryString);
}
@Override
protected Loggable infoToLoggable(String time, String databaseName, String databaseVersion, long seedValue) {
StringBuilder sb = new StringBuilder();
sb.append("-- Time: " + time + "\n");
sb.append("-- Database: " + databaseName + "\n");
sb.append("-- Database version: " + databaseVersion + "\n");
sb.append("-- seed value: " + seedValue + "\n");
return new LoggedString(sb.toString());
}
@Override
public Loggable convertStacktraceToLoggable(Throwable throwable) {
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
throwable.printStackTrace(pw);
return new LoggedString("--" + sw.toString().replace("\n", "\n--"));
}
}