kiri-databases/Command.php

<?php
/**
 * Created by PhpStorm.
 * User: whwyy
 * Date: 2018/3/30 0030
 * Time: 15:23
 */
declare(strict_types=1);

namespace Database;


use Exception;
use Kiri\Abstracts\Component;
use Kiri\Di\Container;
use PDO;
use Throwable;

/**
 * Class Command
 * @package Database
 */
class Command extends Component
{

    public Connection $connection;
    public ?string    $sql    = '';
    public array      $params = [];

    /**
     * @param array $params
     * @throws
     */
    public function __construct(array $params = [])
    {
        parent::__construct();
        Container::configure($this, $params);
    }


    /**
     * @return bool
     * @throws
     */
    public function incrOrDecr(): bool
    {
        return (bool)$this->_prepare();
    }


    /**
     * @return bool|array
     * @throws
     */
    public function all(): bool|array
    {
        return $this->search('fetchAll');
    }

    /**
     * @return array|bool|null
     * @throws
     */
    public function one(): array|null|bool
    {
        return $this->search('fetch');
    }

    /**
     * @return bool
     * @throws Exception
     */
    public function exists(): bool
    {
        $data = $this->search('fetch');
        if (!$data) {
            return false;
        }
        return true;
    }

    /**
     * @return mixed
     * @throws
     */
    public function fetchColumn(): mixed
    {
        return $this->search('fetchColumn');
    }

    /**
     * @return mixed
     * @throws
     */
    public function rowCount(): int
    {
        $data = $this->search('fetch');

        return !$data ? 0 : +current($data);
    }


    /**
     * @param string $method
     * @return mixed
     * @throws
     */
    protected function search(string $method): mixed
    {
        $client = $this->connection->getConnection();
        try {
            $startTime = microtime(true);
            if (($prepare = $client->prepare($this->sql)) === false) {
                throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
            }

            $prepare->execute($this->params);

            $result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC);
            $prepare->closeCursor();

            $this->connection->println($startTime, microtime(true), $this->sql, $this->params);

            return $result;
        } catch (Throwable $throwable) {
            $this->getLogger()->json_log($throwable);

            if ($this->isRefresh($throwable)) {
				return $this->search($method);
			}

            $errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);

            return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
        } finally {
            $this->connection->release($client);
        }
    }


    /**
     * @return bool
     * @throws
     */
    public function flush(): bool
    {
        return (bool)$this->_prepare();
    }


    /**
     * @return int|bool
     * @throws
     */
    private function _prepare(): int|bool
    {
        $client = $this->connection->getConnection();
        try {
            $startTime = microtime(true);
            if (($prepare = $client->prepare($this->sql)) === false) {
                throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
            }
            if ($prepare->execute($this->params) === false) {
                throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
            }
            $prepare->closeCursor();

            $result = $client->lastInsertId();

            $this->connection->println($startTime, microtime(true), $this->sql, $this->params);

            return $result == 0 ? $prepare->rowCount() : (int)$result;
        } catch (Throwable $throwable) {
            $this->getLogger()->json_log($throwable);

            if ($this->isRefresh($throwable)) {
				return $this->_prepare();
			}

            $errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);

            return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
        } finally {
            $this->connection->release($client);
        }
    }


    /**
     * @param Throwable $throwable
     * @return bool
     */
    protected function isRefresh(Throwable $throwable): bool
    {
        $message = $throwable->getMessage();
        
        // MySQL 错误处理
        if (str_contains($message, 'MySQL server has gone away')) {
            return true;
        }
        if (str_contains($message, 'Send of 14 bytes failed with errno=32 Broken pipe')) {
            return true;
        }
        if (str_contains($message, 'Lost connection to MySQL server during query')) {
            return true;
        }
        
        // PostgreSQL 错误处理
        if (str_contains($message, 'server closed the connection unexpectedly')) {
            return true;
        }
        if (str_contains($message, 'Connection refused')) {
            return true;
        }
        if (str_contains($message, 'Broken pipe')) {
            return true;
        }
        if (str_contains($message, 'connection to server was lost')) {
            return true;
        }
        
        return false;
    }


    /**
     * @return bool
     * @throws
     */
    public function delete(): bool
    {
        return (bool)$this->_prepare();
    }


    /**
     * @return int|bool
     */
    public function exec(): int|bool
    {
        return $this->_prepare();
    }

    /**
     * @param array $data
     * @return $this
     */
    public function bindValues(array $data = []): static
    {
        if (count($data) > 0) {
            $this->params = array_merge($this->params, $data);
        }
        return $this;
    }

}