handles HTTP client and server exceptions

src/BatchHandler.php

@@ -4,7 +4,6 @@
 
 use Dew\Tablestore\Concerns\InteractsWithRequest;
 use Dew\Tablestore\Exceptions\BatchHandlerException;
-use Google\Protobuf\Internal\Message;
 use Protos\BatchGetRowRequest;
 use Protos\BatchGetRowResponse;
 use Protos\BatchWriteRowRequest;
@@ -229,14 +228,4 @@ protected function toChangesRequest(BatchBuilder $builder): RowInBatchWriteRowRe
             ->setCondition($this->toCondition($builder))
             ->setReturnContent($this->toReturnContent($builder));
     }
-
-    /**
-     * Communicate with Tablestore with the given message.
-     */
-    protected function send(string $endpoint, Message $message): string
-    {
-        return $this->tablestore->send($endpoint, $message)
-            ->getBody()
-            ->getContents();
-    }
 }

src/Concerns/InteractsWithRequest.php

@@ -5,14 +5,37 @@
 use Dew\Tablestore\BatchBuilder;
 use Dew\Tablestore\Builder;
 use Dew\Tablestore\ConditionFilter;
+use Dew\Tablestore\Exceptions\TablestoreException;
 use Dew\Tablestore\PaginationFilter;
+use Dew\Tablestore\Tablestore;
+use Google\Protobuf\Internal\Message;
+use GuzzleHttp\Exception\BadResponseException;
 use Protos\Condition;
 use Protos\Filter;
 use Protos\ReturnContent;
 use RuntimeException;
 
 trait InteractsWithRequest
 {
+    /**
+     * The Tablestore client.
+     */
+    protected Tablestore $tablestore;
+
+    /**
+     * Communicate with Tablestore with the given message.
+     */
+    protected function send(string $endpoint, Message $message): string
+    {
+        try {
+            return $this->tablestore()->send($endpoint, $message)
+                ->getBody()
+                ->getContents();
+        } catch (BadResponseException $e) {
+            throw TablestoreException::fromResponse($e->getResponse());
+        }
+    }
+
     /**
      * Build a condition Protobuf message.
      */
@@ -76,4 +99,12 @@ public function buildFilter(BatchBuilder|Builder $builder): Filter
 
         throw new RuntimeException('Missing filter data to build with.');
     }
+
+    /**
+     * The Tablestore client.
+     */
+    public function tablestore(): Tablestore
+    {
+        return $this->tablestore;
+    }
 }

src/Exceptions/TablestoreException.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace Dew\Tablestore\Exceptions;
+
+use Exception;
+use Protos\Error;
+use Psr\Http\Message\ResponseInterface;
+
+class TablestoreException extends Exception
+{
+    /**
+     * Create a Tablestore exception.
+     */
+    public function __construct(
+        protected Error $e
+    ) {
+        parent::__construct($e->getMessage());
+    }
+
+    /**
+     * Create a Tablestore exception from error response.
+     */
+    public static function fromResponse(ResponseInterface $response): self
+    {
+        $error = new Error;
+        $error->mergeFromString($response->getBody()->getContents());
+
+        return new self($error);
+    }
+
+    /**
+     * Get the error message.
+     */
+    public function getError(): Error
+    {
+        return $this->e;
+    }
+}

src/Handler.php

@@ -4,7 +4,6 @@
 
 use Dew\Tablestore\Concerns\InteractsWithRequest;
 use Dew\Tablestore\Responses\RowDecodableResponse;
-use Google\Protobuf\Internal\Message;
 use Protos\DeleteRowRequest;
 use Protos\DeleteRowResponse;
 use Protos\GetRowRequest;
@@ -121,22 +120,4 @@ public function getRow(Builder $builder): RowDecodableResponse
 
         return new RowDecodableResponse($response);
     }
-
-    /**
-     * Communicate with Tablestore with the given message.
-     */
-    protected function send(string $endpoint, Message $message): string
-    {
-        return $this->tablestore->send($endpoint, $message)
-            ->getBody()
-            ->getContents();
-    }
-
-    /**
-     * The tablestore client.
-     */
-    public function tablestore(): Tablestore
-    {
-        return $this->tablestore;
-    }
 }

tests/HandlerTest.php

@@ -2,12 +2,17 @@
 
 use Dew\Tablestore\Attribute;
 use Dew\Tablestore\Builder;
+use Dew\Tablestore\Exceptions\TablestoreException;
 use Dew\Tablestore\Handler;
 use Dew\Tablestore\PrimaryKey;
 use Dew\Tablestore\Tablestore;
+use GuzzleHttp\Exception\ClientException;
+use GuzzleHttp\Exception\ServerException;
 use GuzzleHttp\Psr7\Response;
+use Protos\Error;
 use Protos\Filter;
 use Protos\FilterType;
+use Psr\Http\Message\RequestInterface;
 
 test('filter build determination primary keys', function () {
     $handler = new Handler(Mockery::mock(Tablestore::class));
@@ -169,3 +174,29 @@
         ->maxVersions(2)
         ->get();
 });
+
+test('handles http client error', function () {
+    $error = (new Error)->setCode('foo')->setMessage('bar');
+    $exception = new ClientException('',
+        Mockery::mock(RequestInterface::class),
+        new Response(body: $error->serializeToString())
+    );
+    $mockedTs = Mockery::mock(Tablestore::class);
+    $mockedTs->expects()->send(Mockery::any(), Mockery::any())->andThrows($exception);
+    $handler = new Handler($mockedTs);
+    $builder = (new Builder)->setTable('test')->handlerUsing($handler);
+    expect(fn () => $builder->get())->toThrow(TablestoreException::class, 'bar');
+});
+
+test('handles http server error', function () {
+    $error = (new Error)->setCode('foo')->setMessage('bar');
+    $exception = new ServerException('',
+        Mockery::mock(RequestInterface::class),
+        new Response(body: $error->serializeToString())
+    );
+    $mockedTs = Mockery::mock(Tablestore::class);
+    $mockedTs->expects()->send(Mockery::any(), Mockery::any())->andThrows($exception);
+    $handler = new Handler($mockedTs);
+    $builder = (new Builder)->setTable('test')->handlerUsing($handler);
+    expect(fn () => $builder->get())->toThrow(TablestoreException::class, 'bar');
+});