Added return value URLSessionDataTask for OpenGraph.fetch to have the…

… ability to cancel task if needed
bdhazman · Jan 24, 2020 · 0170d49 · 0170d49
1 parent 93d8393
commit 0170d49
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/OpenGraph/OpenGraph.swift b/OpenGraph/OpenGraph.swift
@@ -4,7 +4,8 @@ public struct OpenGraph {
 
     public let source: [OpenGraphMetadata: String]
 
-    public static func fetch(url: URL, headers: [String: String]? = nil, completion: @escaping (Result<OpenGraph, Error>) -> Void) {
+    @discardableResult
+    public static func fetch(url: URL, headers: [String: String]? = nil, completion: @escaping (Result<OpenGraph, Error>) -> Void) -> URLSessionDataTask {
         var mutableURLRequest = URLRequest(url: url)
         headers?.compactMapValues { $0 }.forEach {
             mutableURLRequest.setValue($1, forHTTPHeaderField: $0)
@@ -19,6 +20,7 @@ public struct OpenGraph {
             }
         })
         task.resume()
+        return task
     }
 
     private static func handleFetchResult(data: Data?, response: URLResponse?, completion: @escaping (Result<OpenGraph, Error>) -> Void) {