kiri-rpc/JsonRpcPoolTransporter.php

<?php

namespace Kiri\Rpc;

use Kiri\Annotation\Inject;
use Exception;
use Kiri\Message\Response;
use Kiri\Message\Stream;
use Kiri\Abstracts\Config;
use Kiri\Exception\ConfigException;
use Psr\Http\Message\RequestInterface;
use Psr\Http\Message\ResponseInterface;
use Swoole\Coroutine\Client;

class JsonRpcPoolTransporter implements RpcClientInterface
{


	use TraitTransporter;


	#[Inject(ClientPool::class)]
	public ClientPool $pool;


	const POOL_NAME = 'rpc.client.pool';


	/**
	 */
	public function init()
	{
	}


	/**
	 * @param RequestInterface $request
	 * @return ResponseInterface
	 * @throws Exception
	 */
	public function sendRequest(RequestInterface $request): ResponseInterface
	{
		$content = $request->getBody()->getContents();

		$client = $this->getClient();

		$response = $this->request($client, $content);

		$this->pool->push($client, $this->config['Address'], $this->config['Port']);

		return (new Response())->withBody(new Stream($response));
	}


	/**
	 * @return Client|\Swoole\Client
	 * @throws ConfigException
	 * @throws Exception
	 */
	private function getClient(): Client|\Swoole\Client
	{
		$this->config['pool'] = Config::get('rpc.pool', ['max' => 10, 'min' => 1, 'waite' => 60]);
		return $this->pool->get($this->config, function () {
			return $this->newClient();
		});
	}


}