Merge pull request #7 from trybeapp/insert-result

feat: store and retrieve result of insert

Author: willtj

src/Collection.php

@@ -6,10 +6,18 @@
 
 class Collection extends BaseCollection
 {
-    public function addToIndex()
+    /**
+     * @var array
+     */
+    protected $addToIndexResult;
+
+    /**
+     * @return bool
+     */
+    public function addToIndex(): bool
     {
         if ($this->isEmpty()) {
-            return;
+            return true;
         }
 
         $instance = $this->first();
@@ -25,8 +33,18 @@ public function addToIndex()
 
         $success = $query->insert($docs->all());
 
+        $this->addToIndexResult = $query->getQuery()->getInsertResult();
+
         unset($docs);
 
         return $success;
     }
+
+    /**
+     * @return array|null
+     */
+    public function getAddToIndexResult(): ?array
+    {
+        return $this->addToIndexResult;
+    }
 }

src/QueryBuilder.php

@@ -38,6 +38,8 @@ class QueryBuilder extends BaseBuilder
 
     protected $routing;
 
+    protected $insertResult;
+
     /**
      * All of the supported clause operators.
      *
@@ -731,9 +733,9 @@ public function cursor()
      */
     public function insert(array $values): bool
     {
-        $result = $this->connection->insert($this->grammar->compileInsert($this, $values));
+        $this->insertResult = $this->connection->insert($this->grammar->compileInsert($this, $values));
 
-        return empty($result['errors']);
+        return empty($this->insertResult['errors']);
     }
 
     /**
@@ -767,6 +769,11 @@ public function delete($id = null): bool
         return !empty($result['found']);
     }
 
+    /**
+     * @param mixed $method
+     * @param mixed $parameters
+     * @return void
+     */
     public function __call($method, $parameters)
     {
         if (Str::startsWith($method, 'filterWhere')) {
@@ -776,6 +783,14 @@ public function __call($method, $parameters)
         return parent::__call($method, $parameters);
     }
 
+    /**
+     * @return array
+     */
+    public function getInsertResult(): array
+    {
+        return array_column($this->insertResult['items'], 'index');
+    }
+
     /**
      * @return bool
      */