Fix possible memory leak

alcatraz · May 15, 2016 · e099f9b · e099f9b
1 parent 046fc10
commit e099f9b
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/Alcatraz/Helpers/ATZDownloader.m b/Alcatraz/Helpers/ATZDownloader.m
@@ -25,7 +25,7 @@
 
 @interface ATZDownloader()
 @property (strong, nonatomic) NSMutableDictionary *callbacks;
-@property (strong, nonatomic) NSURLSession *urlSession;
+@property (strong, nonatomic, readonly) NSURLSession *urlSession;
 @end
 
 
@@ -135,16 +135,14 @@ - (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task
 #pragma mark - Private
 
 - (NSURLSession *)urlSession {
-    if (_urlSession) return _urlSession;
-
-    _urlSession = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration]
-                                                delegate:self
-                                           delegateQueue:[NSOperationQueue mainQueue]];
-    return _urlSession;
+    static dispatch_once_t onceToken;
+    static NSURLSession *session = nil;
+    dispatch_once(&onceToken, ^{
+        session = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration]
+                                                    delegate:self
+                                               delegateQueue:[NSOperationQueue mainQueue]];
+    });
+    return session;
 }
 
-
-
-
-
 @end