使用 NSRecursive Lock 等待操作完成

Question

使用 NSRecursiveLock 有目的地等待后台线程上的操作完成会节省吗？这是一个示例：

我有一个类，我想要一个可以异步或同步的 loadParts 函数。异步函数可能会被提前调用，以便可以在实际需要数据之前加载这些部分。同步的应该检查数据是否已经加载，或者当前正在加载。如果已经加载，则直接返回数据；如果当前已加载，则应等待加载然后返回；如果它甚至没有被加载，那么它应该同步加载它。这是我尝试使用的代码：

// Private function to be run either on main thread or
// background thread
-(void)_loadParts
{
    [_loadingPartsLock lock];
    _loadingParts = YES;

    // Do long loading operation

    _loadingParts = NO;
    _partsLoaded = YES;
    [_loadingPartsLock unlock];
}

// Asynchronous loading of parts
-(void)preloadParts
{
    if( _loadingParts || _partsLoaded )
        return;

    [self performSelectorInBackground:@selector(_loadParts) withObject:nil];
}

// Synchronous loading of parts
-(void)loadParts
{
    if( _loadingParts )
    {
        [_loadingPartsLock lock];
        [_loadingPartsLock unlock];
    }

    if( !_partsLoaded )
    {
         [self _loadParts];
    }
}

这是安全/有效的方法吗？我已经看到了一些可能的问题。在没有锁的情况下设置和测试 BOOL 的值是否是线程安全的？如果在后台线程仍在加载时调用同步函数，我还会在同步函数中锁定两次。

有没有更通用、更好的方法来实现这个功能？

谢谢！

Answer 1

一个更好的解决方案是使用dispatch_queue或NSOperationQueue（配置为串行操作）。

将加载操作排队，然后将完成后应该发生的任何事情排队。如果“完成”操作是“告诉主线程更新”，那很好——在主线程上执行一个方法，该方法实际上是一个触发更新以响应现在加载的数据的事件。

这完全避免了与锁定相关的问题和开销，同时还解决了“完成”通知问题，而不需要某种轮询机制。

Answer 2

受到 bbum 的回答的启发，我找到了一个使用 NSOperationQueue 的解决方案，但并不完全按照他描述的方式。在我的 preloadParts 函数中，我创建并存储一个加载操作，它是运行我的后台线程函数的 NSInitationOperation 的实例。然后我将其添加到 NSOperationQueue 中。

如果在任何时候，另一个类请求数据。我首先检查数据是否已加载（变量是否已设置）。如果没有，我检查该操作是否在队列中。如果是，那么我调用 [_loadOperation waitUntilFinished]。否则，我将其添加到操作队列中，并使用 waitUntilFinished 参数。这是我想出的代码：

-(void)preloadCategories
{
    if( [[_operationQueue operations] containsObject:_loadOperation] )
        return;

    [_operationQueue addOperation:_loadOperation];
}

-(CCPart*)getCategoryForName:(NSString*)name
{
    if( nil == _parts )
    {
        [self loadCategories];
    }
    return [_parts objectForKey:name];
}

-(void)loadCategories
{
    if( nil != _parts )
        return;

    if( [[_operationQueue operations] containsObject:_loadOperation] )
    {
        [_loadOperation waitUntilFinished];
    }
    else
    {
        [_operationQueue addOperations:[NSArray arrayWithObject:_loadOperation]
                     waitUntilFinished:YES];
    }
}

-(void)_loadCategories
{
    // Function that actually does the loading and sets _parts to be an array of the data
    _parts = [NSArray array];
}

在初始化函数中，我设置 _operationQueue 和 _loadOperation 如下：

_operationQueue = [[NSOperationQueue alloc] init];
            _loadOperation = [[NSInvocationOperation alloc] initWithTarget:self 
                                                                  selector:@selector(_loadCategories) 
                                                                    object:nil];