使用ReadabilityHandler Block

Question

基本设置

我使用NSTask来运行优化图像的过程。此过程将输出数据写入stdout。我使用readabilityHandler的{​​{1}}属性来捕获该数据。这是缩写设置：

NSTask

然后我这样打电话给NSTask：

NSTask *task = [[NSTask alloc] init];
[task setArguments:arguments];  // arguments defined above

NSPipe *errorPipe = [NSPipe pipe];
[task setStandardError:errorPipe];
NSFileHandle *errorFileHandle = [errorPipe fileHandleForReading];

NSPipe *outputPipe = [NSPipe pipe];
[task setStandardOutput:outputPipe];
NSFileHandle *outputFileHandle = [outputPipe fileHandleForReading];

NSMutableData *outputData = [[NSMutableData alloc] init];
NSMutableData *errorOutputData = [[NSMutableData alloc] init];

outputFileHandle.readabilityHandler = ^void(NSFileHandle *handle) { 
      NSLog(@"Appending data for %@", inputPath.lastPathComponent); 
      [outputData appendData:handle.availableData]; 
};

errorFileHandle.readabilityHandler = ^void(NSFileHandle *handle) { 
       [errorOutputData appendData:handle.availableData]; 
};

（这都是在后台调度队列上完成的）。接下来，我将检查NSTask的返回值：

[task setLaunchPath:_pathToJPEGOptim];
[task launch];
[task waitUntilExit];

问题

大约98％的时间，这非常有效。但是，似乎if ([task terminationStatus] == 0) { newSize = outputData.length; if (newSize <= 0) { NSString *errorString = [[NSString alloc] initWithData:errorOutputData encoding:NSUTF8StringEncoding]; NSLog(@"ERROR string: %@", errorString); } // Truncated for brevity... } CAN在readabilityHandler块运行之前触发。这是一个屏幕截图，显示可读性处理程序在任务退出后运行：

所以这显然是运行readabilityHandler的调度队列和我已经解雇了我的NSTask的调度队列之间的竞争条件。我的问题是：我怎么能确定readabilityHandler已经完成了？如果当NSTask告诉我它已完成时，我怎么能打败这种竞争条件呢？

注意：

我知道NSTask有一个可选的-waitUntilExit块。但是文档声明这个块不能保证在completionHandler返回之前运行，这意味着它可以开始运行甚至超过-waitUntilExit。这将使竞争条件更有可能。

Answer 1

创建一个初始值为0的信号量。在可读性处理程序中，检查从-lboost_system返回的数据对象的长度是否为0.如果是，则表示文件结束。在这种情况下，请发信号通知信号。

然后，在availableData返回后，等待信号量两次（对于您正在阅读的每个管道一次）。当那些等待返回时，你已经获得了所有数据。

Answer 2

好的，经过多次试错，以下是处理它的正确方法：

1。不要使用-AvailableData

在可读性处理程序块中，请勿使用-availableData方法。这有很奇怪的副作用，有时候不会捕获所有可用的数据，并且会干扰系统尝试使用空的NSData对象调用处理程序以通知管道关闭，因为-availableData阻塞直到数据实际上是可用的。

2。使用-readDataOfLength：

相反，在可读性处理程序块中使用-readDataOfLength:NSUIntegerMax。使用这种方法，处理程序正确接收一个空的NSData对象，您可以使用该对象来检测管道的关闭并发出信号量信号。

3。当心macOS 10.12！

Apple在10.13中修复了一个绝对关键的错误：在旧版本的macOS上，如果没有数据可读，则永远不会调用可读性处理程序。也就是说，它们永远不会被调用零长度数据来表明它们已经完成。这导致使用信号量方法永久挂起，因为信号量永远不会增加。为了解决这个问题，我测试了macOS 10.12或更低版本，如果我在旧操作系统上运行，我会使用一次调用dispatch_semaphore_wait（），该调用与NSTask的completionHandler块中对dispatch_semaphore_signal（）的单次调用配对。我有完成块休眠0.2秒，以允许处理程序执行。这显然是一个非常丑陋的黑客攻击，但它确实有效。如果我在10.13加上，我有不同的可读性处理程序，它们发出信号量信号（一次来自错误处理程序，一次来自正常输出处理程序），我仍然从completionHandler块发出信号。在启动任务后，这些调用与dispatch_semaphore_wait（）的3次调用配对。在这种情况下，完成块中不需要延迟，因为当fileHandle完成时，macOS正确地调用具有零长度数据的可读性处理程序。

实施例

（注意：假设在我的原始问题示例中定义了东西。为了便于阅读，缩短了此代码。）

// Create the semaphore
dispatch_semaphore_t sema = dispatch_semaphore_create(0);

// Define a handler to collect output data from our NSTask
outputFileHandle.readabilityHandler = ^void(NSFileHandle *handle)
{
    // DO NOT use -availableData in these handlers.
    NSData *newData = [handle readDataOfLength:NSUIntegerMax];
    if (newData.length == 0) 
    {
        // end of data signal is an empty data object.
        outputFileHandle.readabilityHandler = nil;
        dispatch_semaphore_signal(sema);
    } 
    else 
    {
       [outputData appendData:newData];
    }
};

// Repeat the above for the 'errorFileHandle' readabilityHandler.


[task launch];

// two calls to wait because we are going to signal the semaphore once when
// our 'outputFileHandle' pipe closes and once when our 'errorFileHandle' pipe closes                     
dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);

// ... do stuff when the task is done AND the pipes have finished handling data.

// After doing stuff, release the semaphore
dispatch_release(sema);
sema = NULL;