Better read semantics

OperatorFoundation · Oct 1, 2020 · 12d1095 · 12d1095
1 parent f3a7065
commit 12d1095
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/Sources/TransmissionLinux/Connection.swift b/Sources/TransmissionLinux/Connection.swift
@@ -10,6 +10,7 @@ public class Connection
     var writeLock = DispatchGroup()
 
     let socket: Socket
+    var buffer: Data = Data()
 
     public init?(host: String, port: Int)
     {
@@ -38,7 +39,7 @@ public class Connection
             return nil
         }
 
-        var data = Data(count: size)
+        var data = Data()
 
         do
         {
@@ -48,8 +49,18 @@ public class Connection
         {
             return nil
         }
+
+        buffer.append(data)
+
+        if size > buffer.count
+        {
+            return nil
+        }
+
+        let result = buffer[0..<size]
+        buffer = buffer[size..<buffer.count]
 
-        return data
+        return result
     }
 
     public func write(string: String) -> Bool