Skip to content

Instantly share code, notes, and snippets.

@extraordinaire
Last active June 27, 2023 11:12
Show Gist options
  • Save extraordinaire/4135119 to your computer and use it in GitHub Desktop.
Save extraordinaire/4135119 to your computer and use it in GitHub Desktop.
Reconnectable PDO
<?php
class ReconnectingPDO
{
protected $dsn, $username, $password, $pdo, $driver_options;
public function __construct($dsn, $username = "", $password = "", $driver_options = array())
{
$this->dsn = $dsn;
$this->username = $username;
$this->password = $password;
$this->driver_options = $driver_options;
}
public function __call($name, array $arguments)
{
try {
$this->connection()->query("SHOW STATUS;")->execute();
} catch(\PDOException $e) {
if($e->getCode() != 'HY000' || !stristr($e->getMessage(), 'server has gone away')) {
throw $e;
}
$this->reconnect();
}
return call_user_func_array(array($this->connection(), $name), $arguments);
}
protected function connection()
{
return $this->pdo instanceof \PDO ? $this->pdo : $this->connect();
}
public function connect()
{
$this->pdo = new PDO($this->dsn, $this->username, $this->password, (array) $this->driver_options);
$this->pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
return $this->pdo;
}
public function reconnect()
{
$this->pdo = null;
return $this->connect();
}
}
@wallacio
Copy link

Nice idea, but this seems to interfere with PDO's lastInsertId() method, in that it always returns '0'.
Is checking for a connection perhaps triggering an auto commit?

@titpetric
Copy link

@wallacio: the class is not transaction aware, in case of a disconnect in the middle of a transaction, you should write your own class which handles this scenario. This one is mainly useful for single statement / non transactional database access. IMHO, as soon as you call ->start() (or equivalent of sql 'start commit'), the reconnect option should be disabled, and you should start handling your exceptions in a specific way to what you are doing.

@andreas83
Copy link

i use it for my little opensource project. thank you !

https://github.com/andreas83/SocialNetwork

@shaobaojie
Copy link

这是一个愚蠢的做法 PHP Warning: PDO::query(): MySQL server has gone away 抛出的是 warning 而不是 exception

@sjoorm
Copy link

sjoorm commented Sep 6, 2017

@jurchiks
@schaobaojie
see $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);

@Maarco73
Copy link

Maarco73 commented Oct 1, 2018

It's a good approch, but if you use FOUND_ROWS() in combination with SQL_CALC_FOUND_ROWS it will always return the number of rows from the "SHOW STATUS" in line 17... is anyone where who has a smart solution for this issue?

@wenpeng
Copy link

wenpeng commented Apr 29, 2019

You can add "@mixin PDO" to the class docblock that IDE can support auto-complete.

@mindplay-dk
Copy link

I hate to spoil your fun, but I see a number of problems with this.

Besides breaking lastInsertId() and transactionality, as pointed out above, you've got the overhead of an extra round-trip to the database for every query you execute - hopefully this is small for SHOW STATUS, but there are blocking wait-states involved for every SQL query executed, so this could substantially impact total time to process a request with many queries.

Secondly, a try/catch block around the SHOW STATUS query only guarantees error handling for that query - the connecting could just as well fail when you execute the user-supplied query next, and you have no error-handling for that. Presumably, what you were trying to achieve, is a connection that automatically reconnects and retries the query on connection failure? This might do that in some cases, but it might not.

What you probably want is the try/catch around the user-supplied query itself, with an automatic reconnect and retry - and you can get rid of the SHOW STATUS query entirely.

As @titpetric pointed out though, reconnecting/retrying only makes sense in the first place if you're not in a transaction - since, if you lose the connection, you also lose the transaction state.

This could be addressed, but it's a somewhat more complicated issue - you would need to buffer all executed queries within a transaction, clear the buffer on commit (or rollback) and, in the catch-handler, retry the all of the buffered queries, including restarting the transaction after a reconnect.

If you were hoping for something that can work in a long-lived CLI app, this needs a lot of work still...

@ninze
Copy link

ninze commented Sep 26, 2019

Adding to what @mindplay-dk wrote, "buffering queries within a transaction" might not always work as well. Some queries might use data from previous queries, such as last_insert_id's, which will become different when the transaction is replayed. Besides that, another process could change the state of the database and make some of the buffered queries fail, resulting in missing data, duplicate keys etc. When you lose the transaction, you also lose all the read/write locks for that transaction.

Generally mysql errors such as "gone away" or "deadlock found" are a part of life and sometimes are difficult to avoid. Your code should be ready to replay entire transaction with all the logic that created sql statements. You can write a simple transaction repeater class, which will repeat an anonymous function up to several times in case of recoverable errors.

@sven100 just put "use PDO;" at the top of the file and you won't need the backslash.

@mindplay-dk
Copy link

In a long-running app, likely the most appropriate thing to do, is don't even attempt to "replay" the query - instead, just throw an exception immediately after reconnecting. Assuming you have an error-handler that can gracefully fail and handle the exception, the current request will fail, but the long-running app itself can continue running and handle the next request.

It's no ideal, but probably only happens if the DB server is down - this at least ensures the long-running app is up and handling requests again as soon as the DB server is back.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment