PDO之MySql持久化自动重连导致内存溢出

前言

最近项目需要一个常驻内存的脚本来执行队列程序，脚本完成后发现Mysql自动重连部分存在内存溢出,导致运行一段时间后，会超出PHP内存限制退出

排查

发现脚本存在内存溢出后排查了一遍代码,基本确认内存溢出在Mysql查询部分.

如果不确认也可以把不必要的模块去掉,从最基本的业务开始测,如果没有内存泄露,则继续增加模块,基本很快就能定位到内存泄露处的代码

解决

我使用的是PHP7.2,所以mysql查询部分使用了PDO,但是PDO并没有提供关闭连接的方法,只是说将连接connection赋值为Null,PHP便会进行垃圾回收，销毁连接.下面是PHP官方手册中的一段话

连接数据成功后，返回一个 PDO 类的实例给脚本，此连接在 PDO 对象的生存周期中保持活动。要想关闭连接，需要销毁对象以确保所有剩余到它的引用都被删除，可以赋一个 NULL 值给对象变量。如果不明确地这么做，PHP 在脚本结束时会自动关闭连接

然而事情没有这么简单,赋值NULL后PHP并没有将PDO实例回收,依然存在.因为还需要将PDOStatement同样赋值为NULL,否则PHP还是不会去执行垃圾回收.

修改后的代码

namespace app\library;

use \PDO;

class Mysql
{
    /** @var string 域名 */
    private $host = '127.0.0.1';

    /** @var int 端口 */
    private $port = 3306;

    /** @var string 数据库 */
    private $dataBase = 'test';

    /** @var string 用户名 */
    private $userName = 'root';

    /** @var string 密码 */
    private $password = 'root';

    /** @var PDO 数据库连接 */
    public $connection;

    /** @var Mysql 单例 */
    private static $instance;

    /** @var \PDOStatement */
    protected $PDOStatement;

    private function __construct()
    {
    }

    /**
     * 连接数据库
     */
    protected function connection():void
    {
        $this->connection = new \pdo(
            "mysql:host={$this->host}:{$this->port};dbname={$this->dataBase}",
            $this->userName,
            $this->password,
            [PDO::ATTR_ERRMODE => PDO::ERRMODE_WARNING]
        );
        //设置PDO抛出异常
        $this->connection->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
    }

    /**
     * 获取数据库连接
     * @return PDO
     */
    private function getConnection(): PDO
    {
        if ($this->connection instanceof PDO) {
            return $this->connection;
        }

        $this->connection();

        return $this->connection;
    }

    /**
     * 获取实例
     * @return Mysql
     */
    public static function getInstance()
    {
        if (is_null(self::$instance)) {
            self::$instance = new Mysql();
        }
        return self::$instance;
    }

    /**
     * 重置链接
     */
    protected function resetConnection():void
    {
        $this->connection   = null;
        //将此赋值操作屏蔽会造成内存溢出
        $this->PDOStatement = null;
    }

    /**
     * 重连
     * @return PDO
     */
    protected function reconnection(): PDO
    {
        $this->resetConnection();

        return $this->getConnection();
    }

    /**
     * 查询
     * @param string $sql   sql文本
     * @return mixed
     */
    public function query(string $sql)
    {
        try {
            $this->PDOStatement =  @$this->getConnection()->query($sql);
            return $this->PDOStatement->fetch();
        } catch (\PDOException $e) {
            if (!$this->isMysqlAlive()) {
                $this->PDOStatement =  $this->reconnection()->query($sql);
                return $this->PDOStatement->fetch();
            }
            throw new $e;
        }
    }

    /**
     * mysql连接连接是否有效
     * @return bool
     */
    protected function isMysqlAlive(): bool
    {
        $errorInfo = $this->connection->errorInfo();

        if ($errorInfo[1] != 2006 or $errorInfo[1] != 2013) {
            return false;
        }

        return true;
    }

    /**
     * 关闭数据库连接
     */
    public function close():void
    {
        $this->connection = null;
        $this->PDOStatement = null;
    }

    public function __destruct():void
    {
        $this->connection = null;
        $this->PDOStatement = null;
    }

    /**
     * 禁止clone
     * @throws \Exception
     */
    public function __clone()
    {
        throw new \Exception('deny clone');
    }
}

WIKI

• PHP PDO Exception + Warning on MySQL Has Gone Away?

• PHP官方手册,详细见用户笔记第一条