detect server gone away in failed statement too

ofbeaton · Aug 5, 2016 · c17d9a2 · c17d9a2
1 parent 4a36abd
commit c17d9a2
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/src/Command/PingCommand.php b/src/Command/PingCommand.php
@@ -270,12 +270,19 @@ protected function ping(InputInterface $input, OutputInterface $output)
         error_reporting($errorLevel);
 
         if ($this->checkStmt === false) {
-            $this->writeReply(
-                'check failed statement: ['.$this->dbh->errorCode().'] '
-                .implode(' ', $this->dbh->errorInfo()),
-                $input,
-                $output
-            );
+            $errorInfo = $this->dbh->errorInfo();
+            // server has gone away
+            if (isset($errorInfo[1]) === true && $errorInfo[1] === 2006) {
+                $this->connected = false;
+                $this->writeReply('connection lost', $input, $output);
+            } else {
+                $this->writeReply(
+                    'check failed statement: [' . $this->dbh->errorCode() . '] '
+                    . implode(' ', $errorInfo),
+                    $input,
+                    $output
+                );
+            }
             return false;
         }