Master.php 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979
  1. <?php
  2. namespace Man\Core;
  3. if(!defined('WORKERMAN_ROOT_DIR'))
  4. {
  5. define('WORKERMAN_ROOT_DIR', realpath(__DIR__."/../../")."/");
  6. }
  7. require_once WORKERMAN_ROOT_DIR . 'Core/Lib/Checker.php';
  8. require_once WORKERMAN_ROOT_DIR . 'Core/Lib/Config.php';
  9. require_once WORKERMAN_ROOT_DIR . 'Core/Lib/Task.php';
  10. require_once WORKERMAN_ROOT_DIR . 'Core/Lib/Log.php';
  11. require_once WORKERMAN_ROOT_DIR . 'Core/Lib/Mutex.php';
  12. /**
  13. *
  14. * 主进程
  15. *
  16. * @package Core
  17. *
  18. * @author walkor <walkor@workerman.net>
  19. * <b>使用示例:</b>
  20. * <pre>
  21. * <code>
  22. * Man\Core\Master::run();
  23. * <code>
  24. * </pre>
  25. *
  26. */
  27. class Master
  28. {
  29. /**
  30. * 版本
  31. * @var string
  32. */
  33. const VERSION = '2.1.4';
  34. /**
  35. * 服务名
  36. * @var string
  37. */
  38. const NAME = 'WorkerMan';
  39. /**
  40. * 服务状态 启动中
  41. * @var integer
  42. */
  43. const STATUS_STARTING = 1;
  44. /**
  45. * 服务状态 运行中
  46. * @var integer
  47. */
  48. const STATUS_RUNNING = 2;
  49. /**
  50. * 服务状态 关闭中
  51. * @var integer
  52. */
  53. const STATUS_SHUTDOWN = 4;
  54. /**
  55. * 服务状态 平滑重启中
  56. * @var integer
  57. */
  58. const STATUS_RESTARTING_WORKERS = 8;
  59. /**
  60. * 整个服务能够启动的最大进程数
  61. * @var integer
  62. */
  63. const SERVER_MAX_WORKER_COUNT = 5000;
  64. /**
  65. * 单个进程打开文件数限制
  66. * @var integer
  67. */
  68. const MIN_SOFT_OPEN_FILES = 10000;
  69. /**
  70. * 单个进程打开文件数限制 硬性限制
  71. * @var integer
  72. */
  73. const MIN_HARD_OPEN_FILES = 10000;
  74. /**
  75. * 共享内存中用于存储主进程统计信息的变量id
  76. * @var integer
  77. */
  78. const STATUS_VAR_ID = 1;
  79. /**
  80. * 发送停止命令多久后worker没退出则发送sigkill信号
  81. * @var integer
  82. */
  83. const KILL_WORKER_TIME_LONG = 4;
  84. /**
  85. * 默认listen的backlog,如果没配置backlog,则使用此值
  86. * @var integer
  87. */
  88. const DEFAULT_BACKLOG= 1024;
  89. /**
  90. * 用于保存所有子进程pid ['worker_name1'=>[pid1=>pid1,pid2=>pid2,..], 'worker_name2'=>[pid7,..], ...]
  91. * @var array
  92. */
  93. protected static $workerPidMap = array();
  94. /**
  95. * 服务的状态,默认是启动中
  96. * @var integer
  97. */
  98. protected static $serviceStatus = self::STATUS_STARTING;
  99. /**
  100. * 用来监听端口的Socket数组,用来fork worker使用
  101. * @var array
  102. */
  103. protected static $listenedSocketsArray = array();
  104. /**
  105. * 要重启r的pid数组 [pid1=>time_stamp, pid2=>time_stamp, ..]
  106. * @var array
  107. */
  108. protected static $pidsToRestart = array();
  109. /**
  110. * 共享内存resource id
  111. * @var resource
  112. */
  113. protected static $shmId = 0;
  114. /**
  115. * 消息队列 resource id
  116. * @var resource
  117. */
  118. protected static $queueId = 0;
  119. /**
  120. * master进程pid
  121. * @var integer
  122. */
  123. protected static $masterPid = 0;
  124. /**
  125. * server统计信息 ['start_time'=>time_stamp, 'worker_exit_code'=>['worker_name1'=>[code1=>count1, code2=>count2,..], 'worker_name2'=>[code3=>count3,...], ..] ]
  126. * @var array
  127. */
  128. protected static $serviceStatusInfo = array(
  129. 'start_time' => 0,
  130. 'worker_exit_code' => array(),
  131. );
  132. /**
  133. * 服务运行
  134. * @return void
  135. */
  136. public static function run()
  137. {
  138. // 输出信息
  139. self::notice("Workerman is starting ...", true);
  140. // 初始化
  141. self::init();
  142. // 检查环境
  143. self::checkEnv();
  144. // 变成守护进程
  145. self::daemonize();
  146. // 保存进程pid
  147. self::savePid();
  148. // 安装信号
  149. self::installSignal();
  150. // 创建监听套接字
  151. self::createListeningSockets();
  152. // 创建worker进程
  153. self::spawnWorkers();
  154. // 输出信息
  155. self::notice("\033[1A\n\033[KWorkerman start success ...\033[0m", true);
  156. // 标记服务状态为运行中
  157. self::$serviceStatus = self::STATUS_RUNNING;
  158. // 初始化任务
  159. \Man\Core\Lib\Task::init();
  160. // 关闭标准输出
  161. self::resetStdFd();
  162. // 主循环
  163. self::loop();
  164. }
  165. /**
  166. * 初始化 配置、进程名、共享内存、消息队列等
  167. * @return void
  168. */
  169. public static function init()
  170. {
  171. // 获取配置文件
  172. $config_path = Lib\Config::$configFile;
  173. // 设置进程名称,如果支持的话
  174. self::setProcTitle(self::NAME.':master with-config:' . $config_path);
  175. // 初始化共享内存消息队列
  176. if(extension_loaded('sysvmsg') && extension_loaded('sysvshm'))
  177. {
  178. self::$shmId = shm_attach(IPC_KEY, DEFAULT_SHM_SIZE, 0666);
  179. self::$queueId = msg_get_queue(IPC_KEY, 0666);
  180. msg_set_queue(self::$queueId,array('msg_qbytes'=>65535));
  181. }
  182. }
  183. /**
  184. * 检查环境配置
  185. * @return void
  186. */
  187. public static function checkEnv()
  188. {
  189. // 检查PID文件
  190. Lib\Checker::checkPidFile();
  191. // 检查扩展支持情况
  192. Lib\Checker::checkExtension();
  193. // 检查函数禁用情况
  194. Lib\Checker::checkDisableFunction();
  195. // 检查log目录是否可读
  196. Lib\Log::init();
  197. // 检查配置和语法错误等
  198. Lib\Checker::checkWorkersConfig();
  199. // 检查文件限制
  200. Lib\Checker::checkLimit();
  201. }
  202. /**
  203. * 使之脱离终端,变为守护进程
  204. * @return void
  205. */
  206. protected static function daemonize()
  207. {
  208. // 设置umask
  209. umask(0);
  210. // fork一次
  211. $pid = pcntl_fork();
  212. if(-1 == $pid)
  213. {
  214. // 出错退出
  215. exit("Can not fork");
  216. }
  217. elseif($pid > 0)
  218. {
  219. // 父进程,退出
  220. exit(0);
  221. }
  222. // 成为session leader
  223. if(-1 == posix_setsid())
  224. {
  225. // 出错退出
  226. exit("Setsid fail");
  227. }
  228. // 再fork一次
  229. $pid2 = pcntl_fork();
  230. if(-1 == $pid2)
  231. {
  232. // 出错退出
  233. exit("Can not fork");
  234. }
  235. elseif(0 !== $pid2)
  236. {
  237. // 禁止进程重新打开控制终端
  238. exit(0);
  239. }
  240. // 记录服务启动时间
  241. self::$serviceStatusInfo['start_time'] = time();
  242. }
  243. /**
  244. * 保存主进程pid
  245. * @return void
  246. */
  247. public static function savePid()
  248. {
  249. // 保存在变量中
  250. self::$masterPid = posix_getpid();
  251. // 保存到文件中,用于实现停止、重启
  252. if(false === @file_put_contents(WORKERMAN_PID_FILE, self::$masterPid))
  253. {
  254. exit("\033[31;40mCan not save pid to pid-file(" . WORKERMAN_PID_FILE . ")\033[0m\n\n\033[31;40mServer start fail\033[0m\n\n");
  255. }
  256. // 更改权限
  257. chmod(WORKERMAN_PID_FILE, 0644);
  258. }
  259. /**
  260. * 获取主进程pid
  261. * @return int
  262. */
  263. public static function getMasterPid()
  264. {
  265. return self::$masterPid;
  266. }
  267. /**
  268. * 根据配置文件,创建监听套接字
  269. * @return void
  270. */
  271. protected static function createListeningSockets()
  272. {
  273. // 循环读取配置创建socket
  274. foreach (Lib\Config::getAllWorkers() as $worker_name=>$config)
  275. {
  276. if(isset($config['listen']))
  277. {
  278. $context = self::getSocketContext($worker_name);
  279. $flags = substr($config['listen'], 0, 3) == 'udp' ? STREAM_SERVER_BIND : STREAM_SERVER_BIND | STREAM_SERVER_LISTEN;
  280. $error_no = 0;
  281. $error_msg = '';
  282. // 创建监听socket
  283. if($context)
  284. {
  285. self::$listenedSocketsArray[$worker_name] = stream_socket_server($config['listen'], $error_no, $error_msg, $flags, $context);
  286. }
  287. else
  288. {
  289. self::$listenedSocketsArray[$worker_name] = stream_socket_server($config['listen'], $error_no, $error_msg, $flags);
  290. }
  291. if(!self::$listenedSocketsArray[$worker_name])
  292. {
  293. Lib\Log::add("can not create socket {$config['listen']} info:{$error_no} {$error_msg}\tServer start fail");
  294. exit("\n\033[31;40mCan not create socket {$config['listen']} {$error_msg}\033[0m\n\n\033[31;40mWorkerman start fail\033[0m\n\n");
  295. }
  296. }
  297. }
  298. }
  299. /**
  300. * 根据配置文件创建Workers
  301. * @return void
  302. */
  303. protected static function spawnWorkers()
  304. {
  305. // 生成一定量的worker进程
  306. foreach (Lib\Config::getAllWorkers() as $worker_name=>$config)
  307. {
  308. // 初始化
  309. if(empty(self::$workerPidMap[$worker_name]))
  310. {
  311. self::$workerPidMap[$worker_name] = array();
  312. }
  313. while(count(self::$workerPidMap[$worker_name]) < $config['start_workers'])
  314. {
  315. // 子进程退出
  316. if(self::createOneWorker($worker_name) == 0)
  317. {
  318. self::notice("Worker exit unexpected");
  319. exit(500);
  320. }
  321. }
  322. }
  323. }
  324. /**
  325. * 创建一个worker进程
  326. * @param string $worker_name 服务名
  327. * @return int 父进程:>0得到新worker的pid ;<0 出错; 子进程:始终为0
  328. */
  329. protected static function createOneWorker($worker_name)
  330. {
  331. // 创建子进程
  332. $pid = pcntl_fork();
  333. // 先处理收到的信号
  334. pcntl_signal_dispatch();
  335. // 父进程
  336. if($pid > 0)
  337. {
  338. // 初始化master的一些东东
  339. self::$workerPidMap[$worker_name][$pid] = $pid;
  340. // 更新进程信息到共享内存
  341. self::updateStatusToShm();
  342. return $pid;
  343. }
  344. // 子进程
  345. elseif($pid === 0)
  346. {
  347. // 忽略信号
  348. self::ignoreSignal();
  349. // 清空任务
  350. Lib\Task::delAll();
  351. // 关闭不用的监听socket
  352. foreach(self::$listenedSocketsArray as $tmp_worker_name => $tmp_socket)
  353. {
  354. if($tmp_worker_name != $worker_name)
  355. {
  356. fclose($tmp_socket);
  357. }
  358. }
  359. // 尝试以指定用户运行worker进程
  360. if($worker_user = Lib\Config::get($worker_name . '.user'))
  361. {
  362. self::setProcUser($worker_user);
  363. }
  364. // 关闭输出
  365. self::resetStdFd(Lib\Config::get($worker_name.'.no_debug'));
  366. // 尝试设置子进程进程名称
  367. self::setWorkerProcTitle($worker_name);
  368. // 包含必要文件
  369. require_once WORKERMAN_ROOT_DIR . 'Core/SocketWorker.php';
  370. // 查找worker文件
  371. $worker_file = \Man\Core\Lib\Config::get($worker_name.'.worker_file');
  372. $class_name = basename($worker_file, '.php');
  373. // 如果有语法错误 sleep 5秒 避免狂刷日志
  374. if(\Man\Core\Lib\Checker::checkSyntaxError($worker_file, $class_name))
  375. {
  376. sleep(5);
  377. }
  378. require_once $worker_file;
  379. // 创建实例
  380. $worker = new $class_name($worker_name);
  381. // 如果该worker有配置监听端口,则将监听端口的socket传递给子进程
  382. if(isset(self::$listenedSocketsArray[$worker_name]))
  383. {
  384. $worker->setListendSocket(self::$listenedSocketsArray[$worker_name]);
  385. }
  386. // 使worker开始服务
  387. $worker->start();
  388. return 0;
  389. }
  390. // 出错
  391. else
  392. {
  393. self::notice("create worker fail worker_name:$worker_name detail:pcntl_fork fail");
  394. return $pid;
  395. }
  396. }
  397. /**
  398. * 安装相关信号控制器
  399. * @return void
  400. */
  401. protected static function installSignal()
  402. {
  403. // 设置终止信号处理函数
  404. pcntl_signal(SIGINT, array('\Man\Core\Master', 'signalHandler'), false);
  405. // 设置SIGUSR1信号处理函数,测试用
  406. pcntl_signal(SIGUSR1, array('\Man\Core\Master', 'signalHandler'), false);
  407. // 设置SIGUSR2信号处理函数,平滑重启Server
  408. pcntl_signal(SIGHUP, array('\Man\Core\Master', 'signalHandler'), false);
  409. // 设置子进程退出信号处理函数
  410. pcntl_signal(SIGCHLD, array('\Man\Core\Master', 'signalHandler'), false);
  411. // 设置忽略信号
  412. pcntl_signal(SIGPIPE, SIG_IGN);
  413. pcntl_signal(SIGTTIN, SIG_IGN);
  414. pcntl_signal(SIGTTOU, SIG_IGN);
  415. pcntl_signal(SIGQUIT, SIG_IGN);
  416. pcntl_signal(SIGALRM, SIG_IGN);
  417. }
  418. /**
  419. * 忽略信号
  420. * @return void
  421. */
  422. protected static function ignoreSignal()
  423. {
  424. // 设置忽略信号
  425. pcntl_signal(SIGPIPE, SIG_IGN);
  426. pcntl_signal(SIGTTIN, SIG_IGN);
  427. pcntl_signal(SIGTTOU, SIG_IGN);
  428. pcntl_signal(SIGQUIT, SIG_IGN);
  429. pcntl_signal(SIGALRM, SIG_IGN);
  430. pcntl_signal(SIGINT, SIG_IGN);
  431. pcntl_signal(SIGUSR1, SIG_IGN);
  432. pcntl_signal(SIGUSR2, SIG_IGN);
  433. pcntl_signal(SIGHUP, SIG_IGN);
  434. }
  435. /**
  436. * 设置server信号处理函数
  437. * @param null $null
  438. * @param int $signal
  439. * @return void
  440. */
  441. public static function signalHandler($signal)
  442. {
  443. switch($signal)
  444. {
  445. // 停止服务信号
  446. case SIGINT:
  447. self::notice("Workerman is shutting down");
  448. self::stop();
  449. break;
  450. // 测试用
  451. case SIGUSR1:
  452. break;
  453. // worker退出信号
  454. case SIGCHLD:
  455. // 这里什么也不做
  456. // self::checkWorkerExit();
  457. break;
  458. // 平滑重启server信号
  459. case SIGHUP:
  460. Lib\Config::reload();
  461. self::notice("Workerman reloading");
  462. $pid_worker_name_map = self::getPidWorkerNameMap();
  463. $pids_to_restart = array();
  464. foreach($pid_worker_name_map as $pid=>$worker_name)
  465. {
  466. // 如果对应进程配置了不热启动则不重启对应进程
  467. if(Lib\Config::get($worker_name.'.no_reload'))
  468. {
  469. // 发送reload信号,以便触发onReload方法
  470. posix_kill($pid, SIGHUP);
  471. continue;
  472. }
  473. $pids_to_restart[] = $pid;
  474. }
  475. self::addToRestartPids($pids_to_restart);
  476. self::restartPids();
  477. break;
  478. }
  479. }
  480. /**
  481. * 设置子进程进程名称
  482. * @param string $worker_name
  483. * @return void
  484. */
  485. public static function setWorkerProcTitle($worker_name)
  486. {
  487. if(isset(self::$listenedSocketsArray[$worker_name]))
  488. {
  489. // 获得socket的信息
  490. $sock_name = stream_socket_get_name(self::$listenedSocketsArray[$worker_name], false);
  491. // 更改进程名,如果支持的话
  492. $mata_data = stream_get_meta_data(self::$listenedSocketsArray[$worker_name]);
  493. $protocol = substr($mata_data['stream_type'], 0, 3);
  494. self::setProcTitle(self::NAME.":worker $worker_name {$protocol}://$sock_name");
  495. }
  496. else
  497. {
  498. self::setProcTitle(self::NAME.":worker $worker_name");
  499. }
  500. }
  501. /**
  502. * 主进程主循环 主要是监听子进程退出、服务终止、平滑重启信号
  503. * @return void
  504. */
  505. public static function loop()
  506. {
  507. while(1)
  508. {
  509. sleep(1);
  510. // 检查是否有进程退出
  511. self::checkWorkerExit();
  512. // 触发信号处理
  513. pcntl_signal_dispatch();
  514. }
  515. }
  516. /**
  517. * 监控worker进程状态,退出重启
  518. * @param resource $channel
  519. * @param int $flag
  520. * @param int $pid 退出的进程id
  521. * @return mixed
  522. */
  523. public static function checkWorkerExit()
  524. {
  525. // 由于SIGCHLD信号可能重叠导致信号丢失,所以这里要循环获取所有退出的进程id
  526. while(($pid = pcntl_waitpid(-1, $status, WUNTRACED | WNOHANG)) != 0)
  527. {
  528. // 如果是重启的进程,则继续重启进程
  529. if(isset(self::$pidsToRestart[$pid]) && self::$serviceStatus != self::STATUS_SHUTDOWN)
  530. {
  531. unset(self::$pidsToRestart[$pid]);
  532. self::restartPids();
  533. }
  534. // 出错
  535. if($pid < 0)
  536. {
  537. self::notice('pcntl_waitpid return '.$pid.' and pcntl_get_last_error = ' . pcntl_get_last_error());
  538. return $pid;
  539. }
  540. // 查找子进程对应的woker_name
  541. $pid_workname_map = self::getPidWorkerNameMap();
  542. $worker_name = isset($pid_workname_map[$pid]) ? $pid_workname_map[$pid] : '';
  543. // 没找到worker_name说明出错了
  544. if(empty($worker_name))
  545. {
  546. self::notice("child exist but not found worker_name pid:$pid");
  547. break;
  548. }
  549. // 进程退出状态不是0,说明有问题了
  550. if($status !== 0)
  551. {
  552. self::notice("worker[$pid:$worker_name] exit with status $status");
  553. }
  554. // 记录进程退出状态
  555. self::$serviceStatusInfo['worker_exit_code'][$worker_name][$status] = isset(self::$serviceStatusInfo['worker_exit_code'][$worker_name][$status]) ? self::$serviceStatusInfo['worker_exit_code'][$worker_name][$status] + 1 : 1;
  556. // 更新状态到共享内存
  557. self::updateStatusToShm();
  558. // 清理这个进程的数据
  559. self::clearWorker($worker_name, $pid);
  560. // 如果服务是不是关闭中
  561. if(self::$serviceStatus != self::STATUS_SHUTDOWN)
  562. {
  563. // 重新创建worker
  564. self::spawnWorkers();
  565. }
  566. // 判断是否都重启完毕
  567. else
  568. {
  569. $all_worker_pid = self::getPidWorkerNameMap();
  570. if(empty($all_worker_pid))
  571. {
  572. // 删除共享内存
  573. self::removeShmAndQueue();
  574. // 发送提示
  575. self::notice("Workerman stoped");
  576. // 删除pid文件
  577. @unlink(WORKERMAN_PID_FILE);
  578. exit(0);
  579. }
  580. }//end if
  581. }//end while
  582. }
  583. /**
  584. * 获取pid 到 worker_name 的映射
  585. * @return array ['pid1'=>'worker_name1','pid2'=>'worker_name2', ...]
  586. */
  587. public static function getPidWorkerNameMap()
  588. {
  589. $all_pid = array();
  590. foreach(self::$workerPidMap as $worker_name=>$pid_array)
  591. {
  592. foreach($pid_array as $pid)
  593. {
  594. $all_pid[$pid] = $worker_name;
  595. }
  596. }
  597. return $all_pid;
  598. }
  599. /**
  600. * 放入重启队列中
  601. * @param array $restart_pids
  602. * @return void
  603. */
  604. public static function addToRestartPids($restart_pids)
  605. {
  606. if(!is_array($restart_pids))
  607. {
  608. self::notice("addToRestartPids(".var_export($restart_pids, true).") \$restart_pids not array");
  609. return false;
  610. }
  611. // 将pid放入重启队列
  612. foreach($restart_pids as $pid)
  613. {
  614. if(!isset(self::$pidsToRestart[$pid]))
  615. {
  616. // 重启时间=0
  617. self::$pidsToRestart[$pid] = 0;
  618. }
  619. }
  620. }
  621. /**
  622. * 重启workers
  623. * @return void
  624. */
  625. public static function restartPids()
  626. {
  627. // 标记server状态
  628. if(self::$serviceStatus != self::STATUS_RESTARTING_WORKERS && self::$serviceStatus != self::STATUS_SHUTDOWN)
  629. {
  630. self::$serviceStatus = self::STATUS_RESTARTING_WORKERS;
  631. }
  632. // 没有要重启的进程了
  633. if(empty(self::$pidsToRestart))
  634. {
  635. self::$serviceStatus = self::STATUS_RUNNING;
  636. self::notice("\nWorker Restart Success");
  637. return true;
  638. }
  639. // 遍历要重启的进程 标记它们重启时间
  640. foreach(self::$pidsToRestart as $pid => $stop_time)
  641. {
  642. if($stop_time == 0)
  643. {
  644. self::$pidsToRestart[$pid] = time();
  645. posix_kill($pid, SIGHUP);
  646. Lib\Task::add(self::KILL_WORKER_TIME_LONG, array('\Man\Core\Master', 'forceKillWorker'), array($pid), false);
  647. break;
  648. }
  649. }
  650. }
  651. /**
  652. * worker进程退出时,master进程的一些清理工作
  653. * @param string $worker_name
  654. * @param int $pid
  655. * @return void
  656. */
  657. protected static function clearWorker($worker_name, $pid)
  658. {
  659. // 释放一些不用了的数据
  660. unset(self::$pidsToRestart[$pid], self::$workerPidMap[$worker_name][$pid]);
  661. }
  662. /**
  663. * 停止服务
  664. * @return void
  665. */
  666. public static function stop()
  667. {
  668. // 如果没有子进程则直接退出
  669. $all_worker_pid = self::getPidWorkerNameMap();
  670. if(empty($all_worker_pid))
  671. {
  672. exit(0);
  673. }
  674. // 标记server开始关闭
  675. self::$serviceStatus = self::STATUS_SHUTDOWN;
  676. // killWorkerTimeLong 秒后如果还没停止则强制杀死所有进程
  677. Lib\Task::add(self::KILL_WORKER_TIME_LONG, array('\Man\Core\Master', 'stopAllWorker'), array(true), false);
  678. // 停止所有worker
  679. self::stopAllWorker();
  680. }
  681. /**
  682. * 停止所有worker
  683. * @param bool $force 是否强制退出
  684. * @return void
  685. */
  686. public static function stopAllWorker($force = false)
  687. {
  688. // 获得所有pid
  689. $all_worker_pid = self::getPidWorkerNameMap();
  690. // 强行杀死
  691. if($force)
  692. {
  693. // 杀死所有子进程
  694. foreach($all_worker_pid as $pid=>$worker_name)
  695. {
  696. // 发送SIGKILL信号
  697. self::forceKillWorker($pid);
  698. self::notice("Kill workers[$worker_name] force");
  699. }
  700. }
  701. else
  702. {
  703. // 向所有子进程发送终止信号
  704. foreach($all_worker_pid as $pid=>$worker_name)
  705. {
  706. // 发送SIGINT信号
  707. posix_kill($pid, SIGINT);
  708. }
  709. }
  710. }
  711. /**
  712. * 强制杀死进程
  713. * @param int $pid
  714. * @return void
  715. */
  716. public static function forceKillWorker($pid)
  717. {
  718. if(posix_kill($pid, 0))
  719. {
  720. self::notice("Kill workers $pid force!");
  721. posix_kill($pid, SIGKILL);
  722. }
  723. }
  724. /**
  725. * 设置运行用户
  726. * @param string $worker_user
  727. * @return void
  728. */
  729. protected static function setProcUser($worker_user)
  730. {
  731. $user_info = posix_getpwnam($worker_user);
  732. if($user_info['uid'] != posix_getuid() || $user_info['gid'] != posix_getgid())
  733. {
  734. // 尝试设置gid uid
  735. if(!posix_setgid($user_info['gid']) || !posix_setuid($user_info['uid']))
  736. {
  737. self::notice( 'Notice : Can not run woker as '.$worker_user." , You shuld be root\n", true);
  738. }
  739. }
  740. }
  741. /**
  742. * 获取共享内存资源id
  743. * @return resource
  744. */
  745. public static function getShmId()
  746. {
  747. return self::$shmId;
  748. }
  749. /**
  750. * 获取消息队列资源id
  751. * @return resource
  752. */
  753. public static function getQueueId()
  754. {
  755. return self::$queueId;
  756. }
  757. /**
  758. * 关闭标准输入输出
  759. * @return void
  760. */
  761. protected static function resetStdFd($force = false)
  762. {
  763. // 如果此进程配置是no_debug,则关闭输出
  764. if(!$force)
  765. {
  766. // 开发环境不关闭标准输出,用于调试
  767. if(Lib\Config::get('workerman.debug') == 1 && posix_ttyname(STDOUT))
  768. {
  769. return;
  770. }
  771. }
  772. global $STDOUT, $STDERR;
  773. @fclose(STDOUT);
  774. @fclose(STDERR);
  775. // 将标准输出重定向到/dev/null
  776. $STDOUT = fopen('/dev/null',"rw+");
  777. $STDERR = fopen('/dev/null',"rw+");
  778. }
  779. /**
  780. * 更新主进程收集的状态信息到共享内存
  781. * @return bool
  782. */
  783. protected static function updateStatusToShm()
  784. {
  785. if(!self::$shmId)
  786. {
  787. return true;
  788. }
  789. return shm_put_var(self::$shmId, self::STATUS_VAR_ID, array_merge(self::$serviceStatusInfo, array('pid_map'=>self::$workerPidMap)));
  790. }
  791. /**
  792. * 销毁共享内存以及消息队列
  793. * @return void
  794. */
  795. protected static function removeShmAndQueue()
  796. {
  797. if(self::$shmId)
  798. {
  799. shm_remove(self::$shmId);
  800. }
  801. if(self::$queueId)
  802. {
  803. msg_remove_queue(self::$queueId);
  804. }
  805. }
  806. /**
  807. * 设置进程名称,需要proctitle支持 或者php>=5.5
  808. * @param string $title
  809. * @return void
  810. */
  811. protected static function setProcTitle($title)
  812. {
  813. // >=php 5.5
  814. if (function_exists('cli_set_process_title'))
  815. {
  816. @cli_set_process_title($title);
  817. }
  818. // 需要扩展
  819. elseif(extension_loaded('proctitle') && function_exists('setproctitle'))
  820. {
  821. @setproctitle($title);
  822. }
  823. }
  824. /**
  825. * 获得socket的上下文选项
  826. * @param string $worker_name
  827. * @return resource
  828. */
  829. protected function getSocketContext($worker_name)
  830. {
  831. $context = null;
  832. // 根据手册5.3.3之前版本stream_socket_server 不支持 backlog 选项
  833. if(version_compare(PHP_VERSION, '5.3.3') < 0)
  834. {
  835. return $context;
  836. }
  837. // 读取worker的backlog
  838. $backlog = (int)Lib\Config::get($worker_name . '.backlog');
  839. // 没有设置或者不合法则尝试使用workerman.conf中的backlog设置
  840. if($backlog <= 0)
  841. {
  842. $backlog = (int)Lib\Config::get('workerman.backlog');
  843. }
  844. // 都没设置backlog,使用默认值
  845. if($backlog <= 0)
  846. {
  847. $backlog = self::DEFAULT_BACKLOG;
  848. }
  849. // backlog选项
  850. $opts = array(
  851. 'socket' => array(
  852. 'backlog' => $backlog,
  853. ),
  854. );
  855. // 返回上下文
  856. $context = stream_context_create($opts);
  857. return $context;
  858. }
  859. /**
  860. * notice,记录到日志
  861. * @param string $msg
  862. * @param bool $display
  863. * @return void
  864. */
  865. public static function notice($msg, $display = false)
  866. {
  867. Lib\Log::add("Server:".trim($msg));
  868. if($display)
  869. {
  870. if(self::$serviceStatus == self::STATUS_STARTING && @posix_ttyname(STDOUT))
  871. {
  872. echo($msg."\n");
  873. }
  874. }
  875. }
  876. }