467 lines
9.4 KiB
PHP
467 lines
9.4 KiB
PHP
|
<?php
|
||
|
|
||
|
namespace Doctrine\Common\Collections;
|
||
|
|
||
|
use ArrayIterator;
|
||
|
use Closure;
|
||
|
use Doctrine\Common\Collections\Expr\ClosureExpressionVisitor;
|
||
|
use ReturnTypeWillChange;
|
||
|
use Traversable;
|
||
|
|
||
|
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;
|
||
|
|
||
|
use const ARRAY_FILTER_USE_BOTH;
|
||
|
|
||
|
/**
|
||
|
* 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>
|
||
|
* @psalm-consistent-constructor
|
||
|
*/
|
||
|
class ArrayCollection implements Collection, Selectable
|
||
|
{
|
||
|
/**
|
||
|
* An array containing the entries of this collection.
|
||
|
*
|
||
|
* @psalm-var array<TKey,T>
|
||
|
* @var mixed[]
|
||
|
*/
|
||
|
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.
|
||
|
* @psalm-param array<K,V> $elements
|
||
|
*
|
||
|
* @return static
|
||
|
* @psalm-return static<K,V>
|
||
|
*
|
||
|
* @psalm-template K of array-key
|
||
|
* @psalm-template V
|
||
|
*/
|
||
|
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.
|
||
|
*
|
||
|
* @param TKey $offset
|
||
|
*
|
||
|
* @return bool
|
||
|
*/
|
||
|
#[ReturnTypeWillChange]
|
||
|
public function offsetExists($offset)
|
||
|
{
|
||
|
return $this->containsKey($offset);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Required by interface ArrayAccess.
|
||
|
*
|
||
|
* @param TKey $offset
|
||
|
*
|
||
|
* @return mixed
|
||
|
*/
|
||
|
#[ReturnTypeWillChange]
|
||
|
public function offsetGet($offset)
|
||
|
{
|
||
|
return $this->get($offset);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Required by interface ArrayAccess.
|
||
|
*
|
||
|
* @param TKey|null $offset
|
||
|
* @param T $value
|
||
|
*
|
||
|
* @return void
|
||
|
*/
|
||
|
#[ReturnTypeWillChange]
|
||
|
public function offsetSet($offset, $value)
|
||
|
{
|
||
|
if (! isset($offset)) {
|
||
|
$this->add($value);
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
$this->set($offset, $value);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Required by interface ArrayAccess.
|
||
|
*
|
||
|
* @param TKey $offset
|
||
|
*
|
||
|
* @return void
|
||
|
*/
|
||
|
#[ReturnTypeWillChange]
|
||
|
public function offsetUnset($offset)
|
||
|
{
|
||
|
$this->remove($offset);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*/
|
||
|
public function containsKey($key)
|
||
|
{
|
||
|
return isset($this->elements[$key]) || array_key_exists($key, $this->elements);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*
|
||
|
* @template TMaybeContained
|
||
|
*/
|
||
|
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}
|
||
|
*
|
||
|
* @psalm-param TMaybeContained $element
|
||
|
*
|
||
|
* @psalm-return (TMaybeContained is T ? TKey|false : false)
|
||
|
*
|
||
|
* @template TMaybeContained
|
||
|
*/
|
||
|
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}
|
||
|
*
|
||
|
* @return int
|
||
|
*/
|
||
|
#[ReturnTypeWillChange]
|
||
|
public function count()
|
||
|
{
|
||
|
return count($this->elements);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*/
|
||
|
public function set($key, $value)
|
||
|
{
|
||
|
$this->elements[$key] = $value;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*
|
||
|
* @psalm-suppress InvalidPropertyAssignmentValue
|
||
|
*
|
||
|
* This breaks assumptions about the template type, but it would
|
||
|
* be a backwards-incompatible change to remove this method
|
||
|
*/
|
||
|
public function add($element)
|
||
|
{
|
||
|
$this->elements[] = $element;
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*/
|
||
|
public function isEmpty()
|
||
|
{
|
||
|
return empty($this->elements);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*
|
||
|
* @return Traversable<int|string, mixed>
|
||
|
* @psalm-return Traversable<TKey,T>
|
||
|
*/
|
||
|
#[ReturnTypeWillChange]
|
||
|
public function getIterator()
|
||
|
{
|
||
|
return new ArrayIterator($this->elements);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*
|
||
|
* @psalm-param Closure(T):U $func
|
||
|
*
|
||
|
* @return static
|
||
|
* @psalm-return static<TKey, U>
|
||
|
*
|
||
|
* @psalm-template U
|
||
|
*/
|
||
|
public function map(Closure $func)
|
||
|
{
|
||
|
return $this->createFrom(array_map($func, $this->elements));
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritDoc}
|
||
|
*
|
||
|
* @return static
|
||
|
* @psalm-return static<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);
|
||
|
}
|
||
|
}
|