109 lines
1.9 KiB
PHP
109 lines
1.9 KiB
PHP
<?php
|
|
|
|
|
|
namespace Rpc;
|
|
|
|
|
|
use Exception;
|
|
use ReflectionException;
|
|
use Snowflake\Abstracts\Component;
|
|
use Snowflake\Abstracts\Config;
|
|
use Snowflake\Exception\ComponentException;
|
|
use Snowflake\Exception\ConfigException;
|
|
use Snowflake\Exception\NotFindClassException;
|
|
use Snowflake\Snowflake;
|
|
|
|
|
|
/**
|
|
* Class Producer
|
|
* @package Rpc
|
|
*/
|
|
class Producer extends Component
|
|
{
|
|
|
|
private array $producers = [];
|
|
|
|
|
|
private array $classAlias = [];
|
|
|
|
|
|
private array $consumers = [];
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @param array $handler
|
|
* @param array $node
|
|
*/
|
|
public function addProducer(string $name, array $handler, array $node)
|
|
{
|
|
$this->classAlias[get_class($handler[0])] = $name;
|
|
|
|
$this->consumers[$name] = $handler[0];
|
|
|
|
$this->producers[$name] = $node;
|
|
|
|
if ($handler[0] instanceof IProducer) {
|
|
$handler[0]->initClient();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function get($name): mixed
|
|
{
|
|
if (!isset($this->consumers[$name])) {
|
|
throw new Exception('Unknown rpc client.');
|
|
}
|
|
return $this->consumers[$name];
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function getClient(string $name): Client
|
|
{
|
|
if (!is_array($producer = $this->producers[$name] ?? null)) {
|
|
throw new Exception('Unknown rpc client config.');
|
|
}
|
|
$producerName = $this->getName($name, $producer);
|
|
|
|
$snowflake = Snowflake::app();
|
|
if (!$snowflake->has($producerName)) {
|
|
return $snowflake->set($producerName, ['class' => Client::class, 'config' => $producer]);
|
|
} else {
|
|
return $snowflake->get($producerName);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @return Client|bool
|
|
* @throws Exception
|
|
*/
|
|
public function __get($name): Client|bool
|
|
{
|
|
return $this->get($name); // TODO: Change the autogenerated stub
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $rand
|
|
* @return string
|
|
*/
|
|
private function getName($name, $rand): string
|
|
{
|
|
return 'rpc.client.' . $name . '.' . $rand['host'];
|
|
}
|
|
|
|
}
|