perl进程管理一例

 1 #!/usr/bin/perl -w
 2 use strict;  3 use warnings;  4 use DBI;  5 
 6 ####  7 # 这里进行服务器任务管理  8 ##  9 #字符串映射函数
 10 our %actions = (  11     "check" => &;pcheck, 12     "run" => &;prun, 13     "run_nohup" => &;run_nohup, 14     "kill" => &;pkill  15 );  16 our $home = "/var/www/html";  17 our $public_home = "/var/www/html/public" ;  18 our $log_home = "/var/www/html/yfcloud_admin/Public/log" ;  19 
 20 our $host = "XXX" ;  21 our $database = "XXX" ;  22 our $db_user = "XXX";  23 our $db_pass = "XXX";  24 
 25 #程序路径
 26 our $ppath = "" ;  27 #程序名
 28 our $pname = "" ;  29 #日志文件
 30 our $log_file = "log" ;  31 
 32 #检测进程状态，返回PID  33 # 如果同进程不在运行，则返回空
 34 sub pcheck  35 {  36     # grep 排除perl本身的进程
 37     my $cmd = "ps -ef | grep " . $pname . " | grep -v grep | grep -v perl | awk '{print $2}'";  38     my $ret = `$cmd` ;  39     chomp $ret;  40     #如果有多个进程PID，则返回第一个pid
 41     if( $ret =~ /n/ ){  42         $ret = ( split /n/,$ret )[0] ;  43  }  44     ($ret eq "") ? print 0 : print $ret ;  45 }  46 
 47 #结束进程，返回成功与否  48 # 0 表示成功
 49 sub pkill  50 {  51     my $cmd = "killall $pname";  52     my $ret = `$cmd`;  53     print $? ;  54 }  55 
 56 # 参数需要是 程序的全路径  57 # 启动进程并后台运行，并返回PID  58 # 启动进程后，需要将进程的PID及日志文件写入数据库  59 # shell输出格式如下： [1] 15758
 60 sub prun  61 {  62     my $cmd = "$pname";  63     run_exec($cmd);  64 }  65 
 66 # 后台以nuhup方式运行
 67 sub run_nohup  68 {  69     my $cmd = "nohup $pname" ;  70     run_exec($cmd);  71 }  72 
 73 # 命令格式如下： perl ProcessManager.pl $pname $action 
 74 sub main  75 {  76     umask(0);  77     #print "beginn";  78  #这里接收参数并执行相应的操作
 79     if( @ARGV < 2 )  80  {  81         print "arguments error!nusage:perl ProcessManager.pl command actionn";  82         exit 1;  83  }  84     $pname = $ARGV[0] ;  #这里使用全局变量算了 ~_~
 85     my $action = $ARGV[1] ;  86     #判断操作是否合法
 87     if( $actions{$action} ){  88         $actions{$action}->() ;  89     }else{  90         print "undeclare action!check you action!n";  91         exit 1;  92  }  93 }  94 
 95 main();  96 
 97 
 98 
 99 ############ 100 ##这里是一些工具函数 101 ############
102 sub gettime 103 { 104    my($sec,$min,$hour,$day,$mon,$year) = (localtime(time)); 105    $mon +=1 ; 106    $day = ($day < 10 )?"0$day":$day; 107    $mon = ($mon < 10 )?"0$mon":$mon; 108    $min = ($min < 10 )?"0$min":$min; 109 
110    return "$mon$day$hour$min" ; 111 } 112 
113 sub run_exec 114 { 115     my $cmd = shift ; 116     #先检测文件是否存在，可执行
117     if( -x $pname ){ 118         $pname =~ m#.+/(.+)# ; #获得程序名（包括后缀名）作为日志文件名称 119  # 建立程序的日志文件目录
120         my $name = $1 ; 121         mkdir "$log_home/$name" or die("cannot create dir:$name") unless ( -e "$log_home/$name" ); 122         $log_file = "$log_home/$name/".gettime() ; 123         # print "log_file : $log_filen";
124         $cmd .= " > $log_file 2>&1 &"; 125         addToDB($name,$log_file); 126         my $pid = fork; 127         if (not $pid) { 128             $pid = $$; 129             print $pid ; 130             exec($cmd); #exec是异步的，执行完后就脱离perl程序了
131  } 132     }else{ 133         print "$pname cannot be excutedn"; 134         exit 1; 135  } 136 } 137 
138 # 将日志写入日志
139 sub addToDB 140 { 141     my ($progName,$log_file) = @_ ; 142     my $createtime = time ; 143     my $db_handle = DBI->connect("DBI:mysql:database=$database;host=$host",$db_user,$db_pass,{'RaiseError' => 1})|| die "Could not connect to database: $DBI::errstr"; 144     
145     my $sql = "insert into log (name,log_file,createtime) values (?,?,?)"; 146     my $sth = $db_handle->prepare($sql); 147     
148     $sth->execute($progName,$log_file,$createtime) or die $DBI::errstr; 149     $sth->finish(); 150     
151     $db_handle->disconnect(); 152 }