Catch guzzle exception instead

georgeboot · Aug 6, 2021 · 957ce4f · 957ce4f
1 parent 90a7d6d
commit 957ce4f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/ConnectionRepository.php b/src/ConnectionRepository.php
@@ -4,6 +4,7 @@
 
 use Aws\ApiGatewayManagementApi\ApiGatewayManagementApiClient;
 use Aws\ApiGatewayManagementApi\Exception\ApiGatewayManagementApiException;
+use GuzzleHttp\Exception\ClientException;
 
 class ConnectionRepository
 {
@@ -26,15 +27,15 @@ public function sendMessage(string $connectionId, string $data): void
                 'ConnectionId' => $connectionId,
                 'Data' => $data,
             ]);
-        } catch (ApiGatewayManagementApiException $e) {
+        } catch (ClientException $guzzleClientException) {
             // GoneException: The connection with the provided id no longer exists.
-            if ($e->getAwsErrorCode() === 'Gone') {
+            if ($guzzleClientException->getResponse()->getStatusCode() === 410) {
                 $this->subscriptionRepository->clearConnection($connectionId);
 
                 return;
             }
 
-            throw $e;
+            throw $guzzleClientException;
         }
     }
 }
diff --git a/tests/HandlerTest.php b/tests/HandlerTest.php
@@ -9,6 +9,7 @@
 use Georgeboot\LaravelEchoApiGateway\ConnectionRepository;
 use Georgeboot\LaravelEchoApiGateway\Handler;
 use Georgeboot\LaravelEchoApiGateway\SubscriptionRepository;
+use GuzzleHttp\Psr7\Response;
 use Mockery\Mock;
 use Psr\Http\Message\RequestInterface;
 
@@ -82,7 +83,7 @@
     $mock = new MockHandler();
 
     $mock->append(function (CommandInterface $cmd, RequestInterface $req) {
-        return new  ApiGatewayManagementApiException('', $cmd, ['code' => 'Gone']);
+        return new \GuzzleHttp\Exception\ClientException('', $req, new Response(410));
     });
 
     /** @var SubscriptionRepository */