as2252258/kiri-core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

改名

Browse Source
This commit is contained in:
Mr·x
2021-08-16 18:11:18 +08:00
parent 01f6491421
commit 863ef72c00
1 changed files with 9 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
System/Pool/Connection.php
+6 -29
View File
@@ -8,11 +8,9 @@ use Exception;
use HttpServer\Http\Context; use HttpServer\Http\Context;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Kiri\Kiri; use Kiri\Kiri;
use PDO;
use Swoole\Coroutine;
use Swoole\Error; use Swoole\Error;
use Swoole\Runtime;
use Throwable; use Throwable;
use Database\Mysql\PDO;
/** /**
* Class Connection * Class Connection
@@ -44,13 +42,8 @@ class Connection extends Component
public function beginTransaction($coroutineName) public function beginTransaction($coroutineName)
{ {
$coroutineName = $this->name('Mysql:' . $coroutineName, true); $coroutineName = $this->name('Mysql:' . $coroutineName, true);
if (!Context::hasContext('begin_' . $coroutineName)) {
Context::setContext('begin_' . $coroutineName, 1);
}
Context::increment('begin_' . $coroutineName);
$connection = Context::getContext($coroutineName); $connection = Context::getContext($coroutineName);
if ($connection instanceof PDO && !$connection->inTransaction()) { if ($connection instanceof PDO) {
$connection->beginTransaction(); $connection->beginTransaction();
} }
} }
@@ -62,13 +55,11 @@ class Connection extends Component
public function commit($coroutineName) public function commit($coroutineName)
{ {
$coroutineName = $this->name('Mysql:' . $coroutineName, true); $coroutineName = $this->name('Mysql:' . $coroutineName, true);
if (Context::decrement('begin_' . $coroutineName) == 1) {
$connection = Context::getContext($coroutineName); $connection = Context::getContext($coroutineName);
if ($connection instanceof PDO && $connection->inTransaction()) { if ($connection instanceof PDO) {
$connection->commit(); $connection->commit();
} }
} }
}
/** /**
@@ -78,14 +69,11 @@ class Connection extends Component
public function rollback($coroutineName) public function rollback($coroutineName)
{ {
$coroutineName = $this->name('Mysql:' . $coroutineName, true); $coroutineName = $this->name('Mysql:' . $coroutineName, true);
if (Context::decrement('begin_' . $coroutineName) == 1) { $connection = Context::getContext($coroutineName);
if (($connection = Context::getContext($coroutineName)) instanceof PDO) { if ($connection instanceof PDO) {
if ($connection->inTransaction()) {
$connection->rollBack(); $connection->rollBack();
} }
} }
}
}
/** /**
@@ -117,18 +105,7 @@ class Connection extends Component
public function create($coroutineName, $config): Closure public function create($coroutineName, $config): Closure
{ {
return static function () use ($coroutineName, $config) { return static function () use ($coroutineName, $config) {
$cds = 'mysql:dbname=' . $config['database'] . ';host=' . $config['cds']; return new PDO($config['database'], $config['cds'], $config['username'], $config['password'], $config['charset'] ?? 'utf8mb4');
$link = new PDO($cds, $config['username'], $config['password'], [
PDO::ATTR_EMULATE_PREPARES => false,
PDO::ATTR_CASE => PDO::CASE_NATURAL,
PDO::ATTR_TIMEOUT => 60,
PDO::MYSQL_ATTR_USE_BUFFERED_QUERY => true,
PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . ($config['charset'] ?? 'utf8mb4')
]);
$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);
return $link;
}; };
} }