123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425 |
- <?php
- namespace Doctrine\Common\Collections;
- use ArrayIterator;
- use Closure;
- use Doctrine\Common\Collections\Expr\ClosureExpressionVisitor;
- use const ARRAY_FILTER_USE_BOTH;
- use function array_filter;
- use function array_key_exists;
- use function array_keys;
- use function array_map;
- use function array_reverse;
- use function array_search;
- use function array_slice;
- use function array_values;
- use function count;
- use function current;
- use function end;
- use function in_array;
- use function key;
- use function next;
- use function reset;
- use function spl_object_hash;
- use function uasort;
- /**
- * An ArrayCollection is a Collection implementation that wraps a regular PHP array.
- *
- * Warning: Using (un-)serialize() on a collection is not a supported use-case
- * and may break when we change the internals in the future. If you need to
- * serialize a collection use {@link toArray()} and reconstruct the collection
- * manually.
- *
- * @psalm-template TKey of array-key
- * @psalm-template T
- * @template-implements Collection<TKey,T>
- * @template-implements Selectable<TKey,T>
- */
- class ArrayCollection implements Collection, Selectable
- {
- /**
- * An array containing the entries of this collection.
- *
- * @psalm-var array<TKey,T>
- * @var array
- */
- private $elements;
- /**
- * Initializes a new ArrayCollection.
- *
- * @param array $elements
- *
- * @psalm-param array<TKey,T> $elements
- */
- public function __construct(array $elements = [])
- {
- $this->elements = $elements;
- }
- /**
- * {@inheritDoc}
- */
- public function toArray()
- {
- return $this->elements;
- }
- /**
- * {@inheritDoc}
- */
- public function first()
- {
- return reset($this->elements);
- }
- /**
- * Creates a new instance from the specified elements.
- *
- * This method is provided for derived classes to specify how a new
- * instance should be created when constructor semantics have changed.
- *
- * @param array $elements Elements.
- *
- * @return static
- *
- * @psalm-param array<TKey,T> $elements
- * @psalm-return ArrayCollection<TKey,T>
- */
- protected function createFrom(array $elements)
- {
- return new static($elements);
- }
- /**
- * {@inheritDoc}
- */
- public function last()
- {
- return end($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function key()
- {
- return key($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function next()
- {
- return next($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function current()
- {
- return current($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function remove($key)
- {
- if (! isset($this->elements[$key]) && ! array_key_exists($key, $this->elements)) {
- return null;
- }
- $removed = $this->elements[$key];
- unset($this->elements[$key]);
- return $removed;
- }
- /**
- * {@inheritDoc}
- */
- public function removeElement($element)
- {
- $key = array_search($element, $this->elements, true);
- if ($key === false) {
- return false;
- }
- unset($this->elements[$key]);
- return true;
- }
- /**
- * Required by interface ArrayAccess.
- *
- * {@inheritDoc}
- */
- public function offsetExists($offset)
- {
- return $this->containsKey($offset);
- }
- /**
- * Required by interface ArrayAccess.
- *
- * {@inheritDoc}
- */
- public function offsetGet($offset)
- {
- return $this->get($offset);
- }
- /**
- * Required by interface ArrayAccess.
- *
- * {@inheritDoc}
- */
- public function offsetSet($offset, $value)
- {
- if (! isset($offset)) {
- $this->add($value);
- return;
- }
- $this->set($offset, $value);
- }
- /**
- * Required by interface ArrayAccess.
- *
- * {@inheritDoc}
- */
- public function offsetUnset($offset)
- {
- $this->remove($offset);
- }
- /**
- * {@inheritDoc}
- */
- public function containsKey($key)
- {
- return isset($this->elements[$key]) || array_key_exists($key, $this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function contains($element)
- {
- return in_array($element, $this->elements, true);
- }
- /**
- * {@inheritDoc}
- */
- public function exists(Closure $p)
- {
- foreach ($this->elements as $key => $element) {
- if ($p($key, $element)) {
- return true;
- }
- }
- return false;
- }
- /**
- * {@inheritDoc}
- */
- public function indexOf($element)
- {
- return array_search($element, $this->elements, true);
- }
- /**
- * {@inheritDoc}
- */
- public function get($key)
- {
- return $this->elements[$key] ?? null;
- }
- /**
- * {@inheritDoc}
- */
- public function getKeys()
- {
- return array_keys($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function getValues()
- {
- return array_values($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function count()
- {
- return count($this->elements);
- }
- /**
- * {@inheritDoc}
- */
- public function set($key, $value)
- {
- $this->elements[$key] = $value;
- }
- /**
- * {@inheritDoc}
- */
- public function add($element)
- {
- $this->elements[] = $element;
- return true;
- }
- /**
- * {@inheritDoc}
- */
- public function isEmpty()
- {
- return empty($this->elements);
- }
- /**
- * Required by interface IteratorAggregate.
- *
- * {@inheritDoc}
- */
- public function getIterator()
- {
- return new ArrayIterator($this->elements);
- }
- /**
- * {@inheritDoc}
- *
- * @return static
- */
- public function map(Closure $func)
- {
- return $this->createFrom(array_map($func, $this->elements));
- }
- /**
- * {@inheritDoc}
- *
- * @return static
- *
- * @psalm-return ArrayCollection<TKey,T>
- */
- public function filter(Closure $p)
- {
- return $this->createFrom(array_filter($this->elements, $p, ARRAY_FILTER_USE_BOTH));
- }
- /**
- * {@inheritDoc}
- */
- public function forAll(Closure $p)
- {
- foreach ($this->elements as $key => $element) {
- if (! $p($key, $element)) {
- return false;
- }
- }
- return true;
- }
- /**
- * {@inheritDoc}
- */
- public function partition(Closure $p)
- {
- $matches = $noMatches = [];
- foreach ($this->elements as $key => $element) {
- if ($p($key, $element)) {
- $matches[$key] = $element;
- } else {
- $noMatches[$key] = $element;
- }
- }
- return [$this->createFrom($matches), $this->createFrom($noMatches)];
- }
- /**
- * Returns a string representation of this object.
- *
- * @return string
- */
- public function __toString()
- {
- return self::class . '@' . spl_object_hash($this);
- }
- /**
- * {@inheritDoc}
- */
- public function clear()
- {
- $this->elements = [];
- }
- /**
- * {@inheritDoc}
- */
- public function slice($offset, $length = null)
- {
- return array_slice($this->elements, $offset, $length, true);
- }
- /**
- * {@inheritDoc}
- */
- public function matching(Criteria $criteria)
- {
- $expr = $criteria->getWhereExpression();
- $filtered = $this->elements;
- if ($expr) {
- $visitor = new ClosureExpressionVisitor();
- $filter = $visitor->dispatch($expr);
- $filtered = array_filter($filtered, $filter);
- }
- $orderings = $criteria->getOrderings();
- if ($orderings) {
- $next = null;
- foreach (array_reverse($orderings) as $field => $ordering) {
- $next = ClosureExpressionVisitor::sortByField($field, $ordering === Criteria::DESC ? -1 : 1, $next);
- }
- uasort($filtered, $next);
- }
- $offset = $criteria->getFirstResult();
- $length = $criteria->getMaxResults();
- if ($offset || $length) {
- $filtered = array_slice($filtered, (int) $offset, $length);
- }
- return $this->createFrom($filtered);
- }
- }
|