space-ios/Carthage/Checkouts/KTVHTTPCache/KTVHTTPCache/Classes/KTVHCDataStorage/KTVHCDataLoader.m

//
//  KTVHCDataLoader.m
//  KTVHTTPCache
//
//  Created by Single on 2018/6/7.
//  Copyright © 2018 Single. All rights reserved.
//

#import "KTVHCDataLoader.h"
#import "KTVHCDataReader.h"
#import "KTVHCDataResponse.h"
#import "KTVHCLog.h"

@interface KTVHCDataLoader () <KTVHCDataReaderDelegate>

@property (nonatomic, strong) KTVHCDataReader * reader;

@end

@implementation KTVHCDataLoader

+ (instancetype)loaderWithRequest:(KTVHCDataRequest *)request
{
    return [[self alloc] initWithRequest:request];
}

- (instancetype)initWithRequest:(KTVHCDataRequest *)request
{
    if (self = [super init])
    {
        KTVHCLogAlloc(self);
        self.reader = [KTVHCDataReader readerWithRequest:request];
        self.reader.delegate = self;
        KTVHCLogDataLoader(@"%p, Create loader\norignalRequest : %@\nreader : %@", self, request, self.reader);
    }
    return self;
}

- (void)dealloc
{
    KTVHCLogDealloc(self);
    [self close];
    KTVHCLogDataLoader(@"%p, Destory reader\nError : %@\nprogress : %f", self, self.error, self.progress);
}

- (void)prepare
{
    KTVHCLogDataLoader(@"%p, Call prepare", self);
    [self.reader prepare];
}

- (void)close
{
    KTVHCLogDataLoader(@"%p, Call close", self);
    [self.reader close];
}

- (KTVHCDataRequest *)request
{
    return self.reader.request;
}

- (KTVHCDataResponse *)response
{
    return self.reader.response;
}

- (NSError *)error
{
    return self.reader.error;
}

- (BOOL)didClosed
{
    return self.reader.didClosed;
}

- (BOOL)didFinished
{
    return self.reader.didFinished;
}

#pragma mark - KTVHCDataReaderDelegate

- (void)readerDidPrepared:(KTVHCDataReader *)reader
{
    [self readData];
}

- (void)readerHasAvailableData:(KTVHCDataReader *)reader
{
    [self readData];
}

- (void)reader:(KTVHCDataReader *)reader didFailed:(NSError *)error
{
    KTVHCLogDataLoader(@"%p, Callback for failed", self);
    if ([self.delegate respondsToSelector:@selector(loader:didFailed:)])
    {
        [self.delegate loader:self didFailed:error];
    }
}

- (void)readData
{
    while (YES)
    {
        @autoreleasepool
        {
            NSData * data = [self.reader readDataOfLength:1024 * 1024 * 1];
            if (self.reader.didFinished)
            {
                _progress = 1.0f;
                if ([self.delegate respondsToSelector:@selector(loader:didChangeProgress:)])
                {
                    [self.delegate loader:self didChangeProgress:_progress];
                }
                KTVHCLogDataLoader(@"%p, Callback finished", self);
                if ([self.delegate respondsToSelector:@selector(loaderDidFinished:)])
                {
                    [self.delegate loaderDidFinished:self];
                }
            }
            else if (data)
            {
                _progress = (double)self.reader.readOffset / (double)self.response.currentLength;
                if ([self.delegate respondsToSelector:@selector(loader:didChangeProgress:)])
                {
                    [self.delegate loader:self didChangeProgress:_progress];
                }
                KTVHCLogDataLoader(@"%p, read data progress %f", self, _progress);
                continue;
            }
            KTVHCLogDataLoader(@"%p, read data break", self);
            break;
        }
    }
}

@end