Update PhpCollection.php

izica · Oct 27, 2019 · f726517 · f726517
1 parent f82347c
commit f726517
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/PhpCollection.php b/PhpCollection.php
@@ -498,16 +498,22 @@ public function unique()
     /**
      * @return number
      */
-    public function sum($sKey = null)
+    public function sum($keyOrFunc = null)
     {
         $nResult = 0;
         if ($sKey === null) {
             foreach ($this->arCollection as $nItem) {
                 $nResult += $nItem;
             }
         } else {
-            foreach ($this->arCollection as $arItem) {
-                $nResult += $arItem[$sKey];
+            if (is_callable($keyOrFunc)) {
+                foreach ($this->arCollection as $arItem) {
+                    $nResult += $keyOrFunc($arItem);
+                }
+            } else {
+                foreach ($this->arCollection as $arItem) {
+                    $nResult += $arItem[$sKey];
+                }
             }
         }