115 lines
3.0 KiB
PHP
115 lines
3.0 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
namespace Kiri\Di\Context;
|
|
|
|
use Swoole\Coroutine;
|
|
|
|
class CoroutineContext implements ContextInterface
|
|
{
|
|
|
|
|
|
/**
|
|
* @return bool
|
|
*/
|
|
public static function inCoroutine(): bool
|
|
{
|
|
return Coroutine::getCid() > -1;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $key
|
|
* @param mixed $value
|
|
* @param int|null $coroutineId
|
|
* @return mixed
|
|
*/
|
|
public static function set(string $key, mixed $value, ?int $coroutineId = null): mixed
|
|
{
|
|
// TODO: Implement set() method.
|
|
if (is_null($coroutineId)) {
|
|
$coroutineId = Coroutine::getCid();
|
|
}
|
|
return Coroutine::getContext($coroutineId)[$key] = $value;
|
|
}
|
|
|
|
/**
|
|
* @param string $key
|
|
* @param mixed|null $defaultValue
|
|
* @param int|null $coroutineId
|
|
* @return mixed
|
|
*/
|
|
public static function get(string $key, mixed $defaultValue = null, ?int $coroutineId = null): mixed
|
|
{
|
|
// TODO: Implement get() method.
|
|
if (is_null($coroutineId)) {
|
|
$coroutineId = Coroutine::getCid();
|
|
}
|
|
return Coroutine::getContext($coroutineId)[$key] ?? $defaultValue;
|
|
}
|
|
|
|
/**
|
|
* @param string $key
|
|
* @param int|null $coroutineId
|
|
* @return mixed
|
|
*/
|
|
public static function exists(string $key, ?int $coroutineId = null): bool
|
|
{
|
|
// TODO: Implement exists() method.
|
|
if (is_null($coroutineId)) {
|
|
$coroutineId = Coroutine::getCid();
|
|
}
|
|
return isset(Coroutine::getContext($coroutineId)[$key]);
|
|
}
|
|
|
|
/**
|
|
* @param string $key
|
|
* @param int|null $coroutineId
|
|
* @return void
|
|
*/
|
|
public static function remove(string $key, ?int $coroutineId = null): void
|
|
{
|
|
// TODO: Implement remove() method.
|
|
if (is_null($coroutineId)) {
|
|
$coroutineId = Coroutine::getCid();
|
|
}
|
|
Coroutine::getContext($coroutineId)[$key] = null;
|
|
unset(Coroutine::getContext($coroutineId)[$key]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $id
|
|
* @param int $value
|
|
* @param int|null $coroutineId
|
|
* @return int
|
|
*/
|
|
public static function increment(string $id, int $value = 1, ?int $coroutineId = null): int
|
|
{
|
|
if (is_null($coroutineId)) {
|
|
$coroutineId = Coroutine::getCid();
|
|
}
|
|
if (!isset(Coroutine::getContext($coroutineId)[$id])) {
|
|
Coroutine::getContext($coroutineId)[$id] = 0;
|
|
}
|
|
return Coroutine::getContext($coroutineId)[$id] += $value;
|
|
}
|
|
|
|
/**
|
|
* @param string $id
|
|
* @param int $value
|
|
* @param int|null $coroutineId
|
|
* @return int
|
|
*/
|
|
public static function decrement(string $id, int $value = 1, ?int $coroutineId = null): int
|
|
{
|
|
if (is_null($coroutineId)) {
|
|
$coroutineId = Coroutine::getCid();
|
|
}
|
|
if (!isset(Coroutine::getContext($coroutineId)[$id])) {
|
|
Coroutine::getContext($coroutineId)[$id] = 0;
|
|
}
|
|
return Coroutine::getContext($coroutineId)[$id] -= $value;
|
|
}
|
|
}
|