Revert "改名"

This reverts commit fdf58326

Mysql/Columns.php

@@ -0,0 +1,406 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: admin
+ * Date: 2019-03-18
+ * Time: 17:22
+ */
+
+declare(strict_types=1);
+
+namespace Database\Mysql;
+
+
+use Database\Connection;
+use Database\SqlBuilder;
+use Exception;
+use JetBrains\PhpStorm\Pure;
+use Kiri\Abstracts\Component;
+use Kiri\Core\Json;
+
+/**
+ * Class Columns
+ * @package Database\Mysql
+ */
+class Columns extends Component
+{
+
+	/**
+	 * @var array
+	 * field types
+	 */
+	private array $columns = [];
+
+	/**
+	 * @var Connection
+	 * Mysql client
+	 */
+	public Connection $db;
+
+	/**
+	 * @var string
+	 * tableName
+	 */
+	public string $table = '';
+
+	/**
+	 * @var array
+	 * field primary key
+	 */
+	private array $_primary = [];
+
+	/**
+	 * @var array
+	 * by mysql field auto_increment
+	 */
+	private array $_auto_increment = [];
+
+
+	private array $_fields = [];
+
+	/**
+	 * @param string $table
+	 * @return $this
+	 * @throws Exception
+	 */
+	public function table(string $table): static
+	{
+		$this->structure($this->table = $table);
+		return $this;
+	}
+
+	/**
+	 * @return string
+	 */
+	public function getTable(): string
+	{
+		return $this->table;
+	}
+
+	/**
+	 * @param $key
+	 * @param $val
+	 * @return mixed
+	 * @throws Exception
+	 */
+	public function fieldFormat($key, $val): mixed
+	{
+		return $this->encode($val, $this->get_fields($key));
+	}
+
+	/**
+	 * @param $data
+	 * @return array
+	 * @throws
+	 */
+	public function populate($data): array
+	{
+		$column = $this->get_fields();
+		foreach ($data as $key => $val) {
+			if (!isset($column[$key])) {
+				continue;
+			}
+			$data[$key] = $this->decode($val, $column[$key]);
+		}
+		return $data;
+	}
+
+	/**
+	 * @param $val
+	 * @param null $format
+	 * @return mixed
+	 */
+	public function decode($val, $format = null): mixed
+	{
+		if (empty($format) || $val === null) {
+			return $val;
+		}
+		$format = strtolower($format);
+		if ($this->isInt($format)) {
+			return (int)$val;
+		} else if ($this->isJson($format)) {
+			return Json::decode($val, true);
+		} else if ($this->isFloat($format)) {
+			return (float)$val;
+		} else {
+			return stripslashes($val);
+		}
+	}
+
+
+	/**
+	 * @param string $name
+	 * @param $value
+	 * @return mixed
+	 * @throws Exception
+	 */
+	public function _decode(string $name, $value): mixed
+	{
+		return $this->decode($value, $this->get_fields($name));
+	}
+
+
+	/**
+	 * @param $val
+	 * @param null $format
+	 * @return float|bool|int|string
+	 * @throws Exception
+	 */
+	public function encode($val, $format = null): float|bool|int|string
+	{
+		if (empty($format)) {
+			return $val;
+		}
+		$format = strtolower($format);
+		if ($this->isInt($format)) {
+			return (int)$val;
+		} else if ($this->isJson($format)) {
+			return Json::encode($val);
+		} else if ($this->isFloat($format)) {
+			return (float)$val;
+		} else {
+			return addslashes($val);
+		}
+	}
+
+	/**
+	 * @param $format
+	 * @return bool
+	 */
+	#[Pure] public function isInt($format): bool
+	{
+		return in_array($format, ['int', 'bigint', 'tinyint', 'smallint', 'mediumint']);
+	}
+
+	/**
+	 * @param $format
+	 * @return bool
+	 */
+	#[Pure] public function isFloat($format): bool
+	{
+		return in_array($format, ['float', 'double', 'decimal']);
+	}
+
+	/**
+	 * @param $format
+	 * @return bool
+	 */
+	#[Pure] public function isJson($format): bool
+	{
+		return in_array($format, ['json']);
+	}
+
+	/**
+	 * @param $format
+	 * @return bool
+	 */
+	#[Pure] public function isString($format): bool
+	{
+		return in_array($format, ['varchar', 'char', 'text', 'longtext', 'tinytext', 'mediumtext']);
+	}
+
+
+	/**
+	 * @return array
+	 * @throws
+	 */
+	public function format(): array
+	{
+		return $this->columns('Default', 'Field');
+	}
+
+
+	/**
+	 * @return mixed
+	 * @throws Exception
+	 */
+	public function getFields(): array
+	{
+		if (empty($this->_fields)) {
+			$this->structure($this->table);
+		}
+		return $this->_fields[$this->table];
+	}
+
+
+	/**
+	 * @param string $name
+	 * @return bool
+	 * @throws Exception
+	 */
+	public function hasField(string $name): bool
+	{
+		return array_key_exists($name, $this->getFields());
+	}
+
+
+	/**
+	 * @return int|string|null
+	 * @throws Exception
+	 */
+	public function getAutoIncrement(): int|string|null
+	{
+		return $this->_auto_increment[$this->table] ?? null;
+	}
+
+	/**
+	 * @return array|null|string
+	 *
+	 * @throws Exception
+	 */
+	public function getPrimaryKeys(): array|string|null
+	{
+		if (isset($this->_auto_increment[$this->table])) {
+			return $this->_auto_increment[$this->table];
+		}
+		return $this->_primary[$this->table] ?? null;
+	}
+
+	/**
+	 * @return array|null|string
+	 *
+	 * @throws Exception
+	 */
+	#[Pure] public function getFirstPrimary(): array|string|null
+	{
+		if (isset($this->_auto_increment[$this->table])) {
+			return $this->_auto_increment[$this->table];
+		}
+		if (isset($this->_primary[$this->table])) {
+			return current($this->_primary[$this->table]);
+		}
+		return null;
+	}
+
+	/**
+	 * @param $name
+	 * @param null $index
+	 * @return array
+	 * @throws Exception
+	 */
+	private function columns($name, $index = null): array
+	{
+		if (empty($index)) {
+			return array_column($this->getColumns(), $name);
+		} else {
+			return array_column($this->getColumns(), $name, $index);
+		}
+	}
+
+	/**
+	 * @return array|static
+	 * @throws Exception
+	 */
+	private function getColumns(): array|static
+	{
+		return $this->structure($this->getTable());
+	}
+
+
+	/**
+	 * @param $table
+	 * @return array|Columns
+	 * @throws Exception
+	 */
+	private function structure($table): array|static
+	{
+		if (!isset($this->columns[$table]) || empty($this->columns[$table])) {
+			$column = $this->db->createCommand(SqlBuilder::builder(null)->columns($table))->all();
+			if (empty($column)) {
+				throw new Exception("The table " . $table . " not exists.");
+			}
+			return $this->columns[$table] = $this->resolve($column, $table);
+		}
+		return $this->columns[$table];
+	}
+
+
+	/**
+	 * @param array $column
+	 * @param $table
+	 * @return array
+	 */
+	private function resolve(array $column, $table): array
+	{
+		foreach ($column as $key => $item) {
+			$this->addPrimary($item, $table);
+			$column[$key]['Type'] = $this->clean($item['Type']);
+		}
+
+		$this->_fields[$table] = array_column($column, 'Default', 'Field');
+
+		return $column;
+	}
+
+	/**
+	 * @param $item
+	 * @param $table
+	 */
+	private function addPrimary($item, $table)
+	{
+		if (!isset($this->_primary[$table])) {
+			$this->_primary[$table] = [];
+		}
+		if ($item['Key'] === 'PRI') {
+			$this->_primary[$table][] = $item['Field'];
+		}
+		$this->addIncrement($item, $table);
+	}
+
+
+	/**
+	 * @param $item
+	 * @param $table
+	 */
+	private function addIncrement($item, $table)
+	{
+		if ($item['Extra'] !== 'auto_increment') {
+			return;
+		}
+		$this->_auto_increment[$table] = $item['Field'];
+	}
+
+
+	/**
+	 * @param $type
+	 * @return string
+	 */
+	public function clean($type): string
+	{
+		if (!str_contains($type, ')')) {
+			return $type;
+		}
+		$replace = preg_replace('/\(\d+(,\d+)?\)(\s+\w+)*/', '', $type);
+		if (str_contains($replace, ' ')) {
+			$replace = explode(' ', $replace)[1];
+		}
+		return $replace;
+	}
+
+	/**
+	 * @param null $field
+	 * @return array|string|null
+	 * @throws Exception
+	 */
+	public function get_fields($field = null): array|string|null
+	{
+		$fields = $this->getAllField();
+		if (empty($field)) {
+			return $fields;
+		}
+		if (isset($fields[$field])) {
+			return strtolower($fields[$field]);
+		}
+		return null;
+	}
+
+	/**
+	 * @return array
+	 * @throws Exception
+	 */
+	public function getAllField(): array
+	{
+		return $this->columns('Type', 'Field');
+	}
+
+}

Mysql/PDO.php

@@ -0,0 +1,324 @@
+<?php
+
+namespace Database\Mysql;
+
+use Exception;
+use Kiri\Abstracts\Config;
+use Kiri\Abstracts\Logger;
+use Kiri\Kiri;
+use Kiri\Pool\StopHeartbeatCheck;
+use PDOStatement;
+use Swoole\Timer;
+
+/**
+ *
+ */
+class PDO implements StopHeartbeatCheck
+{
+
+	const DB_ERROR_MESSAGE = 'The system is busy, please try again later.';
+
+
+	private ?\PDO $pdo = null;
+
+
+	private int $_transaction = 0;
+
+
+	private int $_timer = -1;
+
+	private int $_last = 0;
+
+	public string $dbname;
+	public string $cds;
+	public string $username;
+	public string $password;
+	public string $charset;
+	public int $connect_timeout;
+	public int $read_timeout;
+
+
+	public array $attributes = [];
+
+
+	/**
+	 * @param array $config
+	 */
+	public function __construct(array $config)
+	{
+		$this->dbname = $config['dbname'];
+		$this->cds = $config['cds'];
+		$this->username = $config['username'];
+		$this->password = $config['password'];
+		$this->connect_timeout = $config['connect_timeout'] ?? 30;
+		$this->read_timeout = $config['read_timeout'] ?? 10;
+		$this->charset = $config['charset'] ?? 'utf8mb4';
+		$this->attributes = $config['attributes'] ?? [];
+	}
+
+
+	public function init()
+	{
+		$this->heartbeat_check();
+	}
+
+
+	/**
+	 * @return bool
+	 */
+	public function inTransaction(): bool
+	{
+		return $this->_transaction > 0;
+	}
+
+
+	/**
+	 *
+	 */
+	public function heartbeat_check(): void
+	{
+		if (env('state', 'start') == 'exit') {
+			return;
+		}
+		if ($this->_timer === -1) {
+			$this->_timer = Timer::tick(1000, fn() => $this->waite());
+		}
+	}
+
+
+	/**
+	 * @throws Exception
+	 */
+	private function waite(): void
+	{
+		try {
+			if (env('state', 'start') == 'exit') {
+				Kiri::getDi()->get(Logger::class)->critical('timer end');
+				$this->stopHeartbeatCheck();
+			}
+			if (time() - $this->_last > (int)Config::get('databases.pool.tick', 60)) {
+				$this->stopHeartbeatCheck();
+				$this->pdo = null;
+			}
+		} catch (\Throwable $throwable) {
+			error($throwable);
+		}
+	}
+
+
+	/**
+	 *
+	 */
+	public function stopHeartbeatCheck(): void
+	{
+		if ($this->_timer > -1) {
+			Timer::clear($this->_timer);
+		}
+		$this->_timer = -1;
+	}
+
+
+	/**
+	 *
+	 */
+	public function beginTransaction()
+	{
+		if ($this->_transaction == 0) {
+			$this->_pdo()->beginTransaction();
+		}
+		$this->_transaction++;
+	}
+
+
+	/**
+	 *
+	 */
+	public function commit()
+	{
+		if ($this->_transaction == 0) {
+			$this->_pdo()->commit();
+		}
+		$this->_transaction--;
+	}
+
+
+	/**
+	 *
+	 */
+	public function rollback()
+	{
+		if ($this->_transaction == 0) {
+			$this->_pdo()->rollBack();
+		}
+		$this->_transaction--;
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array $params
+	 * @return bool|array|null
+	 * @throws Exception
+	 */
+	public function fetchAll(string $sql, array $params = []): bool|null|array
+	{
+		$pdo = $this->queryPrev($sql, $params);
+
+		$result = $pdo->fetchAll(\PDO::FETCH_ASSOC);
+		$pdo->closeCursor();
+		return $result;
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array $params
+	 * @return bool|array|null
+	 * @throws Exception
+	 */
+	public function fetch(string $sql, array $params = []): bool|null|array
+	{
+		$pdo = $this->queryPrev($sql, $params);
+
+		$result = $pdo->fetch(\PDO::FETCH_ASSOC);
+		$pdo->closeCursor();
+		return $result;
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array $params
+	 * @return bool|array|null
+	 * @throws \Exception
+	 */
+	public function fetchColumn(string $sql, array $params = []): bool|null|array
+	{
+		$pdo = $this->queryPrev($sql, $params);
+
+		$result = $pdo->fetchColumn(\PDO::FETCH_ASSOC);
+		$pdo->closeCursor();
+		return $result;
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array $params
+	 * @return int
+	 * @throws Exception
+	 */
+	public function count(string $sql, array $params = []): int
+	{
+		$pdo = $this->queryPrev($sql, $params);
+
+		$result = $pdo->rowCount();
+		$pdo->closeCursor();
+		return $result;
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array $params
+	 * @return PDOStatement
+	 * @throws Exception
+	 */
+	private function queryPrev(string $sql, array $params = []): PDOStatement
+	{
+		$this->_last = time();
+		try {
+			if (($statement = $this->_pdo()->query($sql)) === false) {
+				throw new Exception($this->_pdo()->errorInfo()[1]);
+			}
+			return $this->bindValue($statement, $params);
+		} catch (\PDOException | \Throwable $throwable) {
+			if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
+				$this->pdo = null;
+
+				return $this->queryPrev($sql, $params);
+			}
+			throw new Exception($throwable->getMessage());
+		}
+	}
+
+
+	/**
+	 * @param PDOStatement $statement
+	 * @param array $params
+	 * @return PDOStatement
+	 */
+	private function bindValue(PDOStatement $statement, array $params = []): PDOStatement
+	{
+		if (empty($params)) return $statement;
+		foreach ($params as $key => $param) {
+			$statement->bindValue($key, $param);
+		}
+		return $statement;
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array $params
+	 * @return int
+	 * @throws Exception
+	 */
+	public function execute(string $sql, array $params = []): int
+	{
+		$this->_last = time();
+		$pdo = $this->_pdo();
+		if (!(($prepare = $pdo->prepare($sql)) instanceof PDOStatement)) {
+			throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
+		}
+		if ($prepare->execute($params) === false) {
+			throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
+		}
+		$result = (int)$pdo->lastInsertId();
+		$prepare->closeCursor();
+		if ($result == 0) {
+			return true;
+		}
+		return $result;
+	}
+
+
+	/**
+	 * @return \PDO
+	 */
+	public function _pdo(): \PDO
+	{
+		if ($this->_timer === -1) {
+			$this->heartbeat_check();
+		}
+		if (!($this->pdo instanceof \PDO)) {
+			$this->pdo = $this->newClient();
+		}
+		return $this->pdo;
+	}
+
+
+	/**
+	 * @return \PDO
+	 */
+	private function newClient(): \PDO
+	{
+		$link = new \PDO('mysql:dbname=' . $this->dbname . ';host=' . $this->cds, $this->username, $this->password, [
+			\PDO::ATTR_EMULATE_PREPARES         => false,
+			\PDO::ATTR_CASE                     => \PDO::CASE_NATURAL,
+			\PDO::ATTR_TIMEOUT                  => $this->connect_timeout,
+			\PDO::MYSQL_ATTR_USE_BUFFERED_QUERY => true,
+			\PDO::MYSQL_ATTR_INIT_COMMAND       => 'SET NAMES ' . $this->charset
+		]);
+		$link->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
+		$link->setAttribute(\PDO::ATTR_STRINGIFY_FETCHES, false);
+		$link->setAttribute(\PDO::ATTR_ORACLE_NULLS, \PDO::NULL_EMPTY_STRING);
+		if (!empty($this->attributes) && is_array($this->attributes)) {
+			foreach ($this->attributes as $key => $attribute) {
+				$link->setAttribute($key, $attribute);
+			}
+		}
+		return $link;
+	}
+
+}

Mysql/Schema.php

@@ -0,0 +1,36 @@
+<?php
+declare(strict_types=1);
+namespace Database\Mysql;
+
+
+use Exception;
+use Kiri\Abstracts\Component;
+use Database\Connection;
+
+/**
+ * Class Schema
+ * @package Database\Mysql
+ */
+class Schema extends Component
+{
+
+	/** @var ?Connection */
+	public ?Connection $db = null;
+
+	/** @var ?Columns $_column*/
+	private ?Columns $_column = null;
+
+
+	/**
+	 * @return Columns|null
+	 * @throws Exception
+	 */
+	public function getColumns(): ?Columns
+	{
+		if ($this->_column === null) {
+			$this->_column = new Columns(['db' => $this->db]);
+		}
+
+		return $this->_column;
+	}
+}