|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467 |
- <?php
- // +----------------------------------------------------------------------
- // | ThinkPHP [ WE CAN DO IT JUST THINK ]
- // +----------------------------------------------------------------------
- // | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved.
- // +----------------------------------------------------------------------
- // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
- // +----------------------------------------------------------------------
- // | Author: zhangyajun <448901948@qq.com>
- // +----------------------------------------------------------------------
-
- namespace think;
-
- use ArrayAccess;
- use ArrayIterator;
- use Countable;
- use IteratorAggregate;
- use JsonSerializable;
-
- class Collection implements ArrayAccess, Countable, IteratorAggregate, JsonSerializable
- {
- /**
- * @var array 数据
- */
- protected $items = [];
-
- /**
- * Collection constructor.
- * @access public
- * @param array $items 数据
- */
- public function __construct($items = [])
- {
- $this->items = $this->convertToArray($items);
- }
-
- /**
- * 创建 Collection 实例
- * @access public
- * @param array $items 数据
- * @return static
- */
- public static function make($items = [])
- {
- return new static($items);
- }
-
- /**
- * 判断数据是否为空
- * @access public
- * @return bool
- */
- public function isEmpty()
- {
- return empty($this->items);
- }
-
- /**
- * 将数据转成数组
- * @access public
- * @return array
- */
- public function toArray()
- {
- return array_map(function ($value) {
- return ($value instanceof Model || $value instanceof self) ?
- $value->toArray() :
- $value;
- }, $this->items);
- }
-
- /**
- * 获取全部的数据
- * @access public
- * @return array
- */
- public function all()
- {
- return $this->items;
- }
-
- /**
- * 交换数组中的键和值
- * @access public
- * @return static
- */
- public function flip()
- {
- return new static(array_flip($this->items));
- }
-
- /**
- * 返回数组中所有的键名组成的新 Collection 实例
- * @access public
- * @return static
- */
- public function keys()
- {
- return new static(array_keys($this->items));
- }
-
- /**
- * 返回数组中所有的值组成的新 Collection 实例
- * @access public
- * @return static
- */
- public function values()
- {
- return new static(array_values($this->items));
- }
-
- /**
- * 合并数组并返回一个新的 Collection 实例
- * @access public
- * @param mixed $items 新的数据
- * @return static
- */
- public function merge($items)
- {
- return new static(array_merge($this->items, $this->convertToArray($items)));
- }
-
- /**
- * 比较数组,返回差集生成的新 Collection 实例
- * @access public
- * @param mixed $items 做比较的数据
- * @return static
- */
- public function diff($items)
- {
- return new static(array_diff($this->items, $this->convertToArray($items)));
- }
-
- /**
- * 比较数组,返回交集组成的 Collection 新实例
- * @access public
- * @param mixed $items 比较数据
- * @return static
- */
- public function intersect($items)
- {
- return new static(array_intersect($this->items, $this->convertToArray($items)));
- }
-
- /**
- * 返回并删除数据中的的最后一个元素(出栈)
- * @access public
- * @return mixed
- */
- public function pop()
- {
- return array_pop($this->items);
- }
-
- /**
- * 返回并删除数据中首个元素
- * @access public
- * @return mixed
- */
- public function shift()
- {
- return array_shift($this->items);
- }
-
- /**
- * 在数组开头插入一个元素
- * @access public
- * @param mixed $value 值
- * @param mixed $key 键名
- * @return void
- */
- public function unshift($value, $key = null)
- {
- if (is_null($key)) {
- array_unshift($this->items, $value);
- } else {
- $this->items = [$key => $value] + $this->items;
- }
- }
-
- /**
- * 在数组结尾插入一个元素
- * @access public
- * @param mixed $value 值
- * @param mixed $key 键名
- * @return void
- */
- public function push($value, $key = null)
- {
- if (is_null($key)) {
- $this->items[] = $value;
- } else {
- $this->items[$key] = $value;
- }
- }
-
- /**
- * 通过使用用户自定义函数,以字符串返回数组
- * @access public
- * @param callable $callback 回调函数
- * @param mixed $initial 初始值
- * @return mixed
- */
- public function reduce(callable $callback, $initial = null)
- {
- return array_reduce($this->items, $callback, $initial);
- }
-
- /**
- * 以相反的顺序创建一个新的 Collection 实例
- * @access public
- * @return static
- */
- public function reverse()
- {
- return new static(array_reverse($this->items));
- }
-
- /**
- * 把数据分割为新的数组块
- * @access public
- * @param int $size 分隔长度
- * @param bool $preserveKeys 是否保持原数据索引
- * @return static
- */
- public function chunk($size, $preserveKeys = false)
- {
- $chunks = [];
-
- foreach (array_chunk($this->items, $size, $preserveKeys) as $chunk) {
- $chunks[] = new static($chunk);
- }
-
- return new static($chunks);
- }
-
- /**
- * 给数据中的每个元素执行回调
- * @access public
- * @param callable $callback 回调函数
- * @return $this
- */
- public function each(callable $callback)
- {
- foreach ($this->items as $key => $item) {
- $result = $callback($item, $key);
-
- if (false === $result) {
- break;
- }
-
- if (!is_object($item)) {
- $this->items[$key] = $result;
- }
- }
-
- return $this;
- }
-
- /**
- * 用回调函数过滤数据中的元素
- * @access public
- * @param callable|null $callback 回调函数
- * @return static
- */
- public function filter(callable $callback = null)
- {
- return new static(array_filter($this->items, $callback ?: null));
- }
-
- /**
- * 返回数据中指定的一列
- * @access public
- * @param mixed $columnKey 键名
- * @param null $indexKey 作为索引值的列
- * @return array
- */
- public function column($columnKey, $indexKey = null)
- {
- if (function_exists('array_column')) {
- return array_column($this->items, $columnKey, $indexKey);
- }
-
- $result = [];
- foreach ($this->items as $row) {
- $key = $value = null;
- $keySet = $valueSet = false;
-
- if (null !== $indexKey && array_key_exists($indexKey, $row)) {
- $key = (string) $row[$indexKey];
- $keySet = true;
- }
-
- if (null === $columnKey) {
- $valueSet = true;
- $value = $row;
- } elseif (is_array($row) && array_key_exists($columnKey, $row)) {
- $valueSet = true;
- $value = $row[$columnKey];
- }
-
- if ($valueSet) {
- if ($keySet) {
- $result[$key] = $value;
- } else {
- $result[] = $value;
- }
- }
- }
-
- return $result;
- }
-
- /**
- * 对数据排序,并返回排序后的数据组成的新 Collection 实例
- * @access public
- * @param callable|null $callback 回调函数
- * @return static
- */
- public function sort(callable $callback = null)
- {
- $items = $this->items;
- $callback = $callback ?: function ($a, $b) {
- return $a == $b ? 0 : (($a < $b) ? -1 : 1);
- };
-
- uasort($items, $callback);
- return new static($items);
- }
-
- /**
- * 将数据打乱后组成新的 Collection 实例
- * @access public
- * @return static
- */
- public function shuffle()
- {
- $items = $this->items;
-
- shuffle($items);
- return new static($items);
- }
-
- /**
- * 截取数据并返回新的 Collection 实例
- * @access public
- * @param int $offset 起始位置
- * @param int $length 截取长度
- * @param bool $preserveKeys 是否保持原先的键名
- * @return static
- */
- public function slice($offset, $length = null, $preserveKeys = false)
- {
- return new static(array_slice($this->items, $offset, $length, $preserveKeys));
- }
-
- /**
- * 指定的键是否存在
- * @access public
- * @param mixed $offset 键名
- * @return bool
- */
- public function offsetExists($offset)
- {
- return array_key_exists($offset, $this->items);
- }
-
- /**
- * 获取指定键对应的值
- * @access public
- * @param mixed $offset 键名
- * @return mixed
- */
- public function offsetGet($offset)
- {
- return $this->items[$offset];
- }
-
- /**
- * 设置键值
- * @access public
- * @param mixed $offset 键名
- * @param mixed $value 值
- * @return void
- */
- public function offsetSet($offset, $value)
- {
- if (is_null($offset)) {
- $this->items[] = $value;
- } else {
- $this->items[$offset] = $value;
- }
- }
-
- /**
- * 删除指定键值
- * @access public
- * @param mixed $offset 键名
- * @return void
- */
- public function offsetUnset($offset)
- {
- unset($this->items[$offset]);
- }
-
- /**
- * 统计数据的个数
- * @access public
- * @return int
- */
- public function count()
- {
- return count($this->items);
- }
-
- /**
- * 获取数据的迭代器
- * @access public
- * @return ArrayIterator
- */
- public function getIterator()
- {
- return new ArrayIterator($this->items);
- }
-
- /**
- * 将数据反序列化成数组
- * @access public
- * @return array
- */
- public function jsonSerialize()
- {
- return $this->toArray();
- }
-
- /**
- * 转换当前数据集为 JSON 字符串
- * @access public
- * @param integer $options json 参数
- * @return string
- */
- public function toJson($options = JSON_UNESCAPED_UNICODE)
- {
- return json_encode($this->toArray(), $options);
- }
-
- /**
- * 将数据转换成字符串
- * @access public
- * @return string
- */
- public function __toString()
- {
- return $this->toJson();
- }
-
- /**
- * 将数据转换成数组
- * @access protected
- * @param mixed $items 数据
- * @return array
- */
- protected function convertToArray($items)
- {
- return $items instanceof self ? $items->all() : (array) $items;
- }
- }
|