From 8a40bd713087eee29b1a54dfa23be74c23d9a21c Mon Sep 17 00:00:00 2001
From: "as2252258@163.com" <as22522582>
Date: Sun, 21 Feb 2021 12:58:21 +0800
Subject: [PATCH] modify

---
 Database/Db.php | 542 ++++++++++++++++++++++++------------------------
 1 file changed, 274 insertions(+), 268 deletions(-)

diff --git a/Database/Db.php b/Database/Db.php
index 020de10f..b3781e9f 100644
--- a/Database/Db.php
+++ b/Database/Db.php
@@ -6,10 +6,12 @@
  * Time: 15:40
  */
 declare(strict_types=1);
+
 namespace Database;
 
 use Database\Traits\QueryTrait;
 use Exception;
+use HttpServer\Http\Context;
 use Snowflake\Snowflake;
 
 /**
@@ -18,308 +20,312 @@ use Snowflake\Snowflake;
  */
 class Db
 {
-	use QueryTrait;
+    use QueryTrait;
 
-	private static bool $isActive = false;
+    /**
+     * @return bool
+     */
+    public static function transactionsActive(): bool
+    {
+        return Context::getContext('begin:transaction') === true;
+    }
 
-	/**
-	 * @return bool
-	 */
-	public static function transactionsActive(): bool
-	{
-		return static::$isActive;
-	}
+    /**
+     * @throws Exception
+     */
+    public static function beginTransaction()
+    {
+        Context::setContext('begin:transaction', true);
+    }
 
-	/**
-	 * @throws Exception
-	 */
-	public static function beginTransaction()
-	{
-		static::$isActive = true;
-	}
+    /**
+     * @throws Exception
+     */
+    public static function commit()
+    {
+        if (!static::transactionsActive()) {
+            return;
+        }
+        $event = Snowflake::app()->getEvent();
+        $event->trigger(Connection::TRANSACTION_COMMIT);
+        $event->offName(Connection::TRANSACTION_COMMIT);
+        Context::deleteId('begin:transaction');
+    }
 
-	/**
-	 * @throws Exception
-	 */
-	public static function commit()
-	{
-		$event = Snowflake::app()->getEvent();
-		$event->trigger(Connection::TRANSACTION_COMMIT);
-		$event->offName(Connection::TRANSACTION_COMMIT);
-		static::$isActive = false;
-	}
+    /**
+     * @throws Exception
+     */
+    public static function rollback()
+    {
+        if (!static::transactionsActive()) {
+            return;
+        }
+        $event = Snowflake::app()->getEvent();
+        $event->trigger(Connection::TRANSACTION_ROLLBACK);
+        $event->offName(Connection::TRANSACTION_ROLLBACK);
+        Context::deleteId('begin:transaction');
+    }
 
-	/**
-	 * @throws Exception
-	 */
-	public static function rollback()
-	{
-		$event = Snowflake::app()->getEvent();
-		$event->trigger(Connection::TRANSACTION_ROLLBACK);
-		$event->offName(Connection::TRANSACTION_ROLLBACK);
-		static::$isActive = false;
-	}
+    /**
+     * @param $table
+     *
+     * @return static
+     */
+    public static function table($table): Db|static
+    {
+        $db = new Db();
+        $db->from($table);
+        return $db;
+    }
 
-	/**
-	 * @param $table
-	 *
-	 * @return static
-	 */
-	public static function table($table): Db|static
-	{
-		$db = new Db();
-		$db->from($table);
-		return $db;
-	}
+    /**
+     * @param string $column
+     * @param string $alias
+     * @return string
+     */
+    public static function any_value(string $column, string $alias = ''): string
+    {
+        if (empty($alias)) {
+            $alias = $column . '_any_value';
+        }
+        return 'ANY_VALUE(' . $column . ') as ' . $alias;
+    }
 
-	/**
-	 * @param string $column
-	 * @param string $alias
-	 * @return string
-	 */
-	public static function any_value(string $column, string $alias = ''): string
-	{
-		if (empty($alias)) {
-			$alias = $column . '_any_value';
-		}
-		return 'ANY_VALUE(' . $column . ') as ' . $alias;
-	}
+    /**
+     * @param Connection|null $db
+     * @return mixed
+     * @throws Exception
+     */
+    public function get(Connection $db = NULL): mixed
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
+        $query = $db->getSchema()->getQueryBuilder();
+        return $db->createCommand($query->getQuery($this))
+            ->all();
+    }
 
-	/**
-	 * @param Connection|null $db
-	 * @return mixed
-	 * @throws Exception
-	 */
-	public function get(Connection $db = NULL): mixed
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
-		$query = $db->getSchema()->getQueryBuilder();
-		return $db->createCommand($query->getQuery($this))
-			->all();
-	}
+    /**
+     * @param $column
+     * @return string
+     */
+    public static function raw($column): string
+    {
+        return '`' . $column . '`';
+    }
 
-	/**
-	 * @param $column
-	 * @return string
-	 */
-	public static function raw($column): string
-	{
-		return '`' . $column . '`';
-	}
+    /**
+     * @param Connection|null $db
+     * @return mixed
+     * @throws Exception
+     */
+    public function find(Connection $db = NULL): mixed
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
+        $query = $db->getSchema()->getQueryBuilder();
+        return $db->createCommand($query->getQuery($this))
+            ->one();
+    }
 
-	/**
-	 * @param Connection|null $db
-	 * @return mixed
-	 * @throws Exception
-	 */
-	public function find(Connection $db = NULL): mixed
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
-		$query = $db->getSchema()->getQueryBuilder();
-		return $db->createCommand($query->getQuery($this))
-			->one();
-	}
+    /**
+     * @param Connection|NULL $db
+     * @return bool|int
+     * @throws Exception
+     */
+    public function count(Connection $db = NULL): bool|int
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
+        $query = $db->getSchema()->getQueryBuilder();
+        return $db->createCommand($query->count($this))
+            ->rowCount();
+    }
 
-	/**
-	 * @param Connection|NULL $db
-	 * @return bool|int
-	 * @throws Exception
-	 */
-	public function count(Connection $db = NULL): bool|int
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
-		$query = $db->getSchema()->getQueryBuilder();
-		return $db->createCommand($query->count($this))
-			->rowCount();
-	}
+    /**
+     * @param Connection|NULL $db
+     * @return bool|int
+     * @throws Exception
+     */
+    public function exists(Connection $db = NULL): bool|int
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
+        $query = $db->getSchema()->getQueryBuilder();
+        return $db->createCommand($query->getQuery($this))
+            ->fetchColumn();
+    }
 
-	/**
-	 * @param Connection|NULL $db
-	 * @return bool|int
-	 * @throws Exception
-	 */
-	public function exists(Connection $db = NULL): bool|int
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
-		$query = $db->getSchema()->getQueryBuilder();
-		return $db->createCommand($query->getQuery($this))
-			->fetchColumn();
-	}
+    /**
+     * @param string $sql
+     * @param array $attributes
+     * @param Connection|null $db
+     * @return array|bool|int|string|null
+     * @throws Exception
+     */
+    public static function findAllBySql(string $sql, array $attributes = [], Connection $db = NULL): int|bool|array|string|null
+    {
+        return $db->createCommand($sql, $attributes)->all();
+    }
 
-	/**
-	 * @param string $sql
-	 * @param array $attributes
-	 * @param Connection|null $db
-	 * @return array|bool|int|string|null
-	 * @throws Exception
-	 */
-	public static function findAllBySql(string $sql, array $attributes = [], Connection $db = NULL): int|bool|array|string|null
-	{
-		return $db->createCommand($sql, $attributes)->all();
-	}
+    /**
+     * @param string $sql
+     * @param array $attributes
+     * @param Connection|NULL $db
+     * @return array|mixed
+     * @throws Exception
+     */
+    public static function findBySql(string $sql, array $attributes = [], Connection $db = NULL): mixed
+    {
+        return $db->createCommand($sql, $attributes)->one();
+    }
 
-	/**
-	 * @param string $sql
-	 * @param array $attributes
-	 * @param Connection|NULL $db
-	 * @return array|mixed
-	 * @throws Exception
-	 */
-	public static function findBySql(string $sql, array $attributes = [], Connection $db = NULL): mixed
-	{
-		return $db->createCommand($sql, $attributes)->one();
-	}
+    /**
+     * @param string $field
+     * @return array|null
+     * @throws Exception
+     */
+    public function values(string $field): ?array
+    {
+        $data = $this->get();
+        if (empty($data) || empty($field)) {
+            return NULL;
+        }
+        $first = current($data);
+        if (!isset($first[$field])) {
+            return NULL;
+        }
+        return array_column($data, $field);
+    }
 
-	/**
-	 * @param string $field
-	 * @return array|null
-	 * @throws Exception
-	 */
-	public function values(string $field): ?array
-	{
-		$data = $this->get();
-		if (empty($data) || empty($field)) {
-			return NULL;
-		}
-		$first = current($data);
-		if (!isset($first[$field])) {
-			return NULL;
-		}
-		return array_column($data, $field);
-	}
+    /**
+     * @param $field
+     * @return mixed
+     * @throws Exception
+     */
+    public function value($field): mixed
+    {
+        $data = $this->find();
+        if (!empty($field) && isset($data[$field])) {
+            return $data[$field];
+        }
+        return $data;
+    }
 
-	/**
-	 * @param $field
-	 * @return mixed
-	 * @throws Exception
-	 */
-	public function value($field): mixed
-	{
-		$data = $this->find();
-		if (!empty($field) && isset($data[$field])) {
-			return $data[$field];
-		}
-		return $data;
-	}
+    /**
+     * @param null $db
+     * @return bool|int
+     * @throws Exception
+     */
+    public function delete($db = null): bool|int
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
 
-	/**
-	 * @param null $db
-	 * @return bool|int
-	 * @throws Exception
-	 */
-	public function delete($db = null): bool|int
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
+        $query = $db->getBuild()->builder($this);
 
-		$query = $db->getBuild()->builder($this);
+        return $db->createCommand($query)->delete();
+    }
 
-		return $db->createCommand($query)->delete();
-	}
+    /**
+     * @param $table
+     * @param null $db
+     * @return bool|int
+     * @throws Exception
+     */
+    public static function drop($table, $db = null): bool|int
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
+        return $db->createCommand('DROP TABLE ' . $table)->delete();
+    }
 
-	/**
-	 * @param $table
-	 * @param null $db
-	 * @return bool|int
-	 * @throws Exception
-	 */
-	public static function drop($table, $db = null): bool|int
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
-		return $db->createCommand('DROP TABLE ' . $table)->delete();
-	}
+    /**
+     * @param $table
+     * @param null $db
+     * @return bool|int
+     * @throws Exception
+     */
+    public static function truncate($table, $db = null): bool|int
+    {
 
-	/**
-	 * @param $table
-	 * @param null $db
-	 * @return bool|int
-	 * @throws Exception
-	 */
-	public static function truncate($table, $db = null): bool|int
-	{
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
 
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
+        return $db->createCommand('TRUNCATE ' . $table)->exec();
+    }
 
-		return $db->createCommand('TRUNCATE ' . $table)->exec();
-	}
+    /**
+     * @param $table
+     * @param Connection|NULL $db
+     * @return mixed
+     * @throws Exception
+     */
+    public static function showCreateSql($table, Connection $db = NULL): mixed
+    {
 
-	/**
-	 * @param $table
-	 * @param Connection|NULL $db
-	 * @return mixed
-	 * @throws Exception
-	 */
-	public static function showCreateSql($table, Connection $db = NULL): mixed
-	{
-
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
 
 
-		if (empty($table)) {
-			return null;
-		}
+        if (empty($table)) {
+            return null;
+        }
 
-		return $db->createCommand('SHOW CREATE TABLE ' . $table)->one();
-	}
+        return $db->createCommand('SHOW CREATE TABLE ' . $table)->one();
+    }
 
-	/**
-	 * @param $table
-	 * @param Connection|NULL $db
-	 * @return bool|int|null
-	 * @throws Exception
-	 */
-	public static function desc($table, Connection $db = NULL): bool|int|null
-	{
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
+    /**
+     * @param $table
+     * @param Connection|NULL $db
+     * @return bool|int|null
+     * @throws Exception
+     */
+    public static function desc($table, Connection $db = NULL): bool|int|null
+    {
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
 
-		if (empty($table)) {
-			return null;
-		}
+        if (empty($table)) {
+            return null;
+        }
 
-		return $db->createCommand('SHOW FULL FIELDS FROM ' . $table)->all();
-	}
+        return $db->createCommand('SHOW FULL FIELDS FROM ' . $table)->all();
+    }
 
 
-	/**
-	 * @param string $table
-	 * @param Connection|NULL $db
-	 * @return mixed
-	 * @throws Exception
-	 */
-	public static function show(string $table, Connection $db = NULL): mixed
-	{
-		if (empty($table)) {
-			return null;
-		}
+    /**
+     * @param string $table
+     * @param Connection|NULL $db
+     * @return mixed
+     * @throws Exception
+     */
+    public static function show(string $table, Connection $db = NULL): mixed
+    {
+        if (empty($table)) {
+            return null;
+        }
 
-		if (empty($db)) {
-			$db = Snowflake::app()->database;
-		}
+        if (empty($db)) {
+            $db = Snowflake::app()->database;
+        }
 
-		$table = ['	const TABLE = \'select * from %s  where REFERENCED_TABLE_NAME=%s\';'];
+        $table = ['	const TABLE = \'select * from %s  where REFERENCED_TABLE_NAME=%s\';'];
 
-		return $db->createCommand((new Sql())
-			->select('*')
-			->from('INFORMATION_SCHEMA.KEY_COLUMN_USAGE')
-			->where(['REFERENCED_TABLE_NAME' => $table])
-			->getSql())->one();
-	}
+        return $db->createCommand((new Sql())
+            ->select('*')
+            ->from('INFORMATION_SCHEMA.KEY_COLUMN_USAGE')
+            ->where(['REFERENCED_TABLE_NAME' => $table])
+            ->getSql())->one();
+    }
 
 }