Now wraps Transport connections as well as NWConnections

OperatorFoundation · Apr 11, 2021 · e2d4927 · e2d4927
1 parent d888534
commit e2d4927
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Sources/Transmission/Connection.swift b/Sources/Transmission/Connection.swift
@@ -5,7 +5,7 @@ import Transport
 
 public class Connection
 {
-    var connection: NWConnection
+    var connection: Transport.Connection
     var connectLock = DispatchGroup()
     var readLock = DispatchGroup()
     var writeLock = DispatchGroup()
@@ -28,14 +28,14 @@ public class Connection
         }
     }
 
-    init?(connection: NWConnection)
+    convenience init?(connection: NWConnection)
     {
-      return self.init(transport: connection)
+        self.init(transport: connection)
     }
 
     init?(transport: Transport.Connection)
     {
-        self.connection = connection
+        self.connection = transport
 
         var success = false
 
@@ -117,7 +117,7 @@ public class Connection
         var success = false
 
         self.writeLock.enter()
-        self.connection.send(content: data, completion: NWConnection.SendCompletion.contentProcessed(
+        self.connection.send(content: data, contentContext: NWConnection.ContentContext.defaultMessage, isComplete: false, completion: NWConnection.SendCompletion.contentProcessed(
         {
             (maybeError) in