From d8b5990ee8cd62c2658978b443020f8c4c2e48ac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=91=E6=9E=97?= <as2252258@163.com>
Date: Mon, 12 Dec 2022 17:31:11 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=98=E6=9B=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Context.php | 224 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 224 insertions(+)
 create mode 100644 Context.php

diff --git a/Context.php b/Context.php
new file mode 100644
index 0000000..bdb136b
--- /dev/null
+++ b/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;
+	}
+	
+}
+
+
+