From d5353d45783104a8c3f3eba7f51ba640d6622517 Mon Sep 17 00:00:00 2001
From: ArtoLord <artolord@yandex-team.ru>
Date: Thu, 9 Nov 2023 13:01:47 +0300
Subject: [PATCH] Add string formatting in LogStream

---
 .../src/main/java/ai/lzy/env/logs/LogStreamQueue.java | 11 ++++++++++-
 1 file changed, 10 insertions(+), 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
index ee0e0794c..89703cba5 100644
--- 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);
                 }