You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.2 KiB
57 lines
1.2 KiB
// Copyright 2015-present 650 Industries. All rights reserved.
|
|
|
|
#import <EXFileSystem/EXTaskHandlersManager.h>
|
|
|
|
@interface EXTaskHandlersManager ()
|
|
|
|
@property (nonatomic, strong) NSMutableDictionary<NSString *, NSURLSessionTask *> *resumableDownloads;
|
|
|
|
@end
|
|
|
|
@implementation EXTaskHandlersManager
|
|
|
|
- (instancetype)init
|
|
{
|
|
if (self = [super init]) {
|
|
_resumableDownloads = [NSMutableDictionary dictionary];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (void)registerTask:(NSURLSessionTask *)task uuid:(NSString *)uuid
|
|
{
|
|
_resumableDownloads[uuid] = task;
|
|
}
|
|
|
|
- (NSURLSessionTask * _Nullable)taskForId:(NSString *)uuid
|
|
{
|
|
return _resumableDownloads[uuid];
|
|
}
|
|
|
|
- (NSURLSessionDownloadTask * _Nullable)downloadTaskForId:(NSString *)uuid
|
|
{
|
|
NSURLSessionTask *task = [self taskForId:uuid];
|
|
if ([task isKindOfClass:[NSURLSessionDownloadTask class]]) {
|
|
return (NSURLSessionDownloadTask *)task;
|
|
}
|
|
|
|
return nil;
|
|
}
|
|
|
|
- (NSURLSessionUploadTask * _Nullable)uploadTaskForId:(NSString *)uuid
|
|
{
|
|
NSURLSessionTask *task = [self taskForId:uuid];
|
|
if ([task isKindOfClass:[NSURLSessionUploadTask class]]) {
|
|
return (NSURLSessionDownloadTask *)task;
|
|
}
|
|
|
|
return nil;
|
|
}
|
|
|
|
- (void)unregisterTask:(NSString *)uuid
|
|
{
|
|
[_resumableDownloads removeObjectForKey:uuid];
|
|
}
|
|
|
|
@end
|