codispatcher.php 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190
  1. <?php
  2. declare(strict_types=0);
  3. define('APP_ID', 'cordispatcher');
  4. define('MOBILE_SERVER',true);
  5. define('USE_COROUTINE',true);
  6. define('SUPPORT_PTHREAD',false);
  7. define('BASE_ROOT_PATH',str_replace('/rdispatcher','',dirname(__FILE__)));
  8. define('BASE_PATH',BASE_ROOT_PATH . '/rdispatcher');
  9. require_once(BASE_ROOT_PATH . '/global.php');
  10. require_once(BASE_ROOT_PATH . '/fooder.php');
  11. require_once(BASE_HELPER_PATH . '/event_looper.php');
  12. require_once(BASE_HELPER_PATH . '/queue/rdispatcher.php');
  13. require_once(BASE_HELPER_PATH . '/algorithm.php');
  14. require_once(BASE_HELPER_PATH . '/refill/RefillFactory.php');
  15. require_once(BASE_PATH . '/processor.php');
  16. require_once(BASE_PATH . '/proxy.php');
  17. Co::set(['hook_flags' => SWOOLE_HOOK_NATIVE_CURL|SWOOLE_HOOK_SLEEP]);
  18. if (empty($_SERVER['argv'][1])) exit('parameter error');
  19. $process_count = intval($_SERVER['argv'][1]);
  20. function all_channels() {
  21. return ['refill'];
  22. }
  23. function strbool($value) {
  24. return $value ? 'true' : 'false';
  25. }
  26. function handle_error($level, $message, $file, $line)
  27. {
  28. if($level == E_NOTICE) return;
  29. $trace = "handle_error: level={$level},msg={$message} file={$file},line={$line}\n";
  30. $backtrace = debug_backtrace();
  31. foreach ($backtrace as $item) {
  32. $trace .= "{$item['file']}\t{$item['line']}\t{$item['function']}\n";
  33. }
  34. Log::record($trace,Log::ERR);
  35. }
  36. function subscribe_message(&$quit, &$redis, $channels)
  37. {
  38. $redis = new Swoole\Coroutine\Redis();
  39. while (!$quit)
  40. {
  41. try
  42. {
  43. Log::record("subscribe_message start quit=" . strbool($quit),Log::DEBUG);
  44. if(!$redis->connected) {
  45. $ret = $redis->connect(C('coroutine.redis_host'), C('coroutine.redis_port'));
  46. Log::record("subscribe_message redis connected = {$redis->connected}",Log::DEBUG);
  47. }
  48. else {
  49. $ret = true;
  50. }
  51. if(!$ret) {
  52. Log::record("subscribe_message cannot connet redis.",Log::DEBUG);
  53. $redis->close();
  54. }
  55. elseif($redis->subscribe($channels))
  56. {
  57. while ($msg = $redis->recv())
  58. {
  59. [$sub_type, $channel, $content] = $msg;
  60. $content = json_decode($content,true);
  61. $type = $content['type'];
  62. if($channel != 'refill') continue;
  63. if($quit) break;
  64. if($type == 'channel' || $type == 'merchant') {
  65. Log::record("subscribe_message recv mgs:{$sub_type}-{$channel}-{$type}",Log::DEBUG);
  66. refill\RefillFactory::instance()->load();
  67. }
  68. elseif($type == 'ratio') {
  69. Log::record("subscribe_message recv mgs:{$sub_type}-{$channel}-{$type}",Log::DEBUG);
  70. $start = microtime(true);
  71. $ins = Cache::getInstance('cacheredis');
  72. $val = $ins->get_org('channel_ratios');
  73. if(empty($val)) continue;
  74. $val = json_decode($val,true);
  75. if(empty($val)) continue;
  76. $ratios = $val['ratios'];
  77. if(empty($ratios)) continue;
  78. refill\RefillFactory::instance()->UpdateRatio($ratios);
  79. $use_time = microtime(true) - $start;
  80. $msg = sprintf("subscribe_message UpdateRatio use_time=%.6f",$use_time);
  81. Log::record($msg,Log::DEBUG);
  82. }
  83. else {
  84. Log::record("subscribe_message recv mgs:{$sub_type}-{$channel}-{$type}",Log::DEBUG);
  85. }
  86. }
  87. Log::record("subscribe_message redis recv timeout",Log::DEBUG);
  88. }
  89. else {
  90. Log::record("subscribe_message subscribe error",Log::ERR);
  91. }
  92. }
  93. catch (Exception $ex)
  94. {
  95. Log::record($ex->getMessage(),Log::ERR);
  96. }
  97. }
  98. Log::record("subscribe_message quit =" . strbool($quit),Log::DEBUG);
  99. $quit = true;
  100. }
  101. $workers = [];
  102. for ($i = 0; $i < $process_count;$i++)
  103. {
  104. $process = new Swoole\Process(function(Swoole\Process $worker)
  105. {
  106. Base::run_util();
  107. set_error_handler('handle_error');
  108. $sub_quit = false;
  109. $sub_redis = null;
  110. go(function () use (&$sub_quit,&$sub_redis) {
  111. subscribe_message($sub_quit,$sub_redis,['refill']);
  112. });
  113. $looper = new processor(true);
  114. go(function () use ($looper) {
  115. $looper->run();
  116. });
  117. Swoole\Process::signal(SIGTERM, function($signal_num) use ($worker,&$sub_quit,$sub_redis,$looper)
  118. {
  119. Log::record("signal call SIGTERM begin = $signal_num, #{$worker->pid}",Log::DEBUG);
  120. set_error_handler(null);
  121. try {
  122. $sub_quit = true;
  123. $sub_redis->close();
  124. }
  125. catch(Exception $ex) {
  126. Log::record($ex->getMessage(),Log::DEBUG);
  127. }
  128. $looper->stop();
  129. do {
  130. $res = Swoole\Coroutine::stats();
  131. $num = $res['coroutine_num'];
  132. if($num > 1) {
  133. Swoole\Coroutine::sleep(0.1);
  134. }
  135. Log::record("coroutine_num = {$num}",Log::DEBUG);
  136. } while($num > 1);
  137. });
  138. }, false, false, true);
  139. $pid = $process->start();
  140. $workers[$pid] = $process;
  141. }
  142. Log::record("main process start wait sub process....",Log::DEBUG);
  143. while (true)
  144. {
  145. if($status = Swoole\Process::wait(true)) {
  146. Log::record("Sub process #{$status['pid']} quit, code={$status['code']}, signal={$status['signal']}",Log::DEBUG);
  147. }
  148. else
  149. {
  150. foreach ($workers as $pid => $worker)
  151. {
  152. Swoole\Process::kill($pid,SIGTERM);
  153. if($status = Swoole\Process::wait(true)) {
  154. Log::record("Graceful Recycled #{$status['pid']}, code={$status['code']}, signal={$status['signal']}",Log::DEBUG);
  155. }
  156. }
  157. break;
  158. }
  159. }
  160. Log::record("Quit all",Log::DEBUG);