NSStream簡介

stream(流)是編程中的一個基本抽象概念：一系列的位有序的從一個點傳輸?shù)搅硪粋€點。Cocoa提供了三個類代表steam以便于你在程序中使用：NSStream，NSInputStream，NSOutputStream。使用這些類的實例，你可以讀或者寫數(shù)據(jù)從文件或者應(yīng)用程序的內(nèi)存。你也可使用在基于socket連接的網(wǎng)絡(luò)中使用這些對象和遠程主機交換數(shù)據(jù)。你也可繼承stream類而獲取專有的stream操作。常見的Stream應(yīng)用場景有：讀/寫取文件，socket通信, 從NSData中讀/寫數(shù)據(jù), 寫數(shù)據(jù)到buffer中。

NSInputStream

NSInputStream 是輸入流，對客戶端而言，就是讀數(shù)據(jù)。

讀文件

@interface ViewController ()<NSStreamDelegate>
@property (nonatomic,strong)NSInputStream *istream;
@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    //獲取所讀文件的路徑   
    NSString *path = [[NSBundle mainBundle] pathForResource:@"init" ofType:@"json"];
    [self setUpStreamForFile:path];
}

-(void)setUpStreamForFile:(NSString *)path
{
    //創(chuàng)建NSInputStream
    self.istream = [[NSInputStream alloc] initWithFileAtPath:path];
    // 設(shè)置delegate
    self.istream.delegate = self;
    // 加入到Runloop中
    [self.istream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
    // 打開流
    [self.istream open];
}

- (void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode;
{
    switch (eventCode) {
        // 有數(shù)據(jù)可讀
        case NSStreamEventHasBytesAvailable:
        {
            //讀取數(shù)據(jù)并打印
            NSMutableData *data = [[NSMutableData alloc] init];
            uint8_t buf[2048];
            NSInteger len = 0;
            len = [(NSInputStream *)aStream read:buf maxLength:2048];
            if (len) {
                [data appendBytes:(const void *)buf length:len];
                NSString *str = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
                NSLog(@"%@", str);
            }else{
                NSLog(@"no buffer");
            }
            break;
        }
        //讀到了流的結(jié)尾
        case NSStreamEventEndEncountered:{
            // 關(guān)閉流
            [aStream close];
            [aStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
            aStream = nil;
            break;
        }
            
        default:
            break;
    }
}

NSOutputStream

NSOutputStream是輸入流，對于客戶端而言，就是寫數(shù)據(jù)。

#import "ViewController.h"

@interface ViewController ()<NSStreamDelegate>
{
    NSString *pathtxt;
}
@property (nonatomic,strong)NSOutputStream *ostream;
@property (nonatomic,strong)NSData *data;
@property (nonatomic,assign)NSInteger readBytes;
@property (nonatomic,assign)NSInteger byteIndex;

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    NSString *path = [[NSBundle mainBundle] pathForResource:@"init" ofType:@"json"];
    self.data = [NSData dataWithContentsOfFile:path];
    pathtxt =  NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES).firstObject;
    //建議這個路徑一定要是沙盒中的路徑，放在工程目錄里面是無法寫入數(shù)據(jù)的。
    pathtxt =  [pathtxt stringByAppendingPathComponent:@"cache.json"];
    [self createOutputStream];
}


-(void)createOutputStream
{
    self.ostream = [[NSOutputStream alloc] initToFileAtPath:pathtxt append:YES];
    self.ostream.delegate = self;
    [self.ostream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
    [self.ostream open];
}



- (void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode;
{
    switch (eventCode) {
        case NSStreamEventHasSpaceAvailable:
        {
            self.readBytes += self.byteIndex;
            NSUInteger data_len = [_data length];
            NSUInteger len = (data_len - self.readBytes >= 1024) ? 1024 : (data_len - self.readBytes);
            uint8_t buf[len];
            [self.data getBytes:buf range:NSMakeRange(self.readBytes, len)];
            len = [(NSOutputStream *)aStream write:buf maxLength:sizeof(buf)];
            self.byteIndex = len;
            break;
        }
        case NSStreamEventEndEncountered:
        {
            [aStream close];
            [aStream removeFromRunLoop:[NSRunLoop currentRunLoop]
                              forMode:NSDefaultRunLoopMode];
            aStream = nil; // oStream is instance variable
            break;
        }

        default:
            break;
    }
}


@end

RunLoop的作用

我們都知道RunLoop可以保留線程不釋放，有任務(wù)的時候執(zhí)行，沒有任務(wù)的時候休息并且不阻塞UI線程。我們在對流進行讀／寫操作時候，如果沒有runloop我們需要一次性將流中的數(shù)據(jù)讀完或者寫完，這顯然是不現(xiàn)實的，那么我們可以通過另一種方式進行分段讀取。

while (1) {
    if (len == 0) break;
    
    if ([oStream hasSpaceAvailable])
    {
        (void)strncpy(buf, readBytes, len);
        readBytes += len;
        if ([oStream write:(const uint8_t *)buf maxLength:len] == -1)
        {
            [self handleError:[oStream streamError]];
            break;
        }
        [bytesWritten setIntValue:[bytesWritten intValue]+len];
        len = (([data length] - [bytesWritten intValue] >= 1024) ? 1024 : [data length] - [bytesWritten intValue]);
    }
}

使用while循環(huán)，每次讀取一定的字節(jié)并記錄讀取的位置，知道讀取完畢，結(jié)束white循環(huán)。這么做也能達到我們的目的，但是顯然這會造成阻塞線程。所以最好的方法還是使用runloop監(jiān)聽數(shù)據(jù)源是否可讀/寫。

參考

Stream Programming Guide
南峰子的博客

交流群

移動開發(fā)交流群：264706196