IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

more improvements to command processing

Browse Source
This commit is contained in:
Jonathan Shook
2020-04-22 07:24:34 -05:00
parent 908ab7c1a3
commit 113932aed5
9 changed files with 226 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
engine-api/src/main/java/io/nosqlbench/engine/api/scripting/ScriptEnvBuffer.java
View File

@@ -140,6 +140,7 @@ public class ScriptEnvBuffer extends SimpleScriptContext {
private final String prefix;
CharArrayWriter buffer = new CharArrayWriter();
private List<String> timedLog = new ArrayList<String>();
private final StringBuilder sb = new StringBuilder();
public DiagWriter(Writer wrapped, String prefix) {
this.wrapped = wrapped;
@@ -152,9 +153,22 @@ public class ScriptEnvBuffer extends SimpleScriptContext {
buffer.write(cbuf, off, len);
String text = new String(cbuf, off, len);
if (!text.equals("\n")) {
String tslogEntry = tsprefix + prefix + new String(cbuf, off, len);
timedLog.add(tslogEntry);
sb.append(text);
if (text.contains("\n")) {
String msgs = sb.toString();
String extra = msgs.substring(msgs.lastIndexOf("\n")+1);
sb.setLength(0);
sb.append(extra);
String[] parts = msgs.substring(0,msgs.length()-extra.length()).split("\n");
for (String part : parts) {
if (!part.isBlank()) {
String tslogEntry = tsprefix + prefix + part + "\n";
timedLog.add(tslogEntry);
}
}
}
wrapped.write(cbuf, off, len);