Merge pull request #381 from nachoBonafonte/urlsession-configure-delegates

Author: Ignacio Bonafonte

Sources/Instrumentation/URLSession/URLSessionInstrumentation.swift

@@ -65,7 +65,7 @@ public class URLSessionInstrumentation {
             #selector(URLSessionDataDelegate.urlSession(_:dataTask:didBecome:) as (URLSessionDataDelegate) -> ((URLSession, URLSessionDataTask, URLSessionStreamTask) -> Void)?)
         ]
 #endif
-        let classes = InstrumentationUtils.objc_getClassList()
+        let classes = configuration.delegateClassesToInstrument ?? InstrumentationUtils.objc_getClassList()
         let selectorsCount = selectors.count
         DispatchQueue.concurrentPerform(iterations: classes.count) { iteration in
             let theClass: AnyClass = classes[iteration]

Sources/Instrumentation/URLSession/URLSessionInstrumentationConfiguration.swift

@@ -20,7 +20,8 @@ public struct URLSessionInstrumentationConfiguration {
                 injectCustomHeaders: ((inout URLRequest, Span?) -> Void)? = nil,
                 createdRequest: ((URLRequest, Span) -> Void)? = nil,
                 receivedResponse: ((URLResponse, DataOrFile?, Span) -> Void)? = nil,
-                receivedError: ((Error, DataOrFile?, HTTPStatus, Span) -> Void)? = nil)
+                receivedError: ((Error, DataOrFile?, HTTPStatus, Span) -> Void)? = nil,
+                delegateClassesToInstrument: [AnyClass]? = nil)
     {
         self.shouldRecordPayload = shouldRecordPayload
         self.shouldInstrument = shouldInstrument
@@ -31,6 +32,7 @@ public struct URLSessionInstrumentationConfiguration {
         self.createdRequest = createdRequest
         self.receivedResponse = receivedResponse
         self.receivedError = receivedError
+        self.delegateClassesToInstrument = delegateClassesToInstrument
     }
 
     // Instrumentation Callbacks
@@ -65,4 +67,7 @@ public struct URLSessionInstrumentationConfiguration {
 
     ///  Called before the span is ended, it allows to add extra information to the Span
     public var receivedError: ((Error, DataOrFile?, HTTPStatus, Span) -> Void)?
+    
+    ///  The array of URLSession delegate classes that will be instrumented by the library, will autodetect if nil is passed.
+    public var delegateClassesToInstrument: [AnyClass]?
 }