perl进程管理一例

yzqzs

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 }