变更

Context.php

@@ -0,0 +1,224 @@
+<?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;
+	}
+	
+}
+
+
+