From 937a0f14ffea261de8a178d29515ef21d682f1b3 Mon Sep 17 00:00:00 2001 From: YIFUH <954902919@qq.com> Date: Sat, 10 Feb 2024 12:51:52 +0000 Subject: [PATCH] =?UTF-8?q?update=20src/db/connector/Sqlite.php.=20?= =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E7=BB=93=E6=9E=9C=E9=9B=86=E4=B8=AD=E7=9A=84?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E7=B1=BB=E5=9E=8B=E5=8F=AF=E4=BB=A5=E8=87=AA?= =?UTF-8?q?=E5=8A=A8=E8=BD=AC=E6=8D=A2=E6=88=90=E5=AF=B9=E5=BA=94=E7=9A=84?= =?UTF-8?q?PHP=E7=B1=BB=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: YIFUH <954902919@qq.com> --- src/db/connector/Sqlite.php | 207 ++++++++++++++++++++++++++++++++++++ 1 file changed, 207 insertions(+) diff --git a/src/db/connector/Sqlite.php b/src/db/connector/Sqlite.php index 3e42a90..949ce3e 100644 --- a/src/db/connector/Sqlite.php +++ b/src/db/connector/Sqlite.php @@ -12,6 +12,7 @@ namespace think\db\connector; use PDO; +use PDOStatement; use think\db\PDOConnection; /** @@ -93,4 +94,210 @@ class Sqlite extends PDOConnection { return true; } +protected function createPdo($dsn, $username, $password, $params) + { + return new SqliteTypedPDO($dsn, $username, $password, $params); + } + +} +/** + * 查询结果集中的数据类型可以自动转换成对应的PHP类型 + */ +class SqliteTypedPDO extends PDO +{ + function __construct($dsn, $username = null, $password = null, $options = null) + { + parent::__construct($dsn, $username, $password, $options); + } + function prepare($query, $options = []) //: PDOStatement|bool + { + $stmt = parent::prepare($query, $options); + if ($stmt!=false && !($stmt instanceof SqlitePDOStatement)) { + return new SqlitePDOStatement($stmt); + } + return stmt; + } + function query($query, $fetchMode = null, $classname = null, $constructorArgs = null) //: PDOStatement|bool + { + if($fetchMode==PDO::FETCH_COLUMN || $fetchMode==PDO::FETCH_INTO){ + $stmt = parent::query($query, $fetchMode, $classname); + }elseif($fetchMode==PDO::FETCH_CLASS){ + $stmt = parent::query($query, $fetchMode, $classname, $constructorArgs); + }elseif($fetchMode==null){ + $stmt = parent::query($query); + } + if ($stmt!=false && !($stmt instanceof SqlitePDOStatement)) { + return new SqlitePDOStatement($stmt); + } + + return $stmt; + } +} + +class SqlitePDOStatement extends PDOStatement +{ + private PDOStatement $stmt; + private $fetchMode; + function __construct($stmt) + { + $this->stmt = $stmt; + //readonly + // $this->queryString = $stmt->queryString; + } + /** + * @var string + */ + // public $queryString; + + function bindColumn($column, &$var, $type = PDO::PARAM_STR, $maxLength = 0, $driverOptions = null): bool + { + return $this->stmt->bindColumn($column, $var, $type, $maxLength, $driverOptions); + } + + function bindParam($param, &$var, $type = PDO::PARAM_STR, $maxLength = 0, $driverOptions = null): bool + { + return $this->stmt->bindParam($param, $var, $type, $maxLength, $driverOptions); + } + + function bindValue($param, $value, $type = PDO::PARAM_STR): bool + { + return $this->stmt->bindValue($param, $value, $type); + } + + function closeCursor(): bool + { + return $this->stmt->closeCursor(); + } + + function columnCount(): int + { + return $this->stmt->columnCount(); + } + + function debugDumpParams() //: bool|null + { + return $this->stmt->debugDumpParams(); + } + + function errorCode() //: string|null + { + return $this->stmt->errorCode(); + } + + function errorInfo(): array + { + return $this->stmt->errorInfo(); + } + + function execute($params = null): bool + { + return $this->stmt->execute($params); + } + + function fetch($mode = PDO::FETCH_ASSOC, $cursorOrientation = PDO::FETCH_ORI_NEXT, $cursorOffset = 0): mixed + { + $result = $this->stmt->fetch($mode, $cursorOrientation, $cursorOffset); + $columnCount = $this->stmt->columnCount(); + if ($this->fetchMode == PDO::FETCH_ASSOC || $mode == PDO::FETCH_ASSOC) { //TODO:暂时支持一种 + for ($i = 0; $i < $columnCount; $i++) { + $meta = $this->stmt->getColumnMeta($i); + if ($meta['native_type'] == 'integer' || (!empty($meta['sqlite:decl_type']) && $meta['sqlite:decl_type'] == 'INTEGER')) { + $result[$meta['name']] = intval($result[$meta['name']]); + } + } + } + return $result; + } + + + function fetchAll($mode = PDO::FETCH_ASSOC, $class = null, $constructorArgs = null): array + { + if ($mode == PDO::FETCH_COLUMN || $mode == PDO::FETCH_FUNC) { + $result = $this->stmt->fetchAll($mode, $class); + } elseif ($mode == PDO::FETCH_CLASS) { + $result = $this->stmt->fetchAll($mode, $class, $constructorArgs); + } else { + $result = $this->stmt->fetchAll($mode); + } + // $tableName = null; + // $pattern = '/\bSELECT\b.*?\bFROM\b\s+["`\'`]?(\w+)["`\'`]?/i'; + // if (preg_match($pattern, trim($this->stmt->queryString), $matches)) { + // // 匹配成功，$matches[1] 包含提取出的表名 + // $tableName = $matches[1]; + // } + // var_dump($this->stmt->queryString); + // var_dump($result); + // if(!empty($tableName)){ + // // die; + // } + + if ($this->fetchMode == PDO::FETCH_ASSOC || $mode == PDO::FETCH_ASSOC) { //TODO:暂时支持一种 + $columnCount = $this->stmt->columnCount(); + for ($i = 0; $i < $columnCount; $i++) { + $meta = $this->stmt->getColumnMeta($i); + if ($meta['native_type'] == 'integer' || (!empty($meta['sqlite:decl_type']) && $meta['sqlite:decl_type'] == 'INTEGER')) { + foreach ($result as &$r) { + $r[$meta['name']] = intval($r[$meta['name']]); + } + } + } + } + // echo("===================\n"); + return $result; + } + + function fetchColumn($column = 0) + { + $result = $this->stmt->fetchColumn($column); + if ($result === false) + return $result; + $meta = $this->stmt->getColumnMeta(0); + if ($meta['native_type'] == 'integer' || (!empty($meta['sqlite:decl_type']) && $meta['sqlite:decl_type'] == 'INTEGER')) { + $result = intval($result); + } + return $result; + } + + function fetchObject($class = "stdClass", $constructorArgs = []) //: bool|object + { + return $this->stmt->fetchObject($class, $constructorArgs); + } + + function getAttribute($name): mixed + { + return $this->stmt->getAttribute($name); + } + + + function getColumnMeta($column) //: array|bool + { + return $this->stmt->getColumnMeta($column); + } + + function getIterator(): \Iterator + { + return $this->stmt->getIterator(); + } + + function nextRowset(): bool + { + return $this->stmt->nextRowset(); + } + + function rowCount(): int + { + return $this->stmt->rowCount(); + } + + function setAttribute($attribute, $value): bool + { + return $this->stmt->setAttribute($attribute, $value); + } + + + function setFetchMode($mode = null, $class = null, $constructorArgs = null): bool + { + $this->fetchMode = $mode; + return $this->stmt->setFetchMode($mode, $class, $constructorArgs); + } } -- Gitee