DFPlayerResourceLoader.m 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. //
  2. // DFPlayerResourceLoader.m
  3. // DFPlayer
  4. //
  5. // Created by ihoudf on 2017/7/30.
  6. // Copyright © 2017年 ihoudf. All rights reserved.
  7. //
  8. #import "DFPlayerResourceLoader.h"
  9. #import <MobileCoreServices/MobileCoreServices.h>
  10. #import "DFPlayerFileManager.h"
  11. @interface DFPlayerResourceLoader ()
  12. @property (nonatomic, strong) NSMutableArray<AVAssetResourceLoadingRequest *> *requestList;
  13. @property (nonatomic, strong) DFPlayerRequestManager *requestManager;
  14. @end
  15. @implementation DFPlayerResourceLoader
  16. - (instancetype)init {
  17. if (self = [super init]) {
  18. if (_requestList.count >= 0) {
  19. [_requestList removeAllObjects];
  20. }
  21. }
  22. return self;
  23. }
  24. - (NSMutableArray<AVAssetResourceLoadingRequest *> *)requestList{
  25. if (!_requestList) {
  26. _requestList = [NSMutableArray array];
  27. }
  28. return _requestList;
  29. }
  30. - (void)stopDownload{
  31. if (self.requestManager) {
  32. self.requestManager.cancel = YES;
  33. }
  34. }
  35. #pragma mark - RequestTaskDelegate
  36. - (void)requestManagerDidReceiveResponseWithStatusCode:(NSInteger)statusCode{
  37. if (self.checkStatusBlock) {
  38. self.checkStatusBlock(statusCode);
  39. }
  40. }
  41. - (void)requestManagerDidReceiveData {
  42. [self processRequestList];
  43. }
  44. - (void)requestManagerIsCached:(BOOL)isCached{
  45. if (self.delegate && [self.delegate respondsToSelector:@selector(loader:isCached:)]) {
  46. [self.delegate loader:self isCached:isCached];
  47. }
  48. }
  49. - (void)requestManagerDidCompleteWithError:(NSInteger)errorCode{
  50. if (self.delegate && [self.delegate respondsToSelector:@selector(loader:requestError:)]) {
  51. [self.delegate loader:self requestError:errorCode];
  52. }
  53. }
  54. #pragma mark - AVAssetResourceLoaderDelegate
  55. - (BOOL)resourceLoader:(AVAssetResourceLoader *)resourceLoader shouldWaitForLoadingOfRequestedResource:(AVAssetResourceLoadingRequest *)loadingRequest {
  56. [self addLoadingRequest:loadingRequest];
  57. return YES;
  58. }
  59. - (void)resourceLoader:(AVAssetResourceLoader *)resourceLoader didCancelLoadingRequest:(AVAssetResourceLoadingRequest *)loadingRequest {
  60. [self.requestList removeObject:loadingRequest];
  61. }
  62. #pragma mark - 处理LoadingRequest
  63. - (void)addLoadingRequest:(AVAssetResourceLoadingRequest *)loadingRequest {
  64. [self.requestList addObject:loadingRequest];
  65. @synchronized(self) {
  66. if (self.requestManager) {
  67. long long offset = loadingRequest.dataRequest.requestedOffset;
  68. if (offset >= 0 && offset <= self.requestManager.cacheLength) {
  69. //数据已经缓存,则直接完
  70. [self processRequestList];
  71. }else {
  72. //数据还没缓存,则等待数据下载;如果是Seek操作,则重新请求
  73. [self newTaskWithLoadingRequest:loadingRequest cache:NO];
  74. }
  75. }else {
  76. [self newTaskWithLoadingRequest:loadingRequest cache:YES];
  77. }
  78. }
  79. }
  80. - (void)newTaskWithLoadingRequest:(AVAssetResourceLoadingRequest *)loadingRequest cache:(BOOL)cache {
  81. NSUInteger fileLength = 0;
  82. if (self.requestManager) {
  83. fileLength = self.requestManager.fileLength;
  84. self.requestManager.cancel = YES;
  85. }
  86. self.requestManager = [[DFPlayerRequestManager alloc] initWithUrl:loadingRequest.request.URL];
  87. if (fileLength > 0) {
  88. self.requestManager.fileLength = fileLength;
  89. }
  90. self.requestManager.delegate = self;
  91. self.requestManager.isHaveCache = self.isCached;
  92. self.requestManager.isObserveFileModifiedTime = self.isObserveFileModifiedTime;
  93. [self.requestManager requestStart];
  94. }
  95. - (void)processRequestList {
  96. NSMutableArray * finishRequestList = [NSMutableArray array];
  97. for (AVAssetResourceLoadingRequest * loadingRequest in self.requestList) {
  98. if ([self finishLoadingWithLoadingRequest:loadingRequest]) {
  99. [finishRequestList addObject:loadingRequest];
  100. }
  101. }
  102. [self.requestList removeObjectsInArray:finishRequestList];
  103. }
  104. - (BOOL)finishLoadingWithLoadingRequest:(AVAssetResourceLoadingRequest *)loadingRequest {
  105. //填充信息
  106. CFStringRef contentType = UTTypeCreatePreferredIdentifierForTag(kUTTagClassMIMEType, (__bridge CFStringRef)(MimeType), NULL);
  107. loadingRequest.contentInformationRequest.contentType = CFBridgingRelease(contentType);
  108. loadingRequest.contentInformationRequest.byteRangeAccessSupported = YES;
  109. loadingRequest.contentInformationRequest.contentLength = self.requestManager.fileLength;
  110. //读文件,填充数据
  111. NSUInteger cacheLength = self.requestManager.cacheLength;//已经缓存的数据长度
  112. NSUInteger requestedOffset = (long)loadingRequest.dataRequest.requestedOffset;
  113. if (loadingRequest.dataRequest.currentOffset != 0) {//如果当前下载长度不为0
  114. requestedOffset = (long)loadingRequest.dataRequest.currentOffset;
  115. }
  116. //当前下载长度-要请求的长度
  117. NSUInteger canReadLength = cacheLength - requestedOffset;
  118. NSUInteger respondLength = MIN(canReadLength, loadingRequest.dataRequest.requestedLength);
  119. [loadingRequest.dataRequest respondWithData:[DFPlayerFileManager df_readTempFileDataWithOffset:requestedOffset length:respondLength]];
  120. //如果完全响应了所需要的数据,则完成
  121. NSUInteger nowendOffset = requestedOffset + canReadLength;
  122. NSUInteger reqEndOffset = (long)loadingRequest.dataRequest.requestedOffset + (long)loadingRequest.dataRequest.requestedLength;
  123. if (nowendOffset >= reqEndOffset) {
  124. [loadingRequest finishLoading];
  125. return YES;
  126. }
  127. return NO;
  128. }
  129. @end