perl进程管理一例

#!/usr/bin/perl -w
use strict;
use warnings;
use DBI;

####
#   这里进行服务器任务管理
##
#字符串映射函数
our %actions = (
    "check" => &pcheck,
    "run" => &prun,
    "run_nohup" => &run_nohup,
    "kill" => &pkill
);
our $home = "/var/www/html";
our $public_home = "/var/www/html/public" ;
our $log_home = "/var/www/html/yfcloud_admin/Public/log" ;

our $host = "XXX" ; 
our $database = "XXX" ;
our $db_user = "XXX";
our $db_pass = "XXX";

#程序路径
our $ppath = "" ; 
#程序名
our $pname = "" ;
#日志文件
our $log_file = "log" ; 

#检测进程状态，返回PID
# 如果同进程不在运行，则返回空
sub pcheck
{
    # grep 排除perl本身的进程
    my $cmd = "ps -ef | grep " . $pname . " | grep -v grep | grep -v perl | awk '{print $2}'";
    my $ret = `$cmd` ;
    chomp $ret;
    #如果有多个进程PID，则返回第一个pid
    if( $ret =~ /
/ ){
        $ret = ( split /
/, $ret )[0] ; 
    }
    ($ret eq "") ? print 0 : print $ret ;
}

#结束进程，返回成功与否
# 0 表示成功
sub pkill
{
    my $cmd = "killall $pname";
    my $ret = `$cmd`;
    print $? ; 
}

# 参数需要是 程序的全路径 
# 启动进程并后台运行，并返回PID
# 启动进程后，需要将进程的PID及日志文件写入数据库
# shell输出格式如下： [1] 15758
sub prun
{
    my $cmd = "$pname";
    run_exec($cmd);
}

# 后台以nuhup方式运行
sub run_nohup
{
    my $cmd = "nohup $pname" ;
    run_exec($cmd);
}

# 命令格式如下： perl ProcessManager.pl $pname $action 
sub main
{
    umask(0);
    #print "begin
";
    #这里接收参数并执行相应的操作
    if( @ARGV < 2 )
    {
        print "arguments error!
usage:perl ProcessManager.pl command action
";
        exit 1;
    }
    $pname = $ARGV[0] ;  #这里使用全局变量算了 ~_~
    my $action = $ARGV[1] ;
    #判断操作是否合法
    if( $actions{$action} ){
        $actions{$action}->() ;
    }else{
        print "undeclare action!check you action!
";
        exit 1;
    }
}

main();



############
##这里是一些工具函数
############
sub gettime
{
   my($sec,$min,$hour,$day,$mon,$year) = (localtime(time));
   $mon +=1 ;
   $day = ($day < 10 )?"0$day":$day;
   $mon = ($mon < 10 )?"0$mon":$mon;
   $min = ($min < 10 )?"0$min":$min;

   return "$mon$day$hour$min" ; 
}

sub run_exec
{
    my $cmd = shift ;
    #先检测文件是否存在，可执行
    if( -x $pname ){
        $pname =~ m#.+/(.+)# ; #获得程序名（包括后缀名）作为日志文件名称
        # 建立程序的日志文件目录
        my $name = $1 ;
        mkdir "$log_home/$name" or die("cannot create dir:$name") unless ( -e "$log_home/$name" );
        $log_file = "$log_home/$name/".gettime() ;
        # print "log_file : $log_file
";
        $cmd .= " > $log_file 2>&1 &";
        addToDB($name,$log_file);
        my $pid = fork;
        if (not $pid) {
            $pid = $$; 
            print $pid ; 
            exec($cmd); #exec是异步的，执行完后就脱离perl程序了
        }
    }else{
        print "$pname cannot be excuted
";
        exit 1;
    }
}

# 将日志写入日志
sub addToDB
{
    my ($progName,$log_file) = @_ ;
    my $createtime = time ;
    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";
    
    my $sql = "insert into log (name,log_file,createtime) values (?,?,?)"; 
    my $sth = $db_handle->prepare($sql);
    
    $sth->execute($progName, $log_file , $createtime) or die $DBI::errstr;
    $sth->finish();
    
    $db_handle->disconnect();
}