-
Notifications
You must be signed in to change notification settings - Fork 397
Expand file tree
/
Copy pathSQLLoggableFactory.java
More file actions
54 lines (46 loc) · 1.86 KB
/
SQLLoggableFactory.java
File metadata and controls
54 lines (46 loc) · 1.86 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
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 += ";";
}
completeString = completeString.replace("\n", "\\n");
completeString = completeString.replace("\r", "\\r");
if (suffix != null && !suffix.isEmpty()) {
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: ").append(time).append("\n");
sb.append("-- Database: ").append(databaseName).append("\n");
sb.append("-- Database version: ").append(databaseVersion).append("\n");
sb.append("-- seed value: ").append(seedValue).append("\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--"));
}
}