今天看了下swoole相关文档,用PHP也实现了一个,由于swoole
没有golang
的select
,所以实现的有点简单,但是实用性还可以,通过工厂函数实现了通用性。
swoole官方的协程池是用只能用在Redis。因为协程池代码层耦合了Redis实例化逻辑。
class RedisPool
{
/**
* @var \Swoole\Coroutine\Channel
*/
protected $pool;
/**
* RedisPool constructor.
* @param int $size 连接池的尺寸
*/
function __construct($size = 100)
{
$this->pool = new Swoole\Coroutine\Channel($size);
for ($i = 0; $i < $size; $i++)
{
$redis = new Swoole\Coroutine\Redis();
$res = $redis->connect('127.0.0.1', 6379);
if ($res == false)
{
throw new RuntimeException("failed to connect redis server.");
}
else
{
$this->put($redis);
}
}
}
function put($redis)
{
$this->pool->push($redis);
}
function get()
{
return $this->pool->pop();
}
}
利用工厂方法的改造如下:
<?php
/**
* @author xialeistudio
* @date 2019-05-20
*/
namespace swoole\foundation\pool;
use Swoole\Coroutine\Channel;
/**
* Swoole generic connection pool
* Class Pool
* @package swoole\foundation\pool
*/
class GenericPool
{
/**
* @var int pool size
*/
private $size = 0;
/**
* @var callable construct a connection
*/
private $factory = null;
/**
* @var Channel
*/
private $channel = null;
/**
* GenericPool constructor.
* @param int $size
* @param callable $factory
* @throws InvalidParamException
*/
public function __construct($size, callable $factory)
{
$this->size = $size;
$this->factory = $factory;
$this->init();
}
/**
* check pool config
* @throws InvalidParamException
*/
private function init()
{
if ($this->size <= 0) {
throw new InvalidParamException('The "size" property must be greater than zero.');
}
if (empty($this->factory)) {
throw new InvalidParamException('The "factory" property must be set.');
}
if (!is_callable($this->factory)) {
throw new InvalidParamException('The "factory" property must be callable.');
}
$this->bootstrap();
}
/**
* bootstrap pool
*/
private function bootstrap()
{
$this->channel = new Channel($this->size);
for ($i = 0; $i < $this->size; $i++) {
$this->channel->push(call_user_func($this->factory));
}
}
/**
* Acquire a connection
* @param int $timeout
* @return mixed
*/
public function acquire($timeout = 0)
{
return $this->channel->pop($timeout);
}
/**
* Release a resource
* @param mixed $resource
*/
public function release($resource)
{
$this->channel->push($resource);
}
}
我们代码里面不关心创建连接的具体逻辑,只要调用工厂方法即可。
项目地址
项目已经开源到githubhttps://github.com/swoole-foundation/swoole-pool
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。