forked from walkor/workerman
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Worker.php
1389 lines (1281 loc) · 44.6 KB
/
Worker.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
<?php
/**
* This file is part of workerman.
*
* Licensed under The MIT License
* For full copyright and license information, please see the MIT-LICENSE.txt
* Redistributions of files must retain the above copyright notice.
*
* @author walkor<[email protected]>
* @copyright walkor<[email protected]>
* @link http://www.workerman.net/
* @license http://www.opensource.org/licenses/mit-license.php MIT License
*/
namespace Workerman;
use \Workerman\Events\Libevent;
use \Workerman\Events\Select;
use \Workerman\Events\EventInterface;
use \Workerman\Connection\ConnectionInterface;
use \Workerman\Connection\TcpConnection;
use \Workerman\Connection\UdpConnection;
use \Workerman\Lib\Timer;
use \Exception;
/**
* Worker 类
* 是一个容器,用于监听端口,维持客户端连接
*/
class Worker
{
/**
* 版本号
* @var string
*/
const VERSION = '3.1.8';
/**
* 状态 启动中
* @var int
*/
const STATUS_STARTING = 1;
/**
* 状态 运行中
* @var int
*/
const STATUS_RUNNING = 2;
/**
* 状态 停止
* @var int
*/
const STATUS_SHUTDOWN = 4;
/**
* 状态 平滑重启中
* @var int
*/
const STATUS_RELOADING = 8;
/**
* 给子进程发送重启命令 KILL_WORKER_TIMER_TIME 秒后
* 如果对应进程仍然未重启则强行杀死
* @var int
*/
const KILL_WORKER_TIMER_TIME = 1;
/**
* 默认的backlog,即内核中用于存放未被进程认领(accept)的连接队列长度
* @var int
*/
const DEFAUL_BACKLOG = 1024;
/**
* udp最大包长
* @var int
*/
const MAX_UDP_PACKEG_SIZE = 65535;
/**
* worker的名称,用于在运行status命令时标记进程
* @var string
*/
public $name = 'none';
/**
* 设置当前worker实例的进程数
* @var int
*/
public $count = 1;
/**
* 设置当前worker进程的运行用户,启动时需要root超级权限
* @var string
*/
public $user = '';
/**
* 当前worker进程是否可以平滑重启
* @var bool
*/
public $reloadable = true;
/**
* 当worker进程启动时,如果设置了$onWorkerStart回调函数,则运行
* 此钩子函数一般用于进程启动后初始化工作
* @var callback
*/
public $onWorkerStart = null;
/**
* 当有客户端连接时,如果设置了$onConnect回调函数,则运行
* @var callback
*/
public $onConnect = null;
/**
* 当客户端连接上发来数据时,如果设置了$onMessage回调,则运行
* @var callback
*/
public $onMessage = null;
/**
* 当客户端的连接关闭时,如果设置了$onClose回调,则运行
* @var callback
*/
public $onClose = null;
/**
* 当客户端的连接发生错误时,如果设置了$onError回调,则运行
* 错误一般为客户端断开连接导致数据发送失败、服务端的发送缓冲区满导致发送失败等
* 具体错误码及错误详情会以参数的形式传递给回调,参见手册
* @var callback
*/
public $onError = null;
/**
* 当连接的发送缓冲区满时,如果设置了$onBufferFull回调,则执行
* @var callback
*/
public $onBufferFull = null;
/**
* 当链接的发送缓冲区被清空时,如果设置了$onBufferDrain回调,则执行
* @var callback
*/
public $onBufferDrain = null;
/**
* 当前进程退出时(由于平滑重启或者服务停止导致),如果设置了此回调,则运行
* @var callback
*/
public $onWorkerStop = null;
/**
* 传输层协议
* @var string
*/
public $transport = 'tcp';
/**
* 所有的客户端连接
* @var array
*/
public $connections = array();
/**
* 应用层协议,由初始化worker时指定
* 例如 new worker('http://0.0.0.0:8080');指定使用http协议
* @var string
*/
protected $_protocol = '';
/**
* 当前worker实例初始化目录位置,用于设置应用自动加载的根目录
* @var string
*/
protected $_appInitPath = '';
/**
* 是否以守护进程的方式运行。运行start时加上-d参数会自动以守护进程方式运行
* 例如 php start.php start -d
* @var bool
*/
public static $daemonize = false;
/**
* 重定向标准输出,即将所有echo、var_dump等终端输出写到对应文件中
* 注意 此参数只有在以守护进程方式运行时有效
* @var string
*/
public static $stdoutFile = '/dev/null';
/**
* pid文件的路径及名称
* 例如 Worker::$pidFile = '/tmp/workerman.pid';
* 注意 此属性一般不必手动设置,默认会放到php临时目录中
* @var string
*/
public static $pidFile = '';
/**
* 日志目录,默认在workerman根目录下,与Applications同级
* 可以手动设置
* 例如 Worker::$logFile = '/tmp/workerman.log';
* @var mixed
*/
public static $logFile = '';
/**
* 全局事件轮询库,用于监听所有资源的可读可写事件
* @var Select/Libevent
*/
public static $globalEvent = null;
/**
* 主进程pid
* @var int
*/
protected static $_masterPid = 0;
/**
* 监听的socket
* @var resource
*/
protected $_mainSocket = null;
/**
* socket名称,包括应用层协议+ip+端口号,在初始化worker时设置
* 值类似 http://0.0.0.0:80
* @var string
*/
protected $_socketName = '';
/**
* socket的上下文,具体选项设置可以在初始化worker时传递
* @var array
*/
protected $_context = null;
/**
* 所有的worker实例
* @var array
*/
protected static $_workers = array();
/**
* 所有worker进程的pid
* 格式为 [worker_id=>[pid=>pid, pid=>pid, ..], ..]
* @var array
*/
protected static $_pidMap = array();
/**
* 所有需要重启的进程pid
* 格式为 [pid=>pid, pid=>pid]
* @var array
*/
protected static $_pidsToRestart = array();
/**
* 当前worker状态
* @var int
*/
protected static $_status = self::STATUS_STARTING;
/**
* 所有worke名称(name属性)中的最大长度,用于在运行 status 命令时格式化输出
* @var int
*/
protected static $_maxWorkerNameLength = 12;
/**
* 所有socket名称(_socketName属性)中的最大长度,用于在运行 status 命令时格式化输出
* @var int
*/
protected static $_maxSocketNameLength = 12;
/**
* 所有user名称(user属性)中的最大长度,用于在运行 status 命令时格式化输出
* @var int
*/
protected static $_maxUserNameLength = 12;
/**
* 运行 status 命令时用于保存结果的文件名
* @var string
*/
protected static $_statisticsFile = '';
/**
* 启动的全局入口文件
* 例如 php start.php start ,则入口文件为start.php
* @var string
*/
protected static $_startFile = '';
/**
* 全局统计数据,用于在运行 status 命令时展示
* 统计的内容包括 workerman启动的时间戳及每组worker进程的退出次数及退出状态码
* @var array
*/
protected static $_globalStatistics = array(
'start_timestamp' => 0,
'worker_exit_info' => array()
);
/**
* 运行所有worker实例
* @return void
*/
public static function runAll()
{
// 初始化环境变量
self::init();
// 解析命令
self::parseCommand();
// 尝试以守护进程模式运行
self::daemonize();
// 初始化所有worker实例,主要是监听端口
self::initWorkers();
// 初始化所有信号处理函数
self::installSignal();
// 保存主进程pid
self::saveMasterPid();
// 创建子进程(worker进程)并运行
self::forkWorkers();
// 展示启动界面
self::displayUI();
// 尝试重定向标准输入输出
self::resetStd();
// 监控所有子进程(worker进程)
self::monitorWorkers();
}
/**
* 初始化一些环境变量
* @return void
*/
public static function init()
{
// 如果没设置$pidFile,则生成默认值
if(empty(self::$pidFile))
{
$backtrace = debug_backtrace();
self::$_startFile = $backtrace[count($backtrace)-1]['file'];
self::$pidFile = sys_get_temp_dir()."/workerman.".str_replace('/', '_', self::$_startFile).".pid";
}
// 没有设置日志文件,则生成一个默认值
if(empty(self::$logFile))
{
self::$logFile = __DIR__ . '/../workerman.log';
}
// 标记状态为启动中
self::$_status = self::STATUS_STARTING;
// 启动时间戳
self::$_globalStatistics['start_timestamp'] = time();
// 设置status文件位置
self::$_statisticsFile = sys_get_temp_dir().'/workerman.status';
// 尝试设置进程名称(需要php>=5.5或者安装了proctitle扩展)
self::setProcessTitle('WorkerMan: master process start_file=' . self::$_startFile);
// 初始化定时器
Timer::init();
}
/**
* 初始化所有的worker实例,主要工作为获得格式化所需数据及监听端口
* @return void
*/
protected static function initWorkers()
{
/** @var static $worker */
foreach(self::$_workers as $worker)
{
// 没有设置worker名称,则使用none代替
if(empty($worker->name))
{
$worker->name = 'none';
}
// 获得所有worker名称中最大长度
$worker_name_length = strlen($worker->name);
if(self::$_maxWorkerNameLength < $worker_name_length)
{
self::$_maxWorkerNameLength = $worker_name_length;
}
// 获得所有_socketName中最大长度
$socket_name_length = strlen($worker->getSocketName());
if(self::$_maxSocketNameLength < $socket_name_length)
{
self::$_maxSocketNameLength = $socket_name_length;
}
// 获得运行用户名的最大长度
if(empty($worker->user) || posix_getuid() !== 0)
{
$worker->user = self::getCurrentUser();
}
$user_name_length = strlen($worker->user);
if(self::$_maxUserNameLength < $user_name_length)
{
self::$_maxUserNameLength = $user_name_length;
}
// 监听端口
$worker->listen();
}
}
/**
* 获得运行当前进程的用户名
* @return string
*/
protected static function getCurrentUser()
{
$user_info = posix_getpwuid(posix_getuid());
return $user_info['name'];
}
/**
* 展示启动界面
* @return void
*/
protected static function displayUI()
{
echo "\033[1A\n\033[K-----------------------\033[47;30m WORKERMAN \033[0m-----------------------------\n\033[0m";
echo 'Workerman version:' , Worker::VERSION , " PHP version:",PHP_VERSION,"\n";
echo "------------------------\033[47;30m WORKERS \033[0m-------------------------------\n";
echo "\033[47;30muser\033[0m",str_pad('', self::$_maxUserNameLength+2-strlen('user')), "\033[47;30mworker\033[0m",str_pad('', self::$_maxWorkerNameLength+2-strlen('worker')), "\033[47;30mlisten\033[0m",str_pad('', self::$_maxSocketNameLength+2-strlen('listen')), "\033[47;30mprocesses\033[0m \033[47;30m","status\033[0m\n";
/** @var static $worker */
foreach(self::$_workers as $worker)
{
echo str_pad($worker->user, self::$_maxUserNameLength+2),str_pad($worker->name, self::$_maxWorkerNameLength+2),str_pad($worker->getSocketName(), self::$_maxSocketNameLength+2), str_pad(' '.$worker->count, 9), " \033[32;40m [OK] \033[0m\n";;
}
echo "----------------------------------------------------------------\n";
if(self::$daemonize)
{
global $argv;
$start_file = $argv[0];
echo "Input \"php $start_file stop\" to quit. Start success.\n";
}
else
{
echo "Press Ctrl-C to quit. Start success.\n";
}
}
/**
* 解析运行命令
* php yourfile.php start | stop | restart | reload | status
* @return void
*/
public static function parseCommand()
{
// 检查运行命令的参数
global $argv;
$start_file = $argv[0];
if(!isset($argv[1]))
{
exit("Usage: php yourfile.php {start|stop|restart|reload|status}\n");
}
// 命令
$command = trim($argv[1]);
// 子命令,目前只支持-d
$command2 = isset($argv[2]) ? $argv[2] : '';
// 记录日志
$mode = '';
if($command === 'start')
{
if($command2 === '-d')
{
$mode = 'in DAEMON mode';
}
else
{
$mode = 'in DEBUG mode';
}
}
self::log("Workerman[$start_file] $command $mode");
// 检查主进程是否在运行
$master_pid = @file_get_contents(self::$pidFile);
$master_is_alive = $master_pid && @posix_kill($master_pid, 0);
if($master_is_alive)
{
if($command === 'start')
{
self::log("Workerman[$start_file] is running");
}
}
elseif($command !== 'start' && $command !== 'restart')
{
self::log("Workerman[$start_file] not run");
}
// 根据命令做相应处理
switch($command)
{
// 启动 workerman
case 'start':
if($command2 === '-d')
{
Worker::$daemonize = true;
}
break;
// 显示 workerman 运行状态
case 'status':
// 尝试删除统计文件,避免脏数据
if(is_file(self::$_statisticsFile))
{
@unlink(self::$_statisticsFile);
}
// 向主进程发送 SIGUSR2 信号 ,然后主进程会向所有子进程发送 SIGUSR2 信号
// 所有进程收到 SIGUSR2 信号后会向 $_statisticsFile 写入自己的状态
posix_kill($master_pid, SIGUSR2);
// 睡眠100毫秒,等待子进程将自己的状态写入$_statisticsFile指定的文件
usleep(100000);
// 展示状态
readfile(self::$_statisticsFile);
exit(0);
// 重启 workerman
case 'restart':
// 停止 workeran
case 'stop':
self::log("Workerman[$start_file] is stoping ...");
// 想主进程发送SIGINT信号,主进程会向所有子进程发送SIGINT信号
$master_pid && posix_kill($master_pid, SIGINT);
// 如果 $timeout 秒后主进程没有退出则展示失败界面
$timeout = 5;
$start_time = time();
while(1)
{
// 检查主进程是否存活
$master_is_alive = $master_pid && posix_kill($master_pid, 0);
if($master_is_alive)
{
// 检查是否超过$timeout时间
if(time() - $start_time >= $timeout)
{
self::log("Workerman[$start_file] stop fail");
exit;
}
usleep(10000);
continue;
}
self::log("Workerman[$start_file] stop success");
// 是restart命令
if($command === 'stop')
{
exit(0);
}
// -d 说明是以守护进程的方式启动
if($command2 === '-d')
{
Worker::$daemonize = true;
}
break;
}
break;
// 平滑重启 workerman
case 'reload':
posix_kill($master_pid, SIGUSR1);
self::log("Workerman[$start_file] reload");
exit;
// 未知命令
default :
exit("Usage: php yourfile.php {start|stop|restart|reload|status}\n");
}
}
/**
* 安装信号处理函数
* @return void
*/
protected static function installSignal()
{
// stop
pcntl_signal(SIGINT, array('\Workerman\Worker', 'signalHandler'), false);
// reload
pcntl_signal(SIGUSR1, array('\Workerman\Worker', 'signalHandler'), false);
// status
pcntl_signal(SIGUSR2, array('\Workerman\Worker', 'signalHandler'), false);
// ignore
pcntl_signal(SIGPIPE, SIG_IGN, false);
}
/**
* 为子进程重新安装信号处理函数,使用全局事件轮询监听信号
* @return void
*/
protected static function reinstallSignal()
{
// uninstall stop signal handler
pcntl_signal(SIGINT, SIG_IGN, false);
// uninstall reload signal handler
pcntl_signal(SIGUSR1, SIG_IGN, false);
// uninstall status signal handler
pcntl_signal(SIGUSR2, SIG_IGN, false);
// reinstall stop signal handler
self::$globalEvent->add(SIGINT, EventInterface::EV_SIGNAL, array('\Workerman\Worker', 'signalHandler'));
// uninstall reload signal handler
self::$globalEvent->add(SIGUSR1, EventInterface::EV_SIGNAL,array('\Workerman\Worker', 'signalHandler'));
// uninstall status signal handler
self::$globalEvent->add(SIGUSR2, EventInterface::EV_SIGNAL, array('\Workerman\Worker', 'signalHandler'));
}
/**
* 信号处理函数
* @param int $signal
*/
public static function signalHandler($signal)
{
switch($signal)
{
// stop
case SIGINT:
self::stopAll();
break;
// reload
case SIGUSR1:
self::$_pidsToRestart = self::getAllWorkerPids();
self::reload();
break;
// show status
case SIGUSR2:
self::writeStatisticsToStatusFile();
break;
}
}
/**
* 尝试以守护进程的方式运行
* @throws Exception
*/
protected static function daemonize()
{
if(!self::$daemonize)
{
return;
}
umask(0);
$pid = pcntl_fork();
if(-1 === $pid)
{
throw new Exception('fork fail');
}
elseif($pid > 0)
{
exit(0);
}
if(-1 === posix_setsid())
{
throw new Exception("setsid fail");
}
// fork again avoid SVR4 system regain the control of terminal
$pid = pcntl_fork();
if(-1 === $pid)
{
throw new Exception("fork fail");
}
elseif(0 !== $pid)
{
exit(0);
}
}
/**
* 重定向标准输入输出
* @throws Exception
*/
protected static function resetStd()
{
if(!self::$daemonize)
{
return;
}
global $STDOUT, $STDERR;
$handle = fopen(self::$stdoutFile,"a");
if($handle)
{
unset($handle);
@fclose(STDOUT);
@fclose(STDERR);
$STDOUT = fopen(self::$stdoutFile,"a");
$STDERR = fopen(self::$stdoutFile,"a");
}
else
{
throw new Exception('can not open stdoutFile ' . self::$stdoutFile);
}
}
/**
* 保存pid到文件中,方便运行命令时查找主进程pid
* @throws Exception
*/
protected static function saveMasterPid()
{
self::$_masterPid = posix_getpid();
if(false === @file_put_contents(self::$pidFile, self::$_masterPid))
{
throw new Exception('can not save pid to ' . self::$pidFile);
}
}
/**
* 获得所有子进程的pid
* @return array
*/
protected static function getAllWorkerPids()
{
$pid_array = array();
foreach(self::$_pidMap as $worker_pid_array)
{
foreach($worker_pid_array as $worker_pid)
{
$pid_array[$worker_pid] = $worker_pid;
}
}
return $pid_array;
}
/**
* 创建子进程
* @return void
*/
protected static function forkWorkers()
{
/** @var static $worker */
foreach(self::$_workers as $worker)
{
// 启动过程中需要得到运行用户名的最大长度,在status时格式化展示
if(self::$_status === self::STATUS_STARTING)
{
if(empty($worker->name))
{
$worker->name = $worker->getSocketName();
}
$worker_name_length = strlen($worker->name);
if(self::$_maxWorkerNameLength < $worker_name_length)
{
self::$_maxWorkerNameLength = $worker_name_length;
}
}
// 创建子进程
while(count(self::$_pidMap[$worker->workerId]) < $worker->count)
{
static::forkOneWorker($worker);
}
}
}
/**
* 创建一个子进程
* @param Worker $worker
* @throws Exception
*/
protected static function forkOneWorker($worker)
{
$pid = pcntl_fork();
// 主进程记录子进程pid
if($pid > 0)
{
self::$_pidMap[$worker->workerId][$pid] = $pid;
}
// 子进程运行
elseif(0 === $pid)
{
// 启动过程中尝试重定向标准输出
if(self::$_status === self::STATUS_STARTING)
{
self::resetStd();
}
self::$_pidMap = array();
self::$_workers = array($worker->workerId => $worker);
Timer::delAll();
self::setProcessTitle('WorkerMan: worker process ' . $worker->name . ' ' . $worker->getSocketName());
self::setProcessUser($worker->user);
$worker->run();
exit(250);
}
else
{
throw new Exception("forkOneWorker fail");
}
}
/**
* 尝试设置运行当前进程的用户
*
* @param $user_name
*/
protected static function setProcessUser($user_name)
{
if(empty($user_name) || posix_getuid() !== 0)
{
return;
}
$user_info = posix_getpwnam($user_name);
if($user_info['uid'] != posix_getuid() || $user_info['gid'] != posix_getgid())
{
if(!posix_setgid($user_info['gid']) || !posix_setuid($user_info['uid']))
{
self::log( 'Notice : Can not run woker as '.$user_name." , You shuld be root\n", true);
}
}
}
/**
* 设置当前进程的名称,在ps aux命令中有用
* 注意 需要php>=5.5或者安装了protitle扩展
* @param string $title
* @return void
*/
protected static function setProcessTitle($title)
{
// >=php 5.5
if (function_exists('cli_set_process_title'))
{
@cli_set_process_title($title);
}
// 需要扩展
elseif(extension_loaded('proctitle') && function_exists('setproctitle'))
{
@setproctitle($title);
}
}
/**
* 监控所有子进程的退出事件及退出码
* @return void
*/
protected static function monitorWorkers()
{
self::$_status = self::STATUS_RUNNING;
while(1)
{
// 如果有信号到来,尝试触发信号处理函数
pcntl_signal_dispatch();
// 挂起进程,直到有子进程退出或者被信号打断
$status = 0;
$pid = pcntl_wait($status, WUNTRACED);
// 有子进程退出
if($pid > 0)
{
// 查找是哪个进程组的,然后再启动新的进程补上
foreach(self::$_pidMap as $worker_id => $worker_pid_array)
{
if(isset($worker_pid_array[$pid]))
{
$worker = self::$_workers[$worker_id];
// 检查退出状态
if($status !== 0)
{
self::log("worker[".$worker->name.":$pid] exit with status $status");
}
// 统计,运行status命令时使用
if(!isset(self::$_globalStatistics['worker_exit_info'][$worker_id][$status]))
{
self::$_globalStatistics['worker_exit_info'][$worker_id][$status] = 0;
}
self::$_globalStatistics['worker_exit_info'][$worker_id][$status]++;
// 清除子进程信息
unset(self::$_pidMap[$worker_id][$pid]);
break;
}
}
// 如果不是关闭状态,则补充新的进程
if(self::$_status !== self::STATUS_SHUTDOWN)
{
self::forkWorkers();
// 如果该进程是因为运行reload命令退出,则继续执行reload流程
if(isset(self::$_pidsToRestart[$pid]))
{
unset(self::$_pidsToRestart[$pid]);
self::reload();
}
}
else
{
// 如果是关闭状态,并且所有进程退出完毕,则主进程退出
if(!self::getAllWorkerPids())
{
self::exitAndClearAll();
}
}
}
else
{
// 如果是关闭状态,并且所有进程退出完毕,则主进程退出
if(self::$_status === self::STATUS_SHUTDOWN && !self::getAllWorkerPids())
{
self::exitAndClearAll();
}
}
}
}
/**
* 退出当前进程
* @return void
*/
protected static function exitAndClearAll()
{
@unlink(self::$pidFile);
self::log("Workerman[".basename(self::$_startFile)."] has been stopped");
exit(0);
}
/**
* 执行平滑重启流程
* @return void
*/
protected static function reload()
{
// 主进程部分
if(self::$_masterPid === posix_getpid())
{
// 设置为平滑重启状态
if(self::$_status !== self::STATUS_RELOADING && self::$_status !== self::STATUS_SHUTDOWN)
{
self::log("Workerman[".basename(self::$_startFile)."] reloading");
self::$_status = self::STATUS_RELOADING;
}
// 如果有worker设置了reloadable=false,则过滤掉
$reloadable_pid_array = array();
foreach(self::$_pidMap as $worker_id =>$worker_pid_array)
{
$worker = self::$_workers[$worker_id];
if($worker->reloadable)
{
foreach($worker_pid_array as $pid)
{
$reloadable_pid_array[$pid] = $pid;
}
}
}
// 得到所有可以重启的进程
self::$_pidsToRestart = array_intersect(self::$_pidsToRestart , $reloadable_pid_array);
// 平滑重启完毕
if(empty(self::$_pidsToRestart))
{
if(self::$_status !== self::STATUS_SHUTDOWN)
{
self::$_status = self::STATUS_RUNNING;
}
return;
}
// 继续执行平滑重启流程
$one_worker_pid = current(self::$_pidsToRestart );
// 给子进程发送平滑重启信号
posix_kill($one_worker_pid, SIGUSR1);
// 定时器,如果子进程在KILL_WORKER_TIMER_TIME秒后没有退出,则强行杀死
Timer::add(self::KILL_WORKER_TIMER_TIME, 'posix_kill', array($one_worker_pid, SIGKILL), false);
}
// 子进程部分
else
{
// 如果当前worker的reloadable属性为真,则执行退出
$worker = current(self::$_workers);
if($worker->reloadable)
{
self::stopAll();
}
}
}
/**
* 执行关闭流程
* @return void
*/
public static function stopAll()
{
self::$_status = self::STATUS_SHUTDOWN;
// 主进程部分
if(self::$_masterPid === posix_getpid())
{
self::log("Workerman[".basename(self::$_startFile)."] Stopping ...");
$worker_pid_array = self::getAllWorkerPids();
// 向所有子进程发送SIGINT信号,表明关闭服务
foreach($worker_pid_array as $worker_pid)
{
posix_kill($worker_pid, SIGINT);
Timer::add(self::KILL_WORKER_TIMER_TIME, 'posix_kill', array($worker_pid, SIGKILL),false);
}
}
// 子进程部分
else
{
// 执行stop逻辑
/** @var static $worker */