From 521bb2ee7fc2479a926802fd96e3b2aba828dc27 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=91=E6=9E=97?= <as2252258@163.com>
Date: Fri, 25 Feb 2022 18:37:49 +0800
Subject: [PATCH] modify plugin name

---
 Db.php | 46 ++++++++++++++++++++++++++--------------------
 1 file changed, 26 insertions(+), 20 deletions(-)

diff --git a/Db.php b/Db.php
index 24e0c26..12de300 100644
--- a/Db.php
+++ b/Db.php
@@ -15,8 +15,12 @@ use Database\Affair\Rollback;
 use Database\Traits\QueryTrait;
 use Exception;
 use Kiri\Abstracts\Config;
+use Kiri\Context;
 use Kiri\Events\EventDispatch;
 use Kiri\Exception\ConfigException;
+use Psr\Container\ContainerExceptionInterface;
+use Psr\Container\NotFoundExceptionInterface;
+use ReflectionException;
 
 /**
  * Class Db
@@ -32,47 +36,49 @@ class Db implements ISqlBuilder
 	/**
 	 * @return bool
 	 */
-	public static function transactionsActive(): bool
+	public static function inTransactionsActive(): bool
 	{
-		return static::$_inTransaction === true;
+		return Context::hasContext('transactions::status') && Context::getContext('transactions::status') === true;
 	}
 
 	/**
-	 * @throws Exception
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws ReflectionException
 	 */
 	public static function beginTransaction()
 	{
-		if (!static::transactionsActive()) {
-			$event = \Kiri::getDi()->get(EventDispatch::class);
-			$event->dispatch(new BeginTransaction());
-		}
-		static::$_inTransaction = true;
+		Context::setContext('transactions::status', true);
+
+		$event = \Kiri::getDi()->get(EventDispatch::class);
+		$event->dispatch(new BeginTransaction());
 	}
 
 
 	/**
-	 * @throws Exception
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws ReflectionException
 	 */
 	public static function commit()
 	{
-		if (static::transactionsActive()) {
-			$event = \Kiri::getDi()->get(EventDispatch::class);
-			$event->dispatch(new Commit());
-		}
-		static::$_inTransaction = false;
+		$event = \Kiri::getDi()->get(EventDispatch::class);
+		$event->dispatch(new Commit());
+		Context::remove('transactions::status');
 	}
 
 
 	/**
-	 * @throws Exception
+	 * @return void
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws ReflectionException
 	 */
 	public static function rollback()
 	{
-		if (static::transactionsActive()) {
-			$event = \Kiri::getDi()->get(EventDispatch::class);
-			$event->dispatch(new Rollback());
-		}
-		static::$_inTransaction = false;
+		$event = \Kiri::getDi()->get(EventDispatch::class);
+		$event->dispatch(new Rollback());
+		Context::remove('transactions::status');
 	}