EasySwoole RPC 自定义注册中心

EasySwoole 默认为通过UDP广播的方式来实现无主化的服务发现。但有些情况，不方便用UDP广播的情况下，那么EasySwoole支持你自定义一个节点管理器，来变更服务发现方式。

例如用Redis来实现

  1. <?php
  2. /**
  3.  * Created by PhpStorm.
  4.  * User: yf
  5.  * Date: 2019-02-25
  6.  * Time: 14:46
  7.  */
  9. namespace EasySwoole\Rpc\NodeManager;
  11. use EasySwoole\Component\Pool\PoolManager;
  12. use EasySwoole\Rpc\Config;
  13. use EasySwoole\Rpc\ServiceNode;
  14. use EasySwoole\Utility\Random;
  16. class RedisManager implements NodeManagerInterface
  17. {
  18.     const KEY = '__rpcRedisKey';
  19.     private $pool;
  21.     /**
  22.      * 初始化服务节点管理器
  23.      * RedisManager constructor.
  24.      * @param Config $config
  25.      */
  26.     public function __construct(Config $config)
  27.     {
  28.         $config = $config->getExtra();
  29.         //获取redis连接
  30.         PoolManager::getInstance()->registerAnonymous('__rpcRedis', function () use ($config) {
  31.             $redis = new \Redis();
  32.             $redis->connect($config['host'], $config['port']);
  33.             if (!empty($config['auth'])) {
  34.                 $redis->auth($config['auth']);
  35.             }
  36.             $redis->setOption(\Redis::OPT_SERIALIZER, \Redis::SERIALIZER_PHP);
  37.             return $redis;
  38.         });
  39.         $this->pool = PoolManager::getInstance()->getPool('__rpcRedis');
  40.     }
  42.     /**
  43.      * 获取所有服务
  44.      * getServiceNodes
  45.      * @param string      $serviceName
  46.      * @param string|null $version
  47.      * @return array
  48.      * @author Tioncico
  49.      * Time: 9:38
  50.      */
  51.     function getServiceNodes(string $serviceName, ?string $version = null): array
  52.     {
  53.         $list = [];
  54.         $nodeList = $this->allServiceNodes();//获取所有服务
  55.         foreach ($nodeList as $item) {
  56.             $serviceNode = new ServiceNode($item);
  57.             if ($serviceNode->getServiceName() == $serviceName) {
  58.                 if ($version !== null && $serviceNode->getServiceVersion() != $version) {
  59.                     continue;
  60.                 }
  61.                 $list[$serviceNode->getNodeId()] = $serviceNode->toArray();
  62.             }
  63.         }
  64.         return $list;
  65.     }
  67.     /**
  68.      * 获取某个服务节点的详情
  69.      * getServiceNode
  70.      * @param string      $serviceName
  71.      * @param string|null $version
  72.      * @return ServiceNode|null
  73.      * @author Tioncico
  74.      * Time: 9:39
  75.      */
  76.     function getServiceNode(string $serviceName, ?string $version = null): ?ServiceNode
  77.     {
  78.         $list = $this->getServiceNodes($serviceName, $version);
  79.         $num = count($list);
  80.         if ($num == 0) {
  81.             return null;
  82.         }
  83.         return new ServiceNode(Random::arrayRandOne($list));
  84.     }
  86.     /**
  87.      * 取所有节点的列表数据
  88.      * allServiceNodes
  89.      * @return array
  90.      * @author Tioncico
  91.      * Time: 9:39
  92.      */
  93.     function allServiceNodes(): array
  94.     {
  95.         $list = [];
  96.         if ($obj = $this->pool->getObj()) {
  97.             $nodeList = $obj->hGetAll(self::KEY);
  98.             $this->pool->recycleObj($obj);
  99.             foreach ($nodeList as $key => $serviceNode) {
  100.                 if ($serviceNode->getNodeExpire() !== null && time() > $serviceNode->getNodeExpire()) {
  101.                     $this->deleteServiceNode($serviceNode);//超时删除
  102.                     continue;
  103.                 }
  104.                 $list[] = $serviceNode->toArray();
  105.             }
  106.         }
  107.         return $list;
  108.     }
  110.     /**
  111.      * 删除某个节点
  112.      * deleteServiceNode
  113.      * @param ServiceNode $serviceNode
  114.      * @return bool
  115.      * @author Tioncico
  116.      * Time: 9:39
  117.      */
  118.     function deleteServiceNode(ServiceNode $serviceNode): bool
  119.     {
  120.         if ($obj = $this->pool->getObj()) {
  121.             $obj->hDel(self::KEY, $serviceNode->getNodeId());
  122.             $this->pool->recycleObj($obj);
  123.             return true;
  124.         }
  125.         return false;
  126.     }
  128.     /**
  129.      * 注册服务节点
  130.      * registerServiceNode
  131.      * @param ServiceNode $serviceNode
  132.      * @return bool
  133.      * @author Tioncico
  134.      * Time: 9:39
  135.      */
  136.     function registerServiceNode(ServiceNode $serviceNode): bool
  137.     {
  138.         if ($obj = $this->pool->getObj()) {
  139.             $obj->hSet(self::KEY, $serviceNode->getNodeId(), $serviceNode);
  140.             $this->pool->recycleObj($obj);
  141.             return true;
  142.         }
  143.         return false;
  144.     }
  145. }

注意，设置自定义节点管理器后，就不再需要启用UDP定时广播进程了。请在创建RPC实例后，自己创建一个ServiceNode对象，刷新到注册中心。节点下线也是同理，在你服务关闭的时候，向节点管理器下线该节点。