mirror of
https://github.com/torrentpier/torrentpier
synced 2025-08-22 22:33:55 -07:00
Db exceptions, query with binding
This commit is contained in:
parent
f4a70a1769
commit
c3eb61d449
4 changed files with 143 additions and 25 deletions
73
src/Db.php
73
src/Db.php
|
@ -28,6 +28,8 @@ namespace TorrentPier;
|
|||
use \PDO;
|
||||
use \PDOException;
|
||||
use \PDOStatement;
|
||||
use TorrentPier\Db\Exception;
|
||||
use TorrentPier\Db\IntegrityViolationException;
|
||||
|
||||
class Db extends PDO
|
||||
{
|
||||
|
@ -120,39 +122,78 @@ class Db extends PDO
|
|||
$this->setAttribute(static::ATTR_STATEMENT_CLASS, ['\\TorrentPier\\Db\\Statement', [$this]]);
|
||||
}
|
||||
|
||||
public function prepare($statement, $options = [])
|
||||
public function prepare($statement, /** @noinspection PhpSignatureMismatchDuringInheritanceInspection */
|
||||
$options = [])
|
||||
{
|
||||
if (!$this->stat) {
|
||||
return parent::prepare($statement);
|
||||
}
|
||||
try {
|
||||
if ($this->stat) {
|
||||
$t = microtime(true);
|
||||
$ret = parent::prepare($statement);
|
||||
}
|
||||
return parent::prepare($statement, $options);
|
||||
} catch (PDOException $e) {
|
||||
throw new Exception($e);
|
||||
} finally {
|
||||
if (isset($t)) {
|
||||
$this->sqlTimeTotal += microtime(true) - $t;
|
||||
return $ret;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/** @noinspection PhpSignatureMismatchDuringInheritanceInspection */
|
||||
/**
|
||||
* @param string $statement
|
||||
* @param array ...$args
|
||||
* @return PDOStatement
|
||||
*/
|
||||
public function query($statement, ...$args)
|
||||
{
|
||||
if (!$this->stat) {
|
||||
return parent::query($statement, ...$args);
|
||||
}
|
||||
try {
|
||||
if ($this->stat) {
|
||||
$t = microtime(true);
|
||||
$ret = parent::query($statement, ...$args);
|
||||
}
|
||||
if (func_num_args() > 1) {
|
||||
$input = func_get_arg(1);
|
||||
if (is_array($input)) {
|
||||
$stmt = parent::prepare($statement);
|
||||
if (func_num_args() > 2) {
|
||||
$stmt->setFetchMode(...array_slice(func_get_args(), 2));
|
||||
}
|
||||
$stmt->execute($input);
|
||||
return $stmt;
|
||||
}
|
||||
}
|
||||
return parent::query($statement, ...$args);
|
||||
} catch (PDOException $e) {
|
||||
if ($e->getCode() == '23000') {
|
||||
throw new IntegrityViolationException($e);
|
||||
}
|
||||
throw new Exception($e);
|
||||
} finally {
|
||||
if (isset($t)) {
|
||||
$this->sqlTimeTotal += microtime(true) - $t;
|
||||
$this->numQueries++;
|
||||
return $ret;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public function exec($statement)
|
||||
{
|
||||
if (!$this->stat) {
|
||||
return parent::exec($statement);
|
||||
}
|
||||
try {
|
||||
if ($this->stat) {
|
||||
$t = microtime(true);
|
||||
$ret = parent::exec($statement);
|
||||
}
|
||||
return parent::exec($statement);
|
||||
} catch (PDOException $e) {
|
||||
if ($e->getCode() == '23000') {
|
||||
throw new IntegrityViolationException($e);
|
||||
}
|
||||
throw new Exception($e);
|
||||
} finally {
|
||||
if (isset($t)) {
|
||||
$this->sqlTimeTotal += microtime(true) - $t;
|
||||
$this->numQueries++;
|
||||
return $ret;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/** @deprecated
|
||||
|
|
38
src/Db/Exception.php
Normal file
38
src/Db/Exception.php
Normal file
|
@ -0,0 +1,38 @@
|
|||
<?php
|
||||
/**
|
||||
* MIT License
|
||||
*
|
||||
* Copyright (c) 2005-2017 TorrentPier
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in all
|
||||
* copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
* SOFTWARE.
|
||||
*/
|
||||
|
||||
namespace TorrentPier\Db;
|
||||
|
||||
use \PDOException;
|
||||
|
||||
class Exception extends PDOException
|
||||
{
|
||||
public function __construct(PDOException $exception)
|
||||
{
|
||||
parent::__construct($exception->getMessage(), 0, $exception);
|
||||
$this->code = $exception->getCode();
|
||||
$this->errorInfo = $exception->errorInfo;
|
||||
}
|
||||
}
|
30
src/Db/IntegrityViolationException.php
Normal file
30
src/Db/IntegrityViolationException.php
Normal file
|
@ -0,0 +1,30 @@
|
|||
<?php
|
||||
/**
|
||||
* MIT License
|
||||
*
|
||||
* Copyright (c) 2005-2017 TorrentPier
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in all
|
||||
* copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
* SOFTWARE.
|
||||
*/
|
||||
|
||||
namespace TorrentPier\Db;
|
||||
|
||||
class IntegrityViolationException extends Exception
|
||||
{
|
||||
}
|
|
@ -26,6 +26,7 @@
|
|||
namespace TorrentPier\Db;
|
||||
|
||||
use \PDOStatement;
|
||||
use \PDOException;
|
||||
use TorrentPier\Db;
|
||||
|
||||
class Statement extends PDOStatement
|
||||
|
@ -39,13 +40,21 @@ class Statement extends PDOStatement
|
|||
|
||||
public function execute($input_parameters = null)
|
||||
{
|
||||
if (!$this->db->stat) {
|
||||
return parent::execute($input_parameters);
|
||||
}
|
||||
try {
|
||||
if ($this->db->stat) {
|
||||
$t = microtime(true);
|
||||
$ret = parent::execute($input_parameters);
|
||||
}
|
||||
return parent::execute($input_parameters);
|
||||
} catch (PDOException $e) {
|
||||
if ($e->getCode() == '23000') {
|
||||
throw new IntegrityViolationException($e);
|
||||
}
|
||||
throw new Exception($e);
|
||||
} finally {
|
||||
if (isset($t)) {
|
||||
$this->db->sqlTimeTotal += microtime(true) - $t;
|
||||
$this->db->numQueries++;
|
||||
return $ret;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue