add receiver method to TransportLayer

2 files changed, 21 insertions, 0 deletions

diff --git a/src/main/kotlin/TransportLayer.kt b/src/main/kotlin/TransportLayer.kt
index d4a9b12..8778592 100644
--- a/src/main/kotlin/TransportLayer.kt
+++ b/src/main/kotlin/TransportLayer.kt
@@ -47,4 +47,11 @@ interface TransportLayer : Closeable {
      * @throws java.io.IOException if the message could not be correctly received.
      */
     fun receive(length: ULong): Array<UByte>
+
+    /**
+     * Gives the caller a "receiver" (i.e. an instance of Iterable) from which raw data of any length can be read.
+     *
+     * @return The receiver.
+     */
+    fun receiver(): Iterable<UByte>
 }
\ No newline at end of file
diff --git a/src/main/kotlin/TransportLayerJavaNet.kt b/src/main/kotlin/TransportLayerJavaNet.kt
index 1085100..d2ee2af 100644
--- a/src/main/kotlin/TransportLayerJavaNet.kt
+++ b/src/main/kotlin/TransportLayerJavaNet.kt
@@ -30,6 +30,16 @@ class TransportLayerJavaNet : TransportLayer {
     constructor(address: String, port: UShort)
     constructor(fullAddress: String)
 
+    private class InStreamIterator(val inStream: InputStream) : Iterator<UByte> {
+        override fun next(): UByte {
+            return this.inStream.readNBytes(1).first().toUByte()
+        }
+
+        override fun hasNext(): Boolean {
+            return this.inStream.available() > 0
+        }
+    }
+
     override fun close() {
         if (this.socket.isClosed) {
             return
@@ -75,4 +85,8 @@ class TransportLayerJavaNet : TransportLayer {
         }
         return Array(bytes.size) { i -> bytes[i].toUByte() }
     }
+
+    override fun receiver(): Iterable<UByte> {
+        return Iterable { InStreamIterator(this.inStream) }
+    }
 }
\ No newline at end of file