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');

    }

}

巴特西

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

前言

排查

解决

修改后的代码

WIKI

最新文章

热门文章