iOS中使用c函数快速获取一个目录所占的空间大小

iOS中使用c函数快速获取一个目录所占的空间大小

Posted on 2011-12-31 by york_gu

在编写iOS应用时，我们都会大量把服务器的接口调用数据或者图片数据进行缓存，为了更好的使用体验，有时候我们也会把各种缓存的数据所占的空间大小告诉用户，提供给用户参考，并让用户可以手动清除缓存数据，就像我们在千寻影视中 也就提供了这样一个设置项。但是当数据多了之后，就会发现所占空间的大小计算非常耗时，因为缓存下来的是大量的小文件，这就会造成界面很久没有反映的状 况。为了解决这种情况，当然是可以通过多线程的方法来做到的，但是在一个简单的设置界面做这么复杂的事情，我倒认为是一个拿起牛刀来杀鸡的行为了。本篇就 是来探求一些优化手段，把目录的大小计算耗时降到可以接受的程度。

首先，在iOS当中，要实现一个功能，肯定想到去找objective-c的官方库中是否提供了相关功能，如果有，那就直接用objective-c代码来实现了，通过搜索，很容易就写出这样一个函数：

+ ( long long ) fileSizeAtPath:( NSString *) filePath{
  NSFileManager * manager = [ NSFileManager defaultManager];
  if ([manager fileExistsAtPath:filePath]){
    return [[manager attributesOfItemAtPath:filePath error: nil ] fileSize];
  }
  return 0;
}

简单可行，但是，我们这里探讨性能，要优化，很容易想到直接使用c代码，跳过objective-c的封装来做，于是，我们这个函数就变成了：

+ ( long long ) fileSizeAtPath:( NSString *) filePath{
  struct stat st;
  if (lstat([filePath cStringUsingEncoding: NSUTF8StringEncoding ], &st) == 0){
    return st.st_size;
  }
  return 0;
}

循环1000次，我们就可以发现两者之间巨大的性能差距了，在我的测试环境中，结果如下，c函数的耗时仅是第一种方法的5%

2011-12-31 14:17:22.165 otest[85861:903] [<span style= "color: #5d68fc;" >0.041</span>] 使用 NSFileManager 获取文件大小
2011-12-31 14:17:22.168 otest[85861:903] [<span style= "color: #5e6cfc;" >0.002</span>] 使用unix c函数获取文件大小

要获取一个目录的总大小，我们就可以写这么一个函数了：

// 循环调用fileSizeAtPath来获取一个目录所占空间大小
+ ( long long ) folderSizeAtPath2:( NSString *) folderPath{
  NSFileManager * manager = [ NSFileManager defaultManager];
  if (![manager fileExistsAtPath:folderPath]) return 0;
  NSEnumerator *childFilesEnumerator = [[manager subpathsAtPath:folderPath] objectEnumerator];
  NSString * fileName;
  long long folderSize = 0;
  while ((fileName = [childFilesEnumerator nextObject]) != nil ){
    NSString * fileAbsolutePath = [folderPath stringByAppendingPathComponent:fileName];
    folderSize += [ self fileSizeAtPath:fileAbsolutePath];
  }
  return folderSize;
}

但是，通过测试，目录遍历本身所带来的时间消耗大大的弱化了我们改进过的fileSizeAtPath函数所带来的优势，导致后者的测试耗时仅是前者的1/2到1/3之间：

2011-12-31 14:17:23.233 otest[85861:903] [<span style= "color: #4958fc;" >0.932</span>] 使用subpathsAtPath然后循环调用 NSFileManager 获取目录总大小
2011-12-31 14:17:23.582 otest[85861:903] [<span style= "color: #4958fc;" >0.348</span>] 使用subpathsAtPath然后循环调用unix c函数获取目录总大小

既然文件大小的获取可以使用c函数，那么目录的遍历当然可以使用c函数来进行的了，一不做二不休，继续修改优化，代码我就不在继续贴了，附件中有包含了自动测试的整个项目文件，优化后，最终的效果如下：

1 2 3

2011-12-31 14:29:08.520 otest[87180:903] [<span style= "color: #2c2cfb;" >0.915</span>] 使用subpathsAtPath然后循环调用 NSFileManager 获取目录总大小 2011-12-31 14:29:08.860 otest[87180:903] [<span style= "color: #2c2cfb;" >0.339</span>] 使用subpathsAtPath然后循环调用unix c函数获取目录总大小 2011-12-31 14:29:08.953 otest[87180:903] [<span style= "color: #2c2cfb;" >0.092</