Merge pull request #778 from rsolyanik/fix_totalcount_promise

Made it possible for totalCount to be a promise

Author: murtukov

src/Relay/Connection/ConnectionInterface.php

@@ -4,6 +4,8 @@
 
 namespace Overblog\GraphQLBundle\Relay\Connection;
 
+use GraphQL\Executor\Promise\Promise;
+
 interface ConnectionInterface
 {
     /**
@@ -33,14 +35,16 @@ public function getPageInfo(): ?PageInfoInterface;
     public function setPageInfo(PageInfoInterface $pageInfo): void;
 
     /**
-     * Get the total count.
+     * Get the total count or promise that returns the total count.
      *
-     * @return int
+     * @return int|Promise|null
      */
-    public function getTotalCount(): ?int;
+    public function getTotalCount();
 
     /**
-     * Set the total count.
+     * Set the total count or promise that returns the total count.
+     *
+     * @param int|Promise $totalCount
      */
-    public function setTotalCount(int $totalCount): void;
+    public function setTotalCount($totalCount): void;
 }

src/Relay/Connection/Output/Connection.php

@@ -4,6 +4,7 @@
 
 namespace Overblog\GraphQLBundle\Relay\Connection\Output;
 
+use GraphQL\Executor\Promise\Promise;
 use Overblog\GraphQLBundle\Relay\Connection\ConnectionInterface;
 use Overblog\GraphQLBundle\Relay\Connection\EdgeInterface;
 use Overblog\GraphQLBundle\Relay\Connection\PageInfoInterface;
@@ -16,7 +17,9 @@ class Connection implements ConnectionInterface
     protected array $edges;
 
     protected ?PageInfoInterface $pageInfo;
-    protected ?int $totalCount = null;
+
+    /** @var int|Promise|null Total count or promise that returns the total count */
+    protected $totalCount = null;
 
     public function __construct(array $edges = [], PageInfoInterface $pageInfo = null)
     {
@@ -59,15 +62,15 @@ public function setPageInfo(PageInfoInterface $pageInfo): void
     /**
      * {@inheritdoc}
      */
-    public function getTotalCount(): ?int
+    public function getTotalCount()
     {
         return $this->totalCount;
     }
 
     /**
      * {@inheritdoc}
      */
-    public function setTotalCount(int $totalCount): void
+    public function setTotalCount($totalCount): void
     {
         $this->totalCount = $totalCount;
     }