Add string formatting in LogStream

lambdazy · Nov 9, 2023 · d5353d4 · d5353d4
1 parent 5e7c707
commit d5353d4
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lzy/execution-env/src/main/java/ai/lzy/env/logs/LogStreamQueue.java b/lzy/execution-env/src/main/java/ai/lzy/env/logs/LogStreamQueue.java
@@ -78,7 +78,16 @@ public void run() {
 
             if (inputHandle.string() != null) {
                 try {
-                    writeLines(inputHandle.string().getBytes());
+                    var s = inputHandle.string();
+                    var formatter = inputHandle.formatter();
+
+                    if (formatter != null) {
+                        for (var line : s.split("\n")) {
+                            writeLines(formatter.apply(line).getBytes());
+                        }
+                    } else {
+                        writeLines(inputHandle.string().getBytes());
+                    }
                 } catch (IOException e) {
                     LogStream.LOG.warn("Cannot write buffer to stream {}: ", streamName, e);
                 }