modify plugin name

kiri-engine/Redis/Redis.php

@@ -0,0 +1,201 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: whwyy
+ * Date: 2018/4/27 0027
+ * Time: 11:00
+ */
+declare(strict_types=1);
+
+namespace Kiri\Redis;
+
+use Exception;
+use Kiri;
+use Kiri\Abstracts\Component;
+use Kiri\Abstracts\Config;
+use Kiri\Core\Json;
+use Kiri\Exception\ConfigException;
+use Kiri\Pool\Pool;
+use Kiri\Server\Events\OnWorkerExit;
+use Psr\Container\ContainerExceptionInterface;
+use Psr\Container\NotFoundExceptionInterface;
+
+/**
+ * Class Redis
+ * @package Kiri\Cache
+ * @mixin \Redis
+ */
+class Redis extends Component
+{
+
+
+	const REDIS_OPTION_HOST = 'host';
+	const REDIS_OPTION_PORT = 'port';
+	const REDIS_OPTION_PREFIX = 'prefix';
+	const REDIS_OPTION_AUTH = 'auth';
+	const REDIS_OPTION_DATABASES = 'databases';
+	const REDIS_OPTION_TIMEOUT = 'timeout';
+	const REDIS_OPTION_POOL = 'pool';
+	const REDIS_OPTION_POOL_TICK = 'tick';
+	const REDIS_OPTION_POOL_MIN = 'min';
+	const REDIS_OPTION_POOL_MAX = 'max';
+
+
+	private Kiri\Pool\Pool $pool;
+
+
+	/**
+	 * @return void
+	 * @throws ConfigException
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws Exception
+	 */
+	public function init()
+	{
+		$this->pool = Kiri::getDi()->get(Pool::class);
+
+		$config = $this->get_config();
+
+		$length = Config::get('cache.redis.pool.max', 10);
+
+		$this->getEventProvider()->on(OnWorkerExit::class, [$this, 'destroy'], 0);
+
+		$this->pool->initConnections($config['host'], $length);
+	}
+
+
+	/**
+	 * @param $name
+	 * @param $arguments
+	 * @return mixed
+	 * @throws
+	 */
+	public function __call($name, $arguments): mixed
+	{
+		$time = microtime(true);
+		if (method_exists($this, $name)) {
+			$data = $this->{$name}(...$arguments);
+		} else {
+			$data = $this->proxy($name, $arguments);
+		}
+		if (microtime(true) - $time >= 0.02) {
+			$this->logger->warning('Redis:' . Json::encode([$name, $arguments]) . (microtime(true) - $time));
+		}
+		return $data;
+	}
+
+
+	/**
+	 * @param $key
+	 * @param int $timeout
+	 * @return bool
+	 */
+	public function waite($key, int $timeout = 5): bool
+	{
+		$time = time();
+		while (!$this->setNx($key, 1)) {
+			if (time() - $time >= $timeout) {
+				return FALSE;
+			}
+			usleep(1000);
+		}
+		$this->expire($key, $timeout);
+		return TRUE;
+	}
+
+
+	/**
+	 * @param $key
+	 * @param int $timeout
+	 * @return bool|int
+	 * @throws Exception
+	 */
+	public function lock($key, int $timeout = 5): bool|int
+	{
+		$script = <<<SCRIPT
+local _nx = redis.call('setnx',KEYS[1], ARGV[1])
+if (_nx ~= 0) then
+	redis.call('expire',KEYS[1], ARGV[1])
+	return 1
+end
+return 0
+SCRIPT;
+		return $this->eval($script, ['{lock}:' . $key, $timeout], 1);
+	}
+
+
+	/**
+	 * @param $key
+	 * @return int
+	 * @throws Exception
+	 */
+	public function unlock($key): int
+	{
+		return $this->del('{lock}:' . $key);
+	}
+
+
+	/**
+	 * @throws ConfigException
+	 * @throws Exception
+	 */
+	public function release()
+	{
+		$this->pool->clean($this->get_config()['host']);
+	}
+
+	/**
+	 * 销毁连接池
+	 * @throws ConfigException
+	 * @throws Exception
+	 */
+	public function destroy()
+	{
+		$this->pool->clean($this->get_config()['host']);
+	}
+
+	/**
+	 * @param $name
+	 * @param $arguments
+	 * @return mixed
+	 * @throws ConfigException
+	 * @throws Exception
+	 */
+	public function proxy($name, $arguments): mixed
+	{
+		$client = $this->getClient();
+		try {
+			$response = $client->{$name}(...$arguments);
+		} catch (\Throwable $throwable) {
+			$response = $this->logger->addError($throwable->getMessage());
+		} finally {
+			$this->pool->push($this->get_config()['host'], $client);
+		}
+		return $response;
+	}
+
+
+	/**
+	 * @return Helper
+	 * @throws ConfigException
+	 */
+	private function getClient(): Helper
+	{
+		$config = $this->get_config();
+		return $this->pool->get($config['host'], static function () use ($config) {
+			return new Helper($config);
+		}, 10);
+	}
+
+
+	/**
+	 * @return array
+	 * @throws ConfigException
+	 */
+	public function get_config(): array
+	{
+		return Config::get('cache.redis', null, true);
+	}
+
+}