基于Swoole 解决百万级全站消息推送问题

问题场景描述

基于Swoole的WebSocket服务对站内的消息进行的推送，有个全站进行站内消息的推送很棘手，因为峰值的用户服务1600+/QPS,服务器的配置2核8G(的配置)。

难点在于：

• Http的服务是接收主站的请求，需要及时返回，响应时间不能久。

• Redis的执行时间不能太久，（Redis是单进程）慢请求会卡主其他的使用。

• 百万级用户场景，全站用户发送时间不定

• 旧版本是Crontab实现的，因为后台直接请求websocket服务改动大

以上诸多难题。

解决思路

难题有两个思路，先从业务场景分析，在从技术角度下手，因为全站消息通知是很久才会使用一次，有站内信做兜底，需要以保证业务可用性和性能，需要找到一个适中的策略。

需要在接收消息、执行任务(Crontab)的时候分别想办法。

1.接收消息

Http接收服务中加分布式锁，粒度15分钟，15分钟不管提交多少次，都执行有且只有一次。

       public function systemMessage(){
           //验证参数...略
           $lock = \EasySwoole\RedisPool\RedisPool::invoke(function (\EasySwoole\Redis\Redis $redis){
               $lock = $redis->get('systemMessageLock');
               return $lock;
           },self::REDIS_CONN_NAME );
           if($lock){
               return $this->writeJson(Status::CODE_OK,[],Status::getReasonPhrase(Status::CODE_OK));
           }
           //没有锁，继续下面的操作
       }
   
  

2.执行定时任务(Crontab)

Crontab 的执行粒度是1分钟，也就是说最大延迟59s是可以接收的，这里的难于处理的在于第一次没有执行完毕，又执行了第二次，以此类推，这样是不可控的，下面是测试的结果，debug超出内存大小限制了。

[2022-02-22 03:00:23][debug][error] : [Allowed memory size of 134217728 bytes exhausted
  

还有task重复执行的问题,说明第一次没有执行完，又执行了下一次…果然和之前设想的一样…

       $taskId:5workIndex:1
       $taskId:3workIndex:2
       $taskId:3workIndex:2
       $taskId:8workIndex:0
       $taskId:8workIndex:0
       $taskId:1workIndex:3
   
  

解决思路：对脚本执行时间和使用内存进行设置，以防止执行任务时处于可控中，业务代码中加锁，以防止计算任务的重复执行。

对锁回收进行补偿处理，如果15分钟没有回收，锁主动释放，附上Demo。

       function run( $taskId, $workerIndex)
           {
               ini_set('memory_limit', '1024M');
               set_time_limit(0);
               //读取队列中的数据
               $redis = \EasySwoole\RedisPool\RedisPool::defer('redis');
               $taskIdLock = $redis->get($this->taskLock);
               if($taskIdLock){
                   return ;
               }
               $json = $redis->lPop(self::PUSH_MSG_NOTICE_SYSTEM);
               if($json && !$taskIdLock){
                   $redis->setEx($this->taskLock,900,$taskId);
               }
               if(isset($json) && !empty($json)){
                       $server = ServerManager::getInstance()->getSwooleServer();
                       $start_fd = 0;
                       while(true)
                       {
                           $conn_list = $server->getClientList( $start_fd, $this->limit );
                           if ($conn_list===false || count($conn_list) === 0 || empty($conn_list))
                           {
                               //删除锁
                               \EasySwoole\RedisPool\RedisPool::invoke(function (\EasySwoole\Redis\Redis $redis) {
                                  $redis->del($this->taskLock);
                               }, self::REDIS_CONN_NAME);
                               break;
                           }
                           $start_fd = end($conn_list);
                           foreach ($conn_list as $fd){
                               $server->push($fd, json_encode($this->pushMsg));
                           }
                       }
               }
           }
   
  

https://blog.csdn.net/xuezhiwu001/article/details/123074506