Added A::pickRandoms()

meszaros-lajos-gyorgy · Jul 30, 2021 · a62593f · a62593f
1 parent f8f46b9
commit a62593f
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/A.php b/src/A.php
@@ -335,9 +335,35 @@ public static function join(string $separator, array $data): string
     // A::pickRandom([1, 2, 3, 4, 5]) -> 3
     public static function pickRandom(array $data)
     {
+        if (self::isEmpty($data)) {
+            return null;
+        }
+
         return $data[rand(0, self::length($data) - 1)];
     }
 
+    // A::pickRandoms(2, [1, 2, 3, 4, 5]) -> [5, 2]
+    public static function pickRandoms(int $amount = 1, array $data): array
+    {
+        if (self::isEmpty($data) || $amount <= 0) {
+            return [];
+        }
+
+        if ($amount > self::length($data)) {
+            $amount = self::length($data);
+        }
+
+        $remaining = $data;
+        $selected = [];
+        for ($i = 0; $i < $amount; $i++) {
+            $randomItem = self::pickRandom($remaining, 1);
+            $remaining = self::without($randomItem, $remaining);
+            $selected[] = $randomItem;
+        }
+
+        return $selected;
+    }
+
     // A::concat([1, 2], 3, [4, 5]) -> [1, 2, 3, 4, 5]
     public static function concat(...$args): array
     {