Add feedback from code review

nachoBonafonte · Aug 17, 2023 · f85280d · f85280d
1 parent 90baefd
commit f85280d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Sources/Instrumentation/URLSession/URLSessionInstrumentation.swift b/Sources/Instrumentation/URLSession/URLSessionInstrumentation.swift
@@ -30,6 +30,8 @@ public class URLSessionInstrumentation {
     private let queue = DispatchQueue(label: "io.opentelemetry.ddnetworkinstrumentation")
 
     static var instrumentedKey = "io.opentelemetry.instrumentedCall"
+
+    static let avAssetDownloadTask: AnyClass? = NSClassFromString("__NSCFBackgroundAVAssetDownloadTask")
 
     public private(set) var tracer: Tracer
 
@@ -572,8 +574,8 @@ public class URLSessionInstrumentation {
 
     private func urlSessionTaskWillResume(_ task: URLSessionTask) {
         // AV Asset Tasks cannot be auto instrumented, they dont include request attributes, skip them
-        if let avAssetDownloadTask = NSClassFromString("__NSCFBackgroundAVAssetDownloadTask"),
-           task.isKind(of: avAssetDownloadTask) {
+        if let avAssetTaskClass = Self.avAssetDownloadTask,
+           task.isKind(of: avAssetTaskClass) {
             return
         }