RandomIterableAggregate::getIterator()   A
last analyzed

Complexity

Conditions 1
Paths 1

Size

Total Lines 5
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 3
CRAP Score 1

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 1
eloc 2
c 1
b 0
f 0
nc 1
nop 0
dl 0
loc 5
ccs 3
cts 3
cp 1
crap 1
rs 10
1
<?php
2
3
/**
4
 * For the full copyright and license information, please view
5
 * the LICENSE file that was distributed with this source code.
6
 */
7
8
declare(strict_types=1);
9
10
namespace loophp\iterators;
11
12
use Generator;
13
use IteratorAggregate;
14
15
/**
16
 * @template TKey
17
 * @template T
18
 *
19
 * @implements IteratorAggregate<TKey, T>
20
 */
21
final class RandomIterableAggregate implements IteratorAggregate
22
{
23
    /**
24
     * @var PackIterableAggregate<TKey, T>
25
     */
26
    private PackIterableAggregate $iteratorAggregate;
27
28
    /**
29
     * @param iterable<TKey, T> $iterable
30
     */
31 2
    public function __construct(iterable $iterable, private int $seed = 0)
32
    {
33 2
        $this->iteratorAggregate = new PackIterableAggregate($iterable);
34
    }
35
36
    /**
37
     * @return Generator<TKey, T>
38
     */
39 2
    public function getIterator(): Generator
40
    {
41 2
        mt_srand($this->seed);
42
43 2
        yield from $this->randomize($this->iteratorAggregate, $this->seed);
44
    }
45
46
    /**
47
     * @param iterable<int, array{0: TKey, 1: T}> $iterable
48
     *
49
     * @return Generator<TKey, T>
50
     */
51 2
    private function randomize(iterable $iterable, int $seed): Generator
52
    {
53 2
        $queue = [];
54
55 2
        foreach (new UnpackIterableAggregate($iterable) as $key => $value) {
56 2
            if (0 === mt_rand(0, $seed)) {
57 2
                yield $key => $value;
58
59 2
                continue;
60
            }
61
62 1
            $queue[] = [$key, $value];
63
        }
64
65 2
        if ([] !== $queue) {
66 1
            yield from $this->randomize($queue, $seed);
67
        }
68
    }
69
}
70