iOS 視頻緩存KTVHTTPCache原理和實(shí)現(xiàn)

目前iOS端比較常見的視頻緩存的實(shí)現(xiàn)方式主要有兩種：
1、使用iOS自帶的AVURLAsset的AVAssetResourceLoader來(lái)實(shí)現(xiàn)。
2、在客戶端搭建local服務(wù)器，local服務(wù)器作為中間者，代替客戶端請(qǐng)求服務(wù)器數(shù)據(jù)，并將獲取到的數(shù)據(jù)緩存，再提供給客戶端。
我們項(xiàng)目里使用的是KTVHTTPCache來(lái)實(shí)現(xiàn)視頻緩存，KTVHTTPCache的實(shí)現(xiàn)方式就是第二種，項(xiàng)目地址：(https://github.com/ChangbaDevs/KTVHTTPCache)。

具體實(shí)現(xiàn)：

KTVHTTPCache的使用比較簡(jiǎn)單：

NSURL *proxyURL = [KTVHTTPCache proxyURLWithOriginalURL:originalURL];
AVPlayer *player = [AVPlayer playerWithURL:proxyURL];

可以看出，它是將源視頻的URL替換成了自己定義格式的URL，這時(shí)我們其實(shí)請(qǐng)求的就是local服務(wù)器了。
核心的流程大概是這樣：


image.png

幾個(gè)核心類實(shí)現(xiàn)：

1、KTVHCHTTPServer：
用來(lái)搭建local server的，內(nèi)部使用第三方庫(kù)HTTPServer實(shí)現(xiàn):
創(chuàng)建自己的Connection類繼承自HTTPConnection

@interface KTVHCHTTPConnection : HTTPConnection

重寫子類方法，返回相應(yīng)的response類

- (NSObject<HTTPResponse> *)httpResponseForMethod:(NSString *)method URI:(NSString *)path
{
    KTVHCLogHTTPConnection(@"%p, Receive request\nmethod : %@\npath : %@\nURL : %@", self, method, path, request.url);
    NSDictionary<NSString *,NSString *> *parameters = [[KTVHCURLTool tool] parseQuery:request.url.query];
    NSURL *URL = [NSURL URLWithString:[parameters objectForKey:@"url"]];
    KTVHCDataRequest *dataRequest = [[KTVHCDataRequest alloc] initWithURL:URL headers:request.allHeaderFields];
    KTVHCHTTPResponse *response = [[KTVHCHTTPResponse alloc] initWithConnection:self dataRequest:dataRequest];
    return response;
}

創(chuàng)建response作為L(zhǎng)ocal Server數(shù)據(jù)返回體，遵循HTTPResponse協(xié)議，實(shí)現(xiàn)協(xié)議方法

@interface KTVHCHTTPResponse : NSObject <HTTPResponse>

實(shí)現(xiàn)協(xié)議方法

#pragma mark - HTTPResponse
- (NSData *)readDataOfLength:(NSUInteger)length
{
  "讀取數(shù)據(jù)最開始的入口"
   NSData *data = [self.reader readDataOfLength:length];
   KTVHCLogHTTPResponse(@"%p, Read data : %lld", self, (long long)data.length);
   if (self.reader.isFinished) {
       KTVHCLogHTTPResponse(@"%p, Read data did finished", self);
       [self.reader close];
       [self.connection responseDidAbort:self];
   }
   return data;
}
………………（省略，節(jié)省篇幅）

這樣，當(dāng)本地發(fā)生請(qǐng)求時(shí)，就會(huì)獲取KTVHCHTTPResponse內(nèi)部方法返回的數(shù)據(jù)。
2、KTVHCDataReader和KTVHCDataSourceManager
從服務(wù)器返回類可以看到，數(shù)據(jù)的入口是從KTVHCDataReader的readDataOfLength獲取的。

#pragma mark - KTVHCDataReader
- (NSData *)readDataOfLength:(NSUInteger)length
{
    [self lock];
    if (self.isClosed) {
        [self unlock];
        return nil;
    }
    if (self.isFinished) {
        [self unlock];
        return nil;
    }
    if (self.error) {
        [self unlock];
        return nil;
    }
    NSData *data = [self.sourceManager readDataOfLength:length];
    if (data.length > 0) {
        self->_readedLength += data.length;
        if (self.response.contentLength > 0) {
            self->_progress = (double)self.readedLength / (double)self.response.contentLength;
        }
    }
    KTVHCLogDataReader(@"%p, Read data : %lld", self, (long long)data.length);
    if (self.sourceManager.isFinished) {
        KTVHCLogDataReader(@"%p, Read data did finished", self);
        self->_finished = YES;
        [self close];
    }
    [self unlock];
    return data;
}

從這個(gè)方法里我們可以看到，讀取數(shù)據(jù)又走到了KTVHCDataSourceManager中去。

#pragma mark - KTVHCDataReader
- (void)prepareSourceManager
{
   "兩個(gè)數(shù)組保存兩種數(shù)據(jù)來(lái)源"
    NSMutableArray<KTVHCDataFileSource *> *fileSources = [NSMutableArray array];
    NSMutableArray<KTVHCDataNetworkSource *> *networkSources = [NSMutableArray array];
    long long min = self.request.range.start;
    long long max = self.request.range.end;
    NSArray *unitItems = self.unit.unitItems;
    for (KTVHCDataUnitItem *item in unitItems) {
        long long itemMin = item.offset;
        long long itemMax = item.offset + item.length - 1;
        if (itemMax < min || itemMin > max) {
            continue;
        }
        if (min > itemMin) {
            itemMin = min;
        }
        if (max < itemMax) {
            itemMax = max;
        }
        min = itemMax + 1;
        KTVHCRange range = KTVHCMakeRange(item.offset, item.offset + item.length - 1);
        KTVHCRange readRange = KTVHCMakeRange(itemMin - item.offset, itemMax - item.offset);
        KTVHCDataFileSource *source = [[KTVHCDataFileSource alloc] initWithPath:item.absolutePath range:range readRange:readRange];
        [fileSources addObject:source];
    }
    [fileSources sortUsingComparator:^NSComparisonResult(KTVHCDataFileSource *obj1, KTVHCDataFileSource *obj2) {
        if (obj1.range.start < obj2.range.start) {
            return NSOrderedAscending;
        }
        return NSOrderedDescending;
    }];
    "對(duì)比本地已緩存的數(shù)據(jù)和視頻數(shù)據(jù)量"
   "除了本地的如果還有未獲取的數(shù)據(jù)，就需要網(wǎng)絡(luò)請(qǐng)求獲取了"
    long long offset = self.request.range.start;
    long long length = KTVHCRangeIsFull(self.request.range) ? KTVHCRangeGetLength(self.request.range) : (self.request.range.end - offset + 1);
    for (KTVHCDataFileSource *obj in fileSources) {
        long long delta = obj.range.start + obj.readRange.start - offset;
        if (delta > 0) {
            KTVHCRange range = KTVHCMakeRange(offset, offset + delta - 1);
            KTVHCDataRequest *request = [self.request newRequestWithRange:range];
            KTVHCDataNetworkSource *source = [[KTVHCDataNetworkSource alloc] initWithRequest:request];
            [networkSources addObject:source];
            offset += delta;
            length -= delta;
        }
        offset += KTVHCRangeGetLength(obj.readRange);
        length -= KTVHCRangeGetLength(obj.readRange);
    } 
 
    if (length > 0) {
        KTVHCRange range = KTVHCMakeRange(offset, self.request.range.end);
        KTVHCDataRequest *request = [self.request newRequestWithRange:range];
        KTVHCDataNetworkSource *source = [[KTVHCDataNetworkSource alloc] initWithRequest:request];
        [networkSources addObject:source];
    }
    NSMutableArray<id<KTVHCDataSource>> *sources = [NSMutableArray array];
    [sources addObjectsFromArray:fileSources];
    [sources addObjectsFromArray:networkSources];
    self.sourceManager = [[KTVHCDataSourceManager alloc] initWithSources:sources delegate:self delegateQueue:self.internalDelegateQueue];
    [self.sourceManager prepare];
}

看到KTVHCDataSourceManager的初始化過程， 可以看出其實(shí)正常獲取數(shù)據(jù)的是KTVHCDataFileSource和KTVHCDataNetworkSource兩個(gè)類。
再看一下KTVHCDataSourceManager的readDataOfLength方法：

#pragma mark - KTVHCDataSourceManager
- (NSData *)readDataOfLength:(NSUInteger)length
{
    [self lock];
    if (self.isClosed) {
        [self unlock];
        return nil;
    }
    if (self.isFinished) {
        [self unlock];
        return nil;
    }
    if (self.error) {
        [self unlock];
        return nil;
    }
    "從Source里讀取數(shù)據(jù)"
    NSData *data = [self.currentSource readDataOfLength:length];

    self->_readedLength += data.length;
    KTVHCLogDataSourceManager(@"%p, Read data : %lld", self, (long long)data.length);
    if (self.currentSource.isFinished) {
        "一個(gè)source讀完，切換到下一個(gè)Source"
        self.currentSource = [self nextSource];
        if (self.currentSource) {
            KTVHCLogDataSourceManager(@"%p, Switch to next source, %@", self, self.currentSource);
            if ([self.currentSource isKindOfClass:[KTVHCDataFileSource class]]) {
                [self.currentSource prepare];
            }
        } else {
            KTVHCLogDataSourceManager(@"%p, Read data did finished", self);
            self->_finished = YES;
        }
    }
    [self unlock];
    return data;
}

**KTVHCDataNetworkSource和KTVHCDataFileSource
從名字就可以看出：這兩個(gè)類，一個(gè)是負(fù)責(zé)從直接從本地文件提供數(shù)據(jù)，一個(gè)是負(fù)責(zé)從網(wǎng)絡(luò)讀取之后提供數(shù)據(jù)
KTVHCDataFileSource的readDataOfLength實(shí)現(xiàn)比較明顯，就是單純從文件里讀取數(shù)據(jù)。
看下KTVHCDataNetworkSource：

- (void)ktv_download:(KTVHCDownload *)download didReceiveResponse:(KTVHCDataResponse *)response
{
    [self lock];
    if (self.isClosed || self.error) {
        [self unlock];
        return;
    }
    self->_response = response;
    NSString *path = [KTVHCPathTool filePathWithURL:self.request.URL offset:self.request.range.start];
    self.unitItem = [[KTVHCDataUnitItem alloc] initWithPath:path offset:self.request.range.start];
    KTVHCDataUnit *unit = [[KTVHCDataUnitPool pool] unitWithURL:self.request.URL];
    [unit insertUnitItem:self.unitItem];
    KTVHCLogDataNetworkSource(@"%p, Receive response\nResponse : %@\nUnit : %@\nUnitItem : %@", self, response, unit, self.unitItem);
    [unit workingRelease];
    "創(chuàng)建了兩個(gè)文件句柄，讀和寫。"
    self.writingHandle = [NSFileHandle fileHandleForWritingAtPath:self.unitItem.absolutePath];
    self.readingHandle = [NSFileHandle fileHandleForReadingAtPath:self.unitItem.absolutePath];
    [self callbackForPrepared];
    [self unlock];
}

- (void)ktv_download:(KTVHCDownload *)download didReceiveData:(NSData *)data
{
    [self lock];
    if (self.isClosed || self.error) {
        [self unlock];
        return;
    }
    @try {
        "接收到數(shù)據(jù)之后，寫入文件。"
        [self.writingHandle writeData:data];
        self.downloadLength += data.length;
        [self.unitItem updateLength:self.downloadLength];
        KTVHCLogDataNetworkSource(@"%p, Receive data : %lld, %lld, %lld", self, (long long)data.length, self.downloadLength, self.unitItem.length);
       "有可用數(shù)據(jù)了，需要回調(diào)通知。"
        [self callbackForHasAvailableData];
    } @catch (NSException *exception) {
        NSError *error = [KTVHCError errorForException:exception];
        KTVHCLogDataNetworkSource(@"%p, write exception\nError : %@", self, error);
        [self callbackForFailed:error];
        if (!self.downloadCalledComplete) {
            KTVHCLogDataNetworkSource(@"%p, Cancel download task when write exception", self);
            [self.downlaodTask cancel];
            self.downlaodTask = nil;
        }
    }
    [self unlock];
}

可以看出，兩個(gè)source的實(shí)現(xiàn)比較類似，只不過KTVHCDataNetworkSource多了一個(gè)從網(wǎng)絡(luò)獲取數(shù)據(jù)寫入文件的步驟，其實(shí)最終提供數(shù)據(jù)還是通過文件讀取的方式。
一旦有可用數(shù)據(jù)，就通過delegate的方式一直回調(diào)，通知response類有可用數(shù)據(jù)。

#pragma mark -  KTVHCHTTPResponse
- (void)ktv_readerDidPrepare:(KTVHCDataReader *)reader
{
    KTVHCLogHTTPResponse(@"%p, Prepared", self);
    if (self.reader.isPrepared && self.waitingResponse == YES) {
        KTVHCLogHTTPResponse(@"%p, Call connection did prepared", self);
        [self.connection responseHasAvailableData:self];
    }
}
"這個(gè)回調(diào)獲取有可用的數(shù)據(jù)的通知。"

- (void)ktv_readerHasAvailableData:(KTVHCDataReader *)reader
{
    KTVHCLogHTTPResponse(@"%p, Has available data", self);
    "這個(gè)方法就會(huì)觸發(fā)response的readDataOfLength"
    [self.connection responseHasAvailableData:self];
}

- (void)ktv_reader:(KTVHCDataReader *)reader didFailWithError:(NSError *)error
{
    KTVHCLogHTTPResponse(@"%p, Failed\nError : %@", self, error);
    [self.reader close];
    [self.connection responseDidAbort:self];
}
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容