Make Stream class implement io.streams.OutputStream

Author: thekid

src/main/php/com/amazon/aws/lambda/RuntimeApi.class.php

@@ -105,6 +105,14 @@ public function write($bytes) {
         $this->stream->flush();
       }
 
+      public function flush() {
+        // All writes flush, NOOP
+      }
+
+      public function close() {
+        $this->end();
+      }
+
       public function end() {      
         if ($this->response) return; // Already ended
 

src/main/php/com/amazon/aws/lambda/Stream.class.php

@@ -1,6 +1,8 @@
 <?php namespace com\amazon\aws\lambda;
 
-interface Stream {
+use io\streams\OutputStream;
+
+interface Stream extends OutputStream {
 
   /**
    * Transmits a given source to the output asynchronously.

src/main/php/xp/lambda/LocalRuntime.class.php

@@ -35,6 +35,8 @@ public function transmit($source, $mime= null) {
           public function use($mime) { /** NOOP */ }
           public function write($bytes) { $this->api->out->write($bytes); }
           public function end() { /** NOOP */ }
+          public function flush() { /** NOOP */ }
+          public function close() { /** NOOP */ }
         };
         $lambda($event, $stream, $context);
       }

src/test/php/com/amazon/aws/lambda/unittest/StreamingTest.class.php

@@ -1,7 +1,7 @@
 <?php namespace com\amazon\aws\lambda\unittest;
 
 use com\amazon\aws\lambda\{Context, Streaming};
-use io\streams\MemoryInputStream;
+use io\streams\{MemoryInputStream, StreamTransfer};
 use lang\IllegalStateException;
 use test\{Assert, Expect, Test, Values};
 
@@ -126,6 +126,26 @@ public function transmit() {
     );
   }
 
+  #[Test]
+  public function can_be_used_as_output_stream() {
+    $response= $this->invoke(function($event, $stream, $context) {
+      $stream->use('application/json');
+      (new StreamTransfer(new MemoryInputStream('{"test":true}'), $stream))->transferAll();
+    });
+
+    Assert::equals(
+      "POST /2018-06-01/runtime/invocation/3e1afeb0-cde4-1d0e-c3c0-66b15046bb88/response HTTP/1.1\r\n".
+      "Connection: close\r\n".
+      "Host: test\r\n".
+      "Content-Type: application/json\r\n".
+      "Lambda-Runtime-Function-Response-Mode: streaming\r\n".
+      "Transfer-Encoding: chunked\r\n".
+      "\r\n".
+      "{\"test\":true}",
+      $response
+    );
+  }
+
   #[Test]
   public function invoke_handler_multiple_times() {
     $lambda= function($event, $stream, $context) { $stream->write('Test'); };

src/test/php/com/amazon/aws/lambda/unittest/TestStream.class.php

@@ -11,5 +11,9 @@ public function use($mime) { /** NOOP */ }
 
   public function write($bytes) { $this->written.= $bytes; }
 
+  public function flush() { /** NOOP */ }
+
+  public function close() { /** NOOP */ }
+
   public function end() { /** NOOP */ }
 }