fluent-booking / vendor / wpfluent / framework / src / WPFluent / Container / RewindableGenerator.php

RewindableGenerator.php in Fluent Booking – The Ultimate Appointments Scheduling, Events Booking, Events Calendar Solution 2.1.1, at vendor/wpfluent/framework/src/WPFluent/Container/RewindableGenerator.php

63 lines 1.1 KB

Raw Download Zip

1	<?php
2
3	namespace FluentBooking\Framework\Container;
4
5	use Countable;
6	use IteratorAggregate;
7
8	class RewindableGenerator implements Countable, IteratorAggregate
9	{
10	/**
11	* The generator callback.
12	*
13	* @var callable
14	*/
15	protected $generator;
16
17	/**
18	* The number of tagged services.
19	*
20	* @var callable\|int
21	*/
22	protected $count;
23
24	/**
25	* Create a new generator instance.
26	*
27	* @param callable $generator
28	* @param callable\|int $count
29	* @return void
30	*/
31	public function __construct(callable $generator, $count)
32	{
33	$this->count = $count;
34	$this->generator = $generator;
35	}
36
37	/**
38	* Get an iterator from the generator.
39	*
40	* @return mixed
41	*/
42	#[\ReturnTypeWillChange]
43	public function getIterator()
44	{
45	return ($this->generator)();
46	}
47
48	/**
49	* Get the total number of tagged services.
50	*
51	* @return int
52	*/
53	#[\ReturnTypeWillChange]
54	public function count()
55	{
56	if (is_callable($count = $this->count)) {
57	$this->count = $count();
58	}
59
60	return $this->count;
61	}
62	}
63