c1e74d8e68
- 按 docs/renew-code-comment.md 将 PHP 文件头改为带边框的 Author 注释\n- 注释中的 crmeb.com 替换为 uj345.cn（代码字符串中的外链未改）\n- 新增 docs/renew-code-comment.md 说明 Made-with: Cursor
159 lines
3.6 KiB
PHP
159 lines
3.6 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | Author: ScottPan Team
|
|
// +----------------------------------------------------------------------
|
|
|
|
namespace crmeb\services;
|
|
|
|
|
|
use think\facade\Config;
|
|
use Swoole\Coroutine\Channel;
|
|
use Swoole\Coroutine\Redis;
|
|
|
|
/**
|
|
* Class RedisService
|
|
* @author 等风来
|
|
* @email 136327134@qq.com
|
|
* @date 2022/10/29
|
|
* @package crmeb\services
|
|
*/
|
|
class RedisService
|
|
{
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
protected $config;
|
|
|
|
/**
|
|
* 获取进程池最大时间
|
|
* @var int
|
|
*/
|
|
protected $maxWaitTime = 3;
|
|
|
|
/**
|
|
* 最大连接池数量
|
|
* @var int
|
|
*/
|
|
protected $maxActive;
|
|
|
|
/**
|
|
* @var Channel
|
|
*/
|
|
protected $pool;
|
|
|
|
/**
|
|
* RedisService constructor.
|
|
* @param array $config
|
|
* @param array $pool
|
|
*/
|
|
public function __construct(array $config = [], array $pool = [])
|
|
{
|
|
$this->config = $config ?: Config::get('cache.stores.redis');
|
|
$this->maxActive = $pool['maxActive'] ?? Config::get('swoole.pool.redis.cache.max_active', 50);
|
|
$this->maxWaitTime = $pool['maxWaitTime'] ?? Config::get('swoole.pool.redis.cache.max_wait_time', 5);
|
|
}
|
|
|
|
/**
|
|
* 初始化连接池
|
|
* @author 等风来
|
|
* @email 136327134@qq.com
|
|
* @date 2022/10/29
|
|
*/
|
|
public function init()
|
|
{
|
|
$this->pool = new Channel($this->maxActive);
|
|
|
|
go(function () {
|
|
for ($i = 0; $i < $this->maxActive; $i++) {
|
|
$redis = $this->createConnect($this->config);
|
|
$this->pool->push($redis);
|
|
}
|
|
});
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 创建连接
|
|
* @param array $config
|
|
* @return mixed
|
|
* @author 等风来
|
|
* @email 136327134@qq.com
|
|
* @date 2022/10/29
|
|
*/
|
|
public function createConnect(array $config)
|
|
{
|
|
$connection = new Redis();
|
|
$res = $connection->connect($config['host'], $config['port'], \Redis::SERIALIZER_PHP);
|
|
if ($config['password']) {
|
|
$connection->auth($config['password']);
|
|
}
|
|
|
|
$connection->select($config['select']);
|
|
|
|
if ($res === false) {
|
|
throw new \RuntimeException("Failed to connect redis server");
|
|
}
|
|
|
|
return $connection;
|
|
}
|
|
|
|
/**
|
|
* 销毁连接
|
|
* @param $connection
|
|
* @author 等风来
|
|
* @email 136327134@qq.com
|
|
* @date 2022/10/29
|
|
*/
|
|
public function removeConnect($connection)
|
|
{
|
|
\Swoole\Coroutine\defer(function () use ($connection) { //释放
|
|
$connection->close();
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 获取redis连接句柄
|
|
* @return Redis
|
|
* @author 等风来
|
|
* @email 136327134@qq.com
|
|
* @date 2022/10/29
|
|
*/
|
|
public function getRedis()
|
|
{
|
|
|
|
if ($this->pool->isEmpty()) {
|
|
$redis = $this->createConnect($this->config);
|
|
$this->removeConnect($redis);
|
|
return $redis;
|
|
}
|
|
|
|
$redis = $this->pool->pop($this->maxWaitTime);
|
|
|
|
if (false === $redis) {
|
|
$redis = $this->createConnect($this->config);
|
|
$this->removeConnect($redis);
|
|
} else {
|
|
\Swoole\Coroutine\defer(function () use ($redis) { //释放
|
|
$this->pool->push($redis);
|
|
});
|
|
}
|
|
|
|
return $redis;
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $arguments
|
|
* @return Redis
|
|
* @author 等风来
|
|
* @email 136327134@qq.com
|
|
* @date 2022/10/29
|
|
*/
|
|
public function __call($name, $arguments)
|
|
{
|
|
return call_user_func_array([$this->getRedis(), $name], $arguments);
|
|
}
|
|
}
|