perl下多进程的进度显示

hp_truth

经常遇到这样的需求, 一个父进程开启多个子进程, 每个子进程都需要做一个长时间的处理, 所以希望显示这些子进程的进度. 在网上可以找到单进程的进度显示, 但多进程的进度显示没找着.  
所以就自己想着做一个, 希望大家给点意见,要是有更好的方法的话请分享一下.

1. 
   
2. 
   
3. #!/usr/local/bin/perl5
   
4. 
   
5. use warnings;
   
6. use strict;
   
7. use Fcntl;
   
8. use Term::Cap;
   
9. use Parallel::ForkManager;
   
10. 
    
11. #==============================
    
12. # for terminal control
    
13. #==============================
    
14. my $termios = new POSIX::Termios;
    
15. my $terminal = Term::Cap->Tgetent ({ TERM => undef, OSPEED => $termios->getospeed });
    
16. $terminal->Trequire(qw(cm sc rc)); #check if terminal support these capabilities
    
17. my $c = `clear`;
    
18. print $c;
    
19. 
    
20. #==============================
    
21. # for fifo
    
22. #==============================
    
23. my $fpath = "myfifo";
    
24. unless (-p $fpath) {
    
25.     if (-e _) {
    
26.         die "$fpath is something unknow\n";
    
27.     }
    
28.     else {
    
29.         require POSIX;
    
30.         POSIX::mkfifo($fpath, 0666) or die "can not mknod $fpath: $!";
    
31.         print "$0: created $fpath as a named pipe\n";
    
32.     }
    
33. }
    
34. else {
    
35.     print "$0: named pipe $fpath exists\n";
    
36. }
    
37. 
    
38. 
    
39. #==============================
    
40. # for parallel processes
    
41. #==============================
    
42. my @names;
    
43. push @names, "p$_" foreach 0..5;
    
44. my $childs;
    
45. $childs->{$_} = {total=>1, processed=>0, name => $names[$_]} foreach(0.. $#names);
    
46. my $remain_child_process = scalar @names;
    
47. 
    
48. $SIG{PIPE} = sub {
    
49.     -- $remain_child_process;
    
50.     print STDERR "pipe broken\n";
    
51. };
    
52. 
    
53. 
    
54. #==============================
    
55. # main program
    
56. #==============================
    
57. my $now = time;
    
58. my $pm =  new Parallel::ForkManager(scalar @names);
    
59. foreach my $child ( 0 .. $#names ) {
    
60.     my $pid = $pm->start($names[$child]) and next;
    
61.     print "This is $names[$child], Child number $child: $\n";
    
62.     work($child);
    
63.     $pm->finish($child); # pass an exit code to finish
    
64. }
    
65. print "Waiting for Children...\n";
    
66. show_progress();
    
67. $pm->wait_all_children;
    
68. $now = time - $now;
    
69. printf("\n\nTotal run time: %02d:%02d:%02d\n\n", int($now / 3600), int(($now % 3600) / 60), int($now % 60));
    
70. print "Everybody done!\n";
    
71. 
    
72. 
    
73. #==============================
    
74. # real work of child process
    
75. #==============================
    
76. sub work {
    
77.     my $child = shift;
    
78. 
    
79.     my $total = int(rand(20)) + 5;
    
80.     foreach (1..$total) {
    
81.         select(undef, undef, undef, 0.2);
    
82.         sysopen (FIFO_W, $fpath, O_WRONLY) or die "can't write $fpath: $!";
    
83.         print FIFO_W "child:$child,processed:$_,total:$total\n";
    
84.         close FIFO_W;
    
85.     }
    
86. 
    
87.     sysopen (FIFO_W, $fpath, O_WRONLY) or die "can't write $fpath: $!";
    
88.     print FIFO_W "$child over\n";
    
89.     close FIFO_W;
    
90. }
    
91. 
    
92. 
    
93. #==================================================
    
94. # receive progress data from child and display them
    
95. #==================================================
    
96. sub show_progress {
    
97.     while ($remain_child_process) {
    
98.         die "Pipe file disappeared" unless -p $fpath;
    
99.         sysopen (FIFO_R, $fpath, O_RDONLY) or die "can't read $fpath: $!";
    
100.         while(my $m = <FIFO_R>) {
     
101.             if ($m =~ m/over/) {
     
102.                 -- $remain_child_process;
     
103.             }
     
104.             else {
     
105.                 #print $m;
     
106.                 my ($child, $processed, $total) = $m =~ m/child:(\w+),processed:(\d+),total:(\d+)/;
     
107.                 $childs->{$child}->{processed} = $processed;
     
108.                 $childs->{$child}->{total} = $total;
     
109.                 proc_bar($childs);
     
110.             }
     
111.         }
     
112.         close FIFO_R;
     
113.     }
     
114.     $terminal->Tgoto('cm', 0, 5+2*@names, *STDOUT); #move cursor to bottom of screen
     
115. }
     
116. 
     
117. #=====================================
     
118. # indicate the progress of comparation
     
119. #=====================================
     
120. sub proc_bar{
     
121.     my $childs = shift;
     
122.     local $| = 1;
     
123. 
     
124.     $terminal->Tgoto('sc', undef, undef, *STDOUT); #save cursor
     
125.     for my $c (sort keys %$childs) {
     
126.         my $i = $childs->{$c}->{processed};
     
127.         my $n = $childs->{$c}->{total};
     
128.         my $name = $childs->{$c}->{name};
     
129.         print "\r\033[36mchild: $c($name) [\033[33m".("#" x int(($i/$n)*50)).(" " x (50 - int(($i/$n)*50)))."\033[36m]";
     
130.         printf("%2.1f%%\033[0m\n",$i/$n*100);
     
131.     }
     
132.     $terminal->Tgoto('rc', undef, undef, *STDOUT); #restore cursor
     
133. 
     
134.     local $| = 0;
     
135. }
     
136. 
     

复制代码

x9x9

支持一下，慢慢看~

skskevin

慢慢学习