Refactor getCollection and hasCollection

mongodb · Feb 17, 2020 · ae88c82 · ae88c82
1 parent 2dcfa5b
commit ae88c82
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Jenssegers/Mongodb/Schema/Builder.php b/src/Jenssegers/Mongodb/Schema/Builder.php
@@ -40,13 +40,13 @@ public function hasCollection($name)
     {
         $db = $this->connection->getMongoDB();
 
-        $collections = $db->listCollections([
+        $collections = iterator_to_array($db->listCollections([
             'filter' => [
                 'name' => $name,
             ],
-        ]);
+        ]), false);
 
-        return $collections->count() ? true : false;
+        return count($collections) ? true : false;
     }
 
     /**
@@ -143,13 +143,13 @@ public function getCollection($name)
     {
         $db = $this->connection->getMongoDB();
 
-        $collections = $db->listCollections([
+        $collections = iterator_to_array($db->listCollections([
             'filter' => [
                 'name' => $name,
             ],
-        ]);
+        ]), false);
 
-        return $collections->count() ? (($collections = iterator_to_array($collections) ) ? current($collections) : false) : false;
+        return count($collections) ? current($collections) : false;
     }
 
     /**