95 lines
1.6 KiB
PHP
95 lines
1.6 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake;
|
|
|
|
|
|
use Closure;
|
|
use Exception;
|
|
use Snowflake\Abstracts\Component;
|
|
use SplQueue;
|
|
use Swoole\Coroutine\Channel as CChannel;
|
|
|
|
|
|
/**
|
|
* Class Channel
|
|
* @package Snowflake
|
|
*/
|
|
class Channel extends Component
|
|
{
|
|
|
|
|
|
private array $_channels = [];
|
|
|
|
|
|
/**
|
|
* @param mixed $value
|
|
* @param string $name
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function push(mixed $value, string $name = ''): mixed
|
|
{
|
|
$channel = $this->channelInit($name);
|
|
if ($channel->isFull()) {
|
|
return $this->addError('Channel is full.');
|
|
}
|
|
return $channel->push($value);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @return bool|CChannel
|
|
*/
|
|
private function channelInit(string $name = ''): bool|SplQueue
|
|
{
|
|
if (!isset($this->_channels[$name]) || !($this->_channels[$name] instanceof SplQueue)) {
|
|
$this->_channels[$name] = new SplQueue();
|
|
}
|
|
return $this->_channels[$name];
|
|
}
|
|
|
|
|
|
/**
|
|
*
|
|
* 清空缓存
|
|
*/
|
|
public function cleanAll()
|
|
{
|
|
/** @var SplQueue $channel */
|
|
foreach ($this->_channels as $channel) {
|
|
while ($channel->count() > 0) {
|
|
$channel->dequeue();
|
|
}
|
|
}
|
|
$this->_channels = [];
|
|
}
|
|
|
|
/**
|
|
* @param $timeout
|
|
* @param Closure $closure
|
|
* @param string $name
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function pop(string $name, Closure $closure, int|float $timeout = null): mixed
|
|
{
|
|
if (($channel = $this->channelInit($name)) == false) {
|
|
return $this->addError('Channel is full.');
|
|
}
|
|
if (!$channel->isEmpty()) {
|
|
return $channel->pop();
|
|
}
|
|
if ($timeout !== null) {
|
|
$data = $channel->pop($timeout);
|
|
}
|
|
if (empty($data)) {
|
|
$data = call_user_func($closure);
|
|
}
|
|
return $data;
|
|
}
|
|
|
|
|
|
}
|