关于目录的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;
}