iOS 快速计算目录的size

关于目录的size计算，我在网上查了很多种方式去实现过，最常见的就是通过递归方式去逐层计算，但通过测试发现在计算层数多、数量大的目录时，递归的方式会消耗很大的栈空间，甚至出现栈溢出，当然效率也会是很大的瓶颈。最初通过将NSFileManager获取单个文件size修改为FS的函数提升效率，到后来看到老谭 改用效率更高的stat方式，并且不仅可能获得文件实际大小，也能获得占用磁盘的大小。另外通过实现栈的方式去替换掉递归，不仅节省了内存占用，让效率也有大幅的提升。

下面就是具体的代码实现，通过NSMutableArray来模拟栈（经测试用数组头作为栈顶相比数组尾部做栈顶效率更高，这与NSMutableArray内部实现有关，也尝试不依赖NSMutableArray而去实现栈但效率提升有限），并通过diskMode参数可返回是否是磁盘占用的size：

+ (uint64_t)sizeAtPath:(NSString*)filePath diskMode:(BOOL)diskMode

{

uint64_ttotalSize =0;

NSMutableArray*searchPaths = [NSMutableArrayarrayWithArray:NSSearchPathForDirectoriesInDomains(NSLibraryDirectory,NSUserDomainMask,YES)] ;

[searchPathsaddObjectsFromArray:NSSearchPathForDirectoriesInDomains(NSDocumentDirectory,NSUserDomainMask,YES)];

[searchPathsaddObject:NSTemporaryDirectory()];

while([searchPathscount] >0)

{

@autoreleasepool

{

NSString*fullPath = [searchPathsobjectAtIndex:0];

[searchPathsremoveObjectAtIndex:0];

structstatfileStat;

if(lstat([fullPathfileSystemRepresentation], &fileStat) ==0)

{

if(fileStat.st_mode&S_IFDIR)

{

NSArray*childSubPaths = [[NSFileManagerdefaultManager]contentsOfDirectoryAtPath:fullPatherror:nil];

for(NSString*childIteminchildSubPaths)

{

NSString*childPath = [fullPathstringByAppendingPathComponent:childItem];

[searchPathsinsertObject:childPathatIndex:0];

}

}else

{

if(diskMode)

totalSize += fileStat.st_blocks*512;

else

totalSize += fileStat.st_size;

}

}

}

}

returntotalSize;

}