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

变更

Browse Source
This commit is contained in:
向林
2023-04-02 19:53:43 +08:00
parent 685a0d72b2
commit f554b4c81e
1 changed files with 92 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Command.php
+92 -47
View File
@@ -57,38 +57,62 @@ class Command extends Component
{ {
return $this->_execute(); return $this->_execute();
} }
/** /**
* @return array|null * @return array|null
* @throws Exception * @throws Exception
*/ */
public function all(): ?array public function all(): ?array
{ {
[$pdo, $statement] = $this->search(); try {
[$pdo, $statement] = $this->search();
$statement->execute($this->params); $statement->execute($this->params);
$data = $statement->fetchAll(PDO::FETCH_ASSOC); $data = $statement->fetchAll(PDO::FETCH_ASSOC);
$this->db->release($pdo); $this->db->release($pdo);
return $data; return $data;
} catch (\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->db->restore();
return $this->one();
}
if (isset($pdo)) {
$this->db->release($pdo);
}
return $this->printErrorMessage($throwable);
}
} }
/** /**
* @return array|null * @return array|null
* @throws Exception * @throws Exception
*/ */
public function one(): ?array public function one(): ?array
{ {
[$pdo, $statement] = $this->search(); try {
[$pdo, $statement] = $this->search();
$statement->execute($this->params); $statement->execute($this->params);
$data = $statement->fetch(PDO::FETCH_ASSOC); $data = $statement->fetch(PDO::FETCH_ASSOC);
$this->db->release($pdo); $this->db->release($pdo);
return $data; return $data;
} catch (\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->db->restore();
return $this->one();
}
if (isset($pdo)) {
$this->db->release($pdo);
}
return $this->printErrorMessage($throwable);
}
} }
/** /**
@@ -97,30 +121,62 @@ class Command extends Component
*/ */
public function fetchColumn(): mixed public function fetchColumn(): mixed
{ {
[$pdo, $statement] = $this->search(); try {
[$pdo, $statement] = $this->search();
$statement->execute($this->params); $statement->execute($this->params);
$data = $statement->fetchColumn(PDO::FETCH_ASSOC); $data = $statement->fetchColumn(PDO::FETCH_ASSOC);
$this->db->release($pdo); $this->db->release($pdo);
return $data; return $data;
} catch (\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->db->restore();
return $this->fetchColumn();
}
if (isset($pdo)) {
$this->db->release($pdo);
}
return $this->printErrorMessage($throwable);
}
} }
/** /**
* @return ?int * @return int|null
* @throws Exception * @throws Exception
*/ */
public function rowCount(): ?int public function rowCount(): ?int
{ {
[$pdo, $statement] = $this->search(); try {
[$pdo, $statement] = $this->search();
$statement->execute($this->params); $statement->execute($this->params);
$data = $statement->rowCount(); $data = $statement->rowCount();
$this->db->release($pdo); $this->db->release($pdo);
return $data; return $data;
} catch (\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->db->restore();
return $this->rowCount();
}
if (isset($pdo)) {
$this->db->release($pdo);
}
return $this->printErrorMessage($throwable);
}
}
private function printErrorMessage(\Throwable $throwable): mixed
{
$this->logger->addError($this->sql . '. error: ' . $throwable->getMessage(), 'mysql');
return null;
} }
/** /**
@@ -165,7 +221,7 @@ class Command extends Component
return $this->logger->addError($this->sql . '. error: ' . $throwable->getMessage(), 'mysql'); return $this->logger->addError($this->sql . '. error: ' . $throwable->getMessage(), 'mysql');
} }
} }
/** /**
* @return array<PDO, PDOStatement>|bool * @return array<PDO, PDOStatement>|bool
* @throws Exception * @throws Exception
@@ -173,22 +229,11 @@ class Command extends Component
private function search(): bool|array private function search(): bool|array
{ {
$pdo = $this->db->getSlaveClient(); $pdo = $this->db->getSlaveClient();
try {
if (($statement = $pdo->prepare($this->sql)) === false) {
throw new Exception($pdo->errorInfo()[1]);
}
return [$pdo, $statement];
} catch (\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->db->restore();
return $this->search(); if (($statement = $pdo->prepare($this->sql)) === false) {
} throw new Exception($pdo->errorInfo()[1]);
$this->db->release($pdo);
return $this->logger->addError($this->sql . '. error: ' . $throwable->getMessage(), 'mysql');
} }
return [$pdo, $statement];
} }