Added UDP

OperatorFoundation · Sep 14, 2020 · f816a0e · f816a0e
1 parent df8607a
commit f816a0e
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 6 deletions.
diff --git a/Sources/Transmission/Connection.swift b/Sources/Transmission/Connection.swift
@@ -9,16 +9,22 @@ public class Connection
     var readLock = DispatchGroup()
     var writeLock = DispatchGroup()
 
-    public convenience init?(host: String, port: Int)
+    public convenience init?(host: String, port: Int, type: ConnectionType = .tcp)
     {
         let nwhost = NWEndpoint.Host(host)
 
         let port16 = UInt16(port)
         let nwport = NWEndpoint.Port(integerLiteral: port16)
 
-        let nwconnection = NWConnection(host: nwhost, port: nwport, using: .tcp)
-
-        self.init(connection: nwconnection)
+        switch type
+        {
+            case .tcp:
+                let nwconnection = NWConnection(host: nwhost, port: nwport, using: .tcp)
+                self.init(connection: nwconnection)
+            case .udp:
+                let nwconnection = NWConnection(host: nwhost, port: nwport, using: .udp)
+                self.init(connection: nwconnection)
+        }
     }
 
     init?(connection: NWConnection)
@@ -116,3 +122,9 @@ public class Connection
         return success
     }
 }
+
+public enum ConnectionType
+{
+    case udp
+    case tcp
+}
diff --git a/Sources/Transmission/Listener.swift b/Sources/Transmission/Listener.swift
@@ -15,12 +15,21 @@ public class Listener
     let queue: BlockingQueue<Connection> = BlockingQueue<Connection>()
     let lock: DispatchGroup = DispatchGroup()
 
-    public init?(port: Int)
+    public init?(port: Int, type: ConnectionType = .tcp)
     {
         let port16 = UInt16(port)
         let nwport = NWEndpoint.Port(integerLiteral: port16)
+
+        var params: NWParameters!
+        switch type
+        {
+            case .tcp:
+                params = NWParameters.tcp
+            case .udp:
+                params = NWParameters.udp
+        }
 
-        guard let listener = try? NWListener(using: .tcp, on: nwport) else {return nil}
+        guard let listener = try? NWListener(using: params, on: nwport) else {return nil}
         self.listener = listener
 
         self.listener.newConnectionHandler =

diff --git a/Sources/TransmissionLinux/Connection.swift b/Sources/TransmissionLinux/Connection.swift
@@ -67,3 +67,9 @@ public class Connection
         return true
     }
 }
+
+public enum ConnectionType
+{
+    case udp
+    case tcp
+}