[NPWP笔记]自动生成子进程服务器

接上文《使用多进程处理客户请求》中描述，由于它存在严重的问题---超过当前的服务线程时，程序将不再进行处理。本文中结合之前的fork技术对其进行完善，主要的修改内容是如何新增子进程和如何删除子进程。新增子进程是在当前的所有子进程都在处理客户的请求时又有新的子进程的请求到达；删除子进程是在空闲的子进程过多时，为了释放系统资源。这两个值成为服务阈值（下阈值和上阈值）。

父进程如何能够得知当前子进程的个数及其状态呢？父子进程将采用管道进行通信。

#............ use IO::Handle; use IO::File; use strict; use warnings; use IO::Socket; use IO::Select; use File::Basename; use Carp 'croak','cluck'; use Fcntl ':flock'; use POSIX qw(:signal_h WNOHANG); use constant NUM => 1; use constant PIDPATH => '/tmp'; use constant PIDFILE =>'/tmp/prefork.pid'; use constant MAX_QUEST => 2; use constant LO_WATER_MARK => 2; use constant HI_WATER_MARK => 3; my $host = "localhost"; my $port = 27002; my %CHILDREN; my %STATUS; my $DONE = 0; my $child_count = 0; my ($pid,$pidfile,$saved_dir,$CWD); pipe(CHILD_READ,CHILD_WRITE) or die "can not make pipe!/n"; my $IN = IO::Select->new(/*CHILD_READ); $SIG{INT} = $SIG{TERM} = sub{$DONE++}; #$SIG{CHLD} = /&reap_child; my $socket = IO::Socket::INET->new(LocalPort => $port, Listen => SOMAXCONN, Reuse => 1, ) or die "cann't create listen socket:$!"; make_new_child() for(1..NUM); #init_server(PIDFILE); while(!$DONE){ if($IN->can_read){ my $message; next unless sysread(CHILD_READ,$message,4096); my @messages = split "/n",$message; foreach(@messages){ next unless my ($pid,$status) = /^(/d+) (.+)$/; if($status ne 'done'){ $STATUS{$pid} = $status; } else{ delete $STATUS{$pid}; } } } my @idle = sort {$a <=> $b} grep {$STATUS{$_} eq 'idle'} keys %STATUS; if(@idle < LO_WATER_MARK){ make_new_child() for (0..LO_WATER_MARK - @idle - 1); } elsif (@idle > HI_WATER_MARK){ my @goners = @idle[0..@idle - HI_WATER_MARK - 1]; my $killed = kill HUP => @goners; warn "killed $killed children/n"; } print "fork a child!/n"; } #initialize server #launch a child process sub launch_child { my $callback = shift; my $signals = POSIX::SigSet->new(SIGINT,SIGCHLD,SIGTERM,SIGHUP); sigprocmask(SIG_BLOCK,$signals); # block inconvenient signals log_die("Can't fork: $!") unless defined (my $child = fork()); if ($child) { $CHILDREN{$child} = $callback || 1; } else { $SIG{HUP} = $SIG{INT} = $SIG{CHLD} = $SIG{TERM} = 'DEFAULT'; } sigprocmask(SIG_UNBLOCK,$signals); # unblock signals return $child; } #reap child sub reap_child { while ( (my $child = waitpid(-1,WNOHANG)) > 0) { $CHILDREN{$child}->($child) if ref $CHILDREN{$child} eq 'CODE'; delete $CHILDREN{$child}; } } #create a child process sub make_new_child{ my $child = launch_child(/&clean_child); if($child){ $child_count++; } else{ close CHILD_READ; do_child($socket); exit 0; } } sub clean_child{ my $child = shift; delete $STATUS{$child}; } #process child request sub do_child{ my $socket = shift; my $lock = IO::File->new(PIDFILE,O_RDONLY) or die "can not open lock file :$!"; my $cycles = MAX_QUEST; my $done = 0; $SIG{HUP} = sub{$done ++}; while(!$done && $cycles--){ syswrite CHILD_WRITE,"$$ idle/n"; my $c; next unless eval{ local $SIG{HUP} = sub{$done++;die}; flock($lock,LOCK_EX); last unless $c = $socket->accept; flock($lock,LOCK_UN); }; syswrite CHILD_WRITE,"$$ busy/n"; while(<$c>){ print; } close $c; } warn "child $$ done/n"; syswrite CHILD_WRITE,"$$ done/n"; close $socket; close $lock; close /*CHILD_WRITE; } close $socket;

测试程序不变，测试时可以通过多个客户端进行连接。

#!/usr/bin/perl #read from STDIN and send string to server #client request until you terminate use strict; use IO::Socket; use warnings; my $host = "192.168.1.105"; my $port = 27002; $SIG{INT} = sub{exit}; my $sock = IO::Socket::INET->new("$host:$port"); while(defined(my $msg = STDIN->getline)){ print $sock $msg; } $sock->close;