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 "begin\n";
 78     #这里接收参数并执行相应的操作
 79     if( @ARGV < 2 )
 80     {
 81         print "arguments error!\nusage:perl ProcessManager.pl command action\n";
 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_file\n";
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 excuted\n";
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 }