Add optional delegate to network

0xLeif · Apr 3, 2024 · a6dafc2 · a6dafc2
1 parent e125f52
commit a6dafc2
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/Sources/Network/Network.swift b/Sources/Network/Network.swift
@@ -2,7 +2,11 @@ import Foundation
 
 /// A class that implements the `Networking` protocol to perform network requests using URLSession.
 open class Network: Networking {
-    public init() { }
+    public weak var delegate: URLSessionTaskDelegate?
+
+    public init(delegate: URLSessionTaskDelegate? = nil) {
+        self.delegate = delegate
+    }
 
     open func request(
         for url: URL,
@@ -17,7 +21,10 @@ open class Network: Networking {
         request.httpBody = body
 
         return DataResponse(
-            try await URLSession.shared.data(for: request)
+            try await URLSession.shared.data(
+                for: request,
+                delegate: delegate
+            )
         )
     }