Update NaryNode and the way nodes are added.

loophp · Jun 12, 2019 · 43da343 · 43da343
1 parent 5712df1
commit 43da343
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/Node/NaryNode.php b/src/Node/NaryNode.php
@@ -43,10 +43,14 @@ public function __construct(int $capacity = 0, NodeInterface $parent = null, Tra
      */
     public function add(NodeInterface ...$nodes): NodeInterface
     {
-        foreach ($nodes as $node) {
-            $capacity = $this->capacity();
+        $capacity = $this->capacity();
+
+        if (0 === $capacity) {
+            return parent::add(...$nodes);
+        }
 
-            if (0 === $capacity || ($this->degree() < $capacity)) {
+        foreach ($nodes as $node) {
+            if ($this->degree() < $capacity) {
                 parent::add($node);
 
                 continue;