perl如何获取一个目录中所有文件的大小总和

MMMIX

原帖由 smzml 于 2006-3-19 16:34 发表
谢谢5楼大哥。但是这样只能统计某个目录下的文件，如果有子目录和子文件，它数不出来呀。
而且，如果文件有几千个，统计效率好象很慢，有没有快一点的办法呀？

不管你用啥方法，其实都是递归遍历，只不过有些实现的效率比较高，有些比较低而已。

撒哈拉里的鱼

我前几天恰好写过一个类似的东西，http://www.icylife.net/yunshu/show.php?id=647。

1. 
   
2. #!/usr/bin/perl
   
3. use strict;
   
4. use warnings;
   
5. use threads;
   
6. use threads::shared;
   
7. use Thread::Semaphore;
   
8. 
   
9. my ($size, $dircnt, $filecnt) = (0, 0, 0);
   
10. my %size_hash : shared;
    
11. my @sub_dirs;
    
12. 
    
13. if( @ARGV != 3 )
    
14. {
    
15.         print "usage: $0  <rootdir>   <depth>   <topn> \n";
    
16.         print "example:$0   c:\\    3   15\n";
    
17. 
    
18.         exit;
    
19. }
    
20. my $root = $ARGV[0];
    
21. my $sub = $ARGV[1];
    
22. my $topn = $ARGV[2];
    
23. 
    
24. if( $root !~ /\\$/ && $root !~ /\/$/ )
    
25. {
    
26.         $root = $root."\\";
    
27. }
    
28. 
    
29. print "开始搜寻$root下面的$sub级子目录.....\n";
    
30. &GetSubDirectory( );
    
31. 
    
32. print "开始获取目录大小......\n";
    
33. my $max_thread_count = 100;
    
34. my $semaphore = new Thread::Semaphore( $max_thread_count );
    
35. foreach my $tmp_dir( @sub_dirs )
    
36. {
    
37.         $semaphore->down( );
    
38.         my $thread = threads->create( \&GetDirectorySize, $tmp_dir );
    
39.         $thread->detach();
    
40. }
    
41. 
    
42. &Wait2Quit( );
    
43. 
    
44. my $index = 0;
    
45. foreach my $key ( sort { $size_hash{$b} <=> $size_hash{$a} } keys %size_hash)
    
46. {
    
47.         if( $index == $topn )
    
48.         {
    
49.                 last;
    
50.         }
    
51.         printf( "%-50s%18s 字节\n", $key, $size_hash{$key} );
    
52.         $index ++;
    
53. }
    
54. 
    
55. sub Wait2Quit
    
56. {
    
57.         my $num = 0;
    
58.         while( $num < $max_thread_count )
    
59.         {
    
60.                 # 尝试获取信号量，当能够获取到最大线程数个信号量时，表示所有线程都结束了
    
61.                 $semaphore->down( );
    
62.                 $num ++;
    
63.         }
    
64.         #print "All $max_thread_count thread quit\n";
    
65. }
    
66. 
    
67. sub GetSubDirectory
    
68. {
    
69.         my @dirs;
    
70.         push( @dirs, $root );
    
71. 
    
72.         my ( $current_dir, $file );
    
73.         while ( $current_dir = pop(@dirs) )
    
74.         {
    
75.                 # 根据路径中\或者/的次数，确定当前是几级目录
    
76.                 #print "processing $current_dir:";
    
77.                 my $sub_count = 0;
    
78.                 while( $current_dir=~/\\|\//g )
    
79.                 {
    
80.                         $sub_count++;
    
81.                 }
    
82.                 #print " $sub_count\n";
    
83.                 # 等于指定深度的，加入到最终数组中
    
84.                 if( $sub_count == $sub )
    
85.                 {
    
86.                         #print $current_dir."\n";
    
87.                         push( @sub_dirs, $current_dir );
    
88.                         next;
    
89.                 }
    
90.                 # 过深的目录不予处理
    
91.                 elsif( $sub_count > $sub )
    
92.                 {
    
93.                         next;
    
94.                 }
    
95. 
    
96.                 # 目录结尾加上\
    
97.                 if( $current_dir !~ /\\$/ && $current_dir !~ /\/$/ )
    
98.                 {
    
99.                         $current_dir = $current_dir."\\";
    
100.                 }
     
101. 
     
102.                 # 路径不足的打开目录，查找子目录
     
103.                 local *DH;
     
104.                 if ( !opendir(DH, $current_dir) )
     
105.                 {
     
106.                         #print "Cannot opendir $current_dir: $!\n";
     
107.                         next;
     
108.                 }
     
109. 
     
110.                 # 读取文件和子目录列表
     
111.                 foreach( readdir(DH) )
     
112.                 {
     
113.                         # 当前目录和上级目录跳过
     
114.                         if ($_ eq "." || $_ eq "..")
     
115.                         {
     
116.                                 next;
     
117.                         }
     
118.                         $file = $current_dir.$_;
     
119.                         if ( !-l $file && -d $file )
     
120.                         {
     
121.                                 #print "Add $file\n";
     
122.                                 push(@dirs, $file);
     
123.                         }
     
124.                         #&process($file);
     
125.                 }
     
126.                 closedir(DH);
     
127.         }
     
128. }
     
129. 
     
130. sub GetDirectorySize
     
131. {
     
132.         my $first_dir = shift;
     
133.         $size_hash{ $first_dir } = 0;
     
134. 
     
135.         my @dirs;
     
136.         push( @dirs, $first_dir );
     
137. 
     
138.         my ( $current_dir, $file );
     
139.         while ( $current_dir = pop(@dirs) )
     
140.         {
     
141.                 # 目录结尾加上\
     
142.                 if( $current_dir !~ /\\$/ && $current_dir !~ /\/$/ )
     
143.                 {
     
144.                         $current_dir = $current_dir."\\";
     
145.                 }
     
146. 
     
147.                 # 打开目录，查找子目录以及文件
     
148.                 local *DH;
     
149.                 if ( !opendir(DH, $current_dir) )
     
150.                 {
     
151.                         #print "Cannot opendir $current_dir: $!\n";
     
152.                         next;
     
153.                 }
     
154. 
     
155.                 # 读取文件和子目录列表
     
156.                 foreach( readdir(DH) )
     
157.                 {
     
158.                         # 当前目录和上级目录跳过
     
159.                         if ($_ eq "." || $_ eq "..")
     
160.                         {
     
161.                                 next;
     
162.                         }
     
163.                        
     
164.                         # 形成绝对路径
     
165.                         $file = $current_dir.$_;
     
166. 
     
167.                         # 处理目录或符号链接
     
168.                         if ( !-l $file && -d $file )
     
169.                         {
     
170.                                 push(@dirs, $file);
     
171.                         }
     
172.                        
     
173.                         $size_hash{ $first_dir } += -s $file;
     
174.                 }
     
175.                 closedir(DH);
     
176.         }
     
177. 
     
178.         $semaphore->up( );
     
179. }
     

复制代码

flw

你为啥不用 File::Find？

MMMIX

原帖由 flw 于 2008-12-12 11:08 发表
你为啥不用 File::Find？

遍历目录用这个很方便

撒哈拉里的鱼

因为对比过，File::Find没有我自己写的速度快。

huifeideluotuo

原帖由 flw 于 2008-12-12 11:08 发表
你为啥不用 File::Find？

遍历目录的话，File_Find 模块的确不方便，正常情况下，用stat函数获得每个文件的大小，但是在累加求和的时候，发现每次都要去调用wanted子过程，不是多方便。