使用 NSDictionary 元素缓慢创建 NSMutableArray

Question

这段代码是一种创建数组以供多个其他类使用的方法。输入是来自 CoreData 提取的数组，类型为 NSDictionaryResultType。

其中 3 个字段是我需要分解为数组的字符串，因此是 componentsSeparatedByString。

生成的数组 _dataProductionArray 运行良好 --- 但是 --- 这段代码需要整整 5 秒来处理大约 32,000 条记录。

任何帮助指出导致这种缓慢性能的明显错误将不胜感激！

NSMutableArray *dataArray = [NSMutableArray array];
int j = 0;
int maxNumMonths = 0;
for (id obj in _dictionaries) {
    if ([_dictionaries[j] [@"month"] length] >0 ) {
        // get production values
        NSArray *aItems   = [_dictionaries[j] [@"prodA"] componentsSeparatedByString:@","];
        NSArray *bItems   = [_dictionaries[j] [@"prodB"] componentsSeparatedByString:@","];
        NSArray *monthItems = [_dictionaries[j] [@"month"] componentsSeparatedByString:@","];

        NSMutableArray *productionAArray = [NSMutableArray array];
        NSMutableArray *productionBArray = [NSMutableArray array];
        int monthLoop = 1;
        for (NSNumber *month in monthItems) {
            if (monthLoop <= MONTHS_OF_PRODUCTION) {
                if ([month intValue] == monthLoop) {
                    [productionAArray addObject:[aItems objectAtIndex:monthLoop-1]];
                    [productionBArray addObject:[bItems objectAtIndex:monthLoop-1]];
                    productionCount ++;
                    if (monthLoop > maxNumMonths)
                        maxNumMonths = monthLoop;
                }
            }
            monthLoop++;
        }

        NSDictionary *arrayItem = @{@"name":_dictionaries[j] [@"name"],
                                    @"type":[NSString stringWithFormat:@"%@",_dictionaries[j] [@"type"]],
                                    @"height":[NSString stringWithFormat:@"%@",_dictionaries[j] [@"height"]],
                                    @"aArray":productionAArray,
                                    @"bArray":productionBArray,
                                    };
        [dataArray addObject:arrayItem];
    }
    j++;
}

_dataProductionArray = [NSArray arrayWithArray:dataArray];

Answer 1

我可以看到您可以在循环中进行一些优化，但我不确定这些优化有多大帮助（特别是如果编译器正在执行这些优化）。根本问题是 32k 是很多迭代。

您是否需要一次获得所有 32k 结果？通过完成这项工作lazily，您可以显着改善用户体验，因为 UI 需要转换后的记录。

这种方法是使 dataProductionArray 成为一个可变字典，由 NSNumber 索引索引。然后，而不是...

// replace this
self.dataProductionArray[128];

// with this
[self dataProductionAtIndex:@128];

那个新的getter方法调用你懒惰写的代码，像这样......

- (id)dataProductionAtIndex:(NSNumber *)index {

    // replace dataProductionArray with dataProductionDictionary
    id result = self.dataProductionDictionary[index];
    if (!result) {
        result = [self getDataAt:index];
        self.dataProductionDictionary[index] = result;
    }
    return result;
}

然后getDataAt: 是对您发布的代码的简单重构，除了不是循环 32k 元素，它只为传入的一个索引完成工作......

- (id)getDataAt:(NSNumber *)index {

    int j = [index intValue];

    // no loop, just skip to iteration j
    NSArray *aItems   = [_dictionaries[j] [@"prodA"] componentsSeparatedByString:@","];
    NSArray *bItems   = [_dictionaries[j] [@"prodB"] componentsSeparatedByString:@","];

    // and so on, then at the end, don't save arrayItem, just return it

    NSDictionary *arrayItem = @{@"name":_dictionaries[j] [@"name"],
                                @"type":[NSString stringWithFormat:@"%@",_dictionaries[j] [@"type"]],
                                @"height":[NSString stringWithFormat:@"%@",_dictionaries[j] [@"height"]],
                                @"aArray":productionAArray,
                                @"bArray":productionBArray,
                                };
    return arrayItem;
}    

PS - 可变字典是用于惰性求值的良好数据结构。下一个复杂级别是 NSCache，它的作用类似于可变字典并管理内存 (class ref here)。

Answer 2

你的 for 循环很愚蠢。随便写

for (NSDictionary* dict in _dictionaries)...

并使用 dict 代替 _dictionaries [j]。每次保存一个方法调用。

stringWithFormat: 每次都会创建一个新字符串。你不能只添加项目本身而不是把它变成一个字符串吗？

与其将所有项目提取到productionAArray 和productionBArray 中，不如创建一个NSIndexSet，将其填充到循环中——或者更好的是使用块——然后一次性创建数组。