Merge pull request treasure-data#1497 from yoyama/master2v0_10-echo-log

[Master to v0_10] log for echo> (treasure-data#1221)
darcy168 · Dec 15, 2020 · c272399 · c272399
2 parents cfca0b6 + a893e89
commit c272399
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/digdag-standards/src/main/java/io/digdag/standards/operator/EchoOperatorFactory.java b/digdag-standards/src/main/java/io/digdag/standards/operator/EchoOperatorFactory.java
@@ -2,12 +2,16 @@
 
 import com.google.inject.Inject;
 import io.digdag.client.config.Config;
+import io.digdag.core.log.TaskContextLogging;
+import io.digdag.core.log.TaskLogger;
 import io.digdag.spi.Operator;
 import io.digdag.spi.OperatorContext;
 import io.digdag.spi.OperatorFactory;
 import io.digdag.spi.TaskRequest;
 import io.digdag.spi.TaskResult;
 
+import java.nio.charset.StandardCharsets;
+
 public class EchoOperatorFactory
         implements OperatorFactory
 {
@@ -45,6 +49,11 @@ public TaskResult run()
 
             System.out.println(message);
 
+            // To store the message to task log
+            TaskLogger logger = TaskContextLogging.getContext().getLogger();
+            byte[] messageBytes = message.getBytes(StandardCharsets.UTF_8);
+            logger.log(messageBytes, 0, messageBytes.length);
+
             return TaskResult.empty(request);
         }
     }