Merge pull request #31 from mathroc/master

fix code against a subtype
daryl-software · Nov 21, 2015 · 3ae8468 · 3ae8468
2 parents c8ab6ae + 5f97dd2
commit 3ae8468
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/RetryStrategy/MysqlRetryStrategy.php b/src/RetryStrategy/MysqlRetryStrategy.php
@@ -33,6 +33,8 @@ private function errorCodeStrategies() {
 
     private function errorCode(DBALException $exception)
     {
-        return $exception->getErrorCode();
+        if ($exception instanceof DriverException) {
+            return $exception->getErrorCode();
+        }
     }
 }
diff --git a/src/RetryStrategy/PostgreSQLRetryStrategy.php b/src/RetryStrategy/PostgreSQLRetryStrategy.php
@@ -3,7 +3,6 @@
 namespace Ez\DbLinker\RetryStrategy;
 
 use Doctrine\DBAL\DBALException;
-use Doctrine\DBAL\Exception\DriverException;
 use Ez\DbLinker\RetryStrategy as RetryStrategyInterface;
 
 class PostgreSQLRetryStrategy implements RetryStrategyInterface