kiri-container/Context.php

<?php


namespace Kiri\Di;

use Swoole\Coroutine;

/**
 * Class Context
 * @package Yoc\http
 */
class Context
{
	
	protected static array $_contents = [];
	
	
	/**
	 * @param $id
	 * @param $context
	 * @param null $coroutineId
	 * @return mixed
	 */
	public static function setContext($id, $context, $coroutineId = null): mixed
	{
		if (is_null($coroutineId)) {
			$coroutineId = Coroutine::getCid();
		}
		if (Coroutine::getCid() !== -1) {
			return Coroutine::getContext($coroutineId)[$id] = $context;
		}
		return static::$_contents[$id] = $context;
	}
	
	/**
	 * @param $id
	 * @param int $value
	 * @param null $coroutineId
	 * @return bool|int
	 */
	public static function increment($id, int $value = 1, $coroutineId = null): bool|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 $id
	 * @param int $value
	 * @param null $coroutineId
	 * @return bool|int
	 */
	public static function decrement($id, int $value = 1, $coroutineId = null): bool|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 $id
	 * @param null $default
	 * @param null $coroutineId
	 * @return mixed
	 */
	public static function getContext($id, $default = null, $coroutineId = null): mixed
	{
		if (Coroutine::getCid() === -1) {
			return static::loadByStatic($id, $default);
		}
		return static::loadByContext($id, $default, $coroutineId);
	}
	
	
	/**
	 * @param $id
	 * @param null $default
	 * @param null $coroutineId
	 * @return mixed
	 */
	private static function loadByContext($id, $default = null, $coroutineId = null): mixed
	{
		if (is_null($coroutineId)) {
			$coroutineId = Coroutine::getCid();
		}
		return Coroutine::getContext($coroutineId)[$id] ?? $default;
	}
	
	
	/**
	 * @param $id
	 * @param null $default
	 * @return mixed
	 */
	private static function loadByStatic($id, $default = null): mixed
	{
		return static::$_contents[$id] ?? $default;
	}
	
	
	/**
	 * @param null $coroutineId
	 * @return Coroutine\Context|array
	 */
	public static function getAllContext($coroutineId = null): Coroutine\Context|array
	{
		if (Coroutine::getCid() === -1) {
			return Coroutine::getContext((int)$coroutineId) ?? [];
		} else {
			return static::$_contents ?? [];
		}
	}
	
	
	/**
	 * @return void
	 */
	public static function clearAll(): void
	{
		if (Coroutine::getCid() === -1) {
			static::$_contents = [];
		}
	}
	
	/**
	 * @param string $id
	 * @param null $coroutineId
	 */
	public static function remove(string $id, $coroutineId = null)
	{
		if (is_null($coroutineId)) {
			$coroutineId = Coroutine::getCid();
		}
		if (!static::hasContext($id, $coroutineId)) {
			return;
		}
		if (Coroutine::getCid() === -1) {
			static::$_contents[$id] = null;
			
			unset(static::$_contents[$id]);
			
		} else {
			Coroutine::getContext($coroutineId)[$id] = null;
			
			unset(Coroutine::getContext($coroutineId)[$id]);
		}
	}
	
	/**
	 * @param $id
	 * @param null $key
	 * @param null $coroutineId
	 * @return bool
	 */
	public static function hasContext($id, $key = null, $coroutineId = null): bool
	{
		if (Coroutine::getCid() === -1) {
			return static::searchByStatic($id, $key);
		}
		return static::searchByCoroutine($id, $key, $coroutineId);
	}
	
	
	/**
	 * @param $id
	 * @param null $key
	 * @return bool
	 */
	private static function searchByStatic($id, $key = null): bool
	{
		if (!isset(static::$_contents[$id])) {
			return false;
		}
		$value = static::$_contents[$id];
		if (!empty($key) && is_array($value)) {
			return ($value[$key] ?? null) !== null;
		}
		return true;
	}
	
	
	/**
	 * @param $id
	 * @param null $key
	 * @param null $coroutineId
	 * @return bool
	 */
	private static function searchByCoroutine($id, $key = null, $coroutineId = null): bool
	{
		if (is_null($coroutineId)) {
			$coroutineId = Coroutine::getCid();
		}
		if (!isset(Coroutine::getContext($coroutineId)[$id])) {
			return false;
		}
		$value = Coroutine::getContext($coroutineId)[$id];
		if ($key !== null && is_array($value)) {
			return ($value[$key] ?? null) !== null;
		}
		return true;
	}
	
	
	/**
	 * @return bool
	 */
	public static function inCoroutine(): bool
	{
		return Coroutine::getCid() !== -1;
	}
	
}