Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.
 
 
 
 
 
 

468 rindas
11 KiB

  1. <?php
  2. // +----------------------------------------------------------------------
  3. // | ThinkPHP [ WE CAN DO IT JUST THINK ]
  4. // +----------------------------------------------------------------------
  5. // | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved.
  6. // +----------------------------------------------------------------------
  7. // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
  8. // +----------------------------------------------------------------------
  9. // | Author: zhangyajun <448901948@qq.com>
  10. // +----------------------------------------------------------------------
  11. namespace think;
  12. use ArrayAccess;
  13. use ArrayIterator;
  14. use Countable;
  15. use IteratorAggregate;
  16. use JsonSerializable;
  17. class Collection implements ArrayAccess, Countable, IteratorAggregate, JsonSerializable
  18. {
  19. /**
  20. * @var array 数据
  21. */
  22. protected $items = [];
  23. /**
  24. * Collection constructor.
  25. * @access public
  26. * @param array $items 数据
  27. */
  28. public function __construct($items = [])
  29. {
  30. $this->items = $this->convertToArray($items);
  31. }
  32. /**
  33. * 创建 Collection 实例
  34. * @access public
  35. * @param array $items 数据
  36. * @return static
  37. */
  38. public static function make($items = [])
  39. {
  40. return new static($items);
  41. }
  42. /**
  43. * 判断数据是否为空
  44. * @access public
  45. * @return bool
  46. */
  47. public function isEmpty()
  48. {
  49. return empty($this->items);
  50. }
  51. /**
  52. * 将数据转成数组
  53. * @access public
  54. * @return array
  55. */
  56. public function toArray()
  57. {
  58. return array_map(function ($value) {
  59. return ($value instanceof Model || $value instanceof self) ?
  60. $value->toArray() :
  61. $value;
  62. }, $this->items);
  63. }
  64. /**
  65. * 获取全部的数据
  66. * @access public
  67. * @return array
  68. */
  69. public function all()
  70. {
  71. return $this->items;
  72. }
  73. /**
  74. * 交换数组中的键和值
  75. * @access public
  76. * @return static
  77. */
  78. public function flip()
  79. {
  80. return new static(array_flip($this->items));
  81. }
  82. /**
  83. * 返回数组中所有的键名组成的新 Collection 实例
  84. * @access public
  85. * @return static
  86. */
  87. public function keys()
  88. {
  89. return new static(array_keys($this->items));
  90. }
  91. /**
  92. * 返回数组中所有的值组成的新 Collection 实例
  93. * @access public
  94. * @return static
  95. */
  96. public function values()
  97. {
  98. return new static(array_values($this->items));
  99. }
  100. /**
  101. * 合并数组并返回一个新的 Collection 实例
  102. * @access public
  103. * @param mixed $items 新的数据
  104. * @return static
  105. */
  106. public function merge($items)
  107. {
  108. return new static(array_merge($this->items, $this->convertToArray($items)));
  109. }
  110. /**
  111. * 比较数组,返回差集生成的新 Collection 实例
  112. * @access public
  113. * @param mixed $items 做比较的数据
  114. * @return static
  115. */
  116. public function diff($items)
  117. {
  118. return new static(array_diff($this->items, $this->convertToArray($items)));
  119. }
  120. /**
  121. * 比较数组,返回交集组成的 Collection 新实例
  122. * @access public
  123. * @param mixed $items 比较数据
  124. * @return static
  125. */
  126. public function intersect($items)
  127. {
  128. return new static(array_intersect($this->items, $this->convertToArray($items)));
  129. }
  130. /**
  131. * 返回并删除数据中的的最后一个元素(出栈)
  132. * @access public
  133. * @return mixed
  134. */
  135. public function pop()
  136. {
  137. return array_pop($this->items);
  138. }
  139. /**
  140. * 返回并删除数据中首个元素
  141. * @access public
  142. * @return mixed
  143. */
  144. public function shift()
  145. {
  146. return array_shift($this->items);
  147. }
  148. /**
  149. * 在数组开头插入一个元素
  150. * @access public
  151. * @param mixed $value 值
  152. * @param mixed $key 键名
  153. * @return void
  154. */
  155. public function unshift($value, $key = null)
  156. {
  157. if (is_null($key)) {
  158. array_unshift($this->items, $value);
  159. } else {
  160. $this->items = [$key => $value] + $this->items;
  161. }
  162. }
  163. /**
  164. * 在数组结尾插入一个元素
  165. * @access public
  166. * @param mixed $value 值
  167. * @param mixed $key 键名
  168. * @return void
  169. */
  170. public function push($value, $key = null)
  171. {
  172. if (is_null($key)) {
  173. $this->items[] = $value;
  174. } else {
  175. $this->items[$key] = $value;
  176. }
  177. }
  178. /**
  179. * 通过使用用户自定义函数,以字符串返回数组
  180. * @access public
  181. * @param callable $callback 回调函数
  182. * @param mixed $initial 初始值
  183. * @return mixed
  184. */
  185. public function reduce(callable $callback, $initial = null)
  186. {
  187. return array_reduce($this->items, $callback, $initial);
  188. }
  189. /**
  190. * 以相反的顺序创建一个新的 Collection 实例
  191. * @access public
  192. * @return static
  193. */
  194. public function reverse()
  195. {
  196. return new static(array_reverse($this->items));
  197. }
  198. /**
  199. * 把数据分割为新的数组块
  200. * @access public
  201. * @param int $size 分隔长度
  202. * @param bool $preserveKeys 是否保持原数据索引
  203. * @return static
  204. */
  205. public function chunk($size, $preserveKeys = false)
  206. {
  207. $chunks = [];
  208. foreach (array_chunk($this->items, $size, $preserveKeys) as $chunk) {
  209. $chunks[] = new static($chunk);
  210. }
  211. return new static($chunks);
  212. }
  213. /**
  214. * 给数据中的每个元素执行回调
  215. * @access public
  216. * @param callable $callback 回调函数
  217. * @return $this
  218. */
  219. public function each(callable $callback)
  220. {
  221. foreach ($this->items as $key => $item) {
  222. $result = $callback($item, $key);
  223. if (false === $result) {
  224. break;
  225. }
  226. if (!is_object($item)) {
  227. $this->items[$key] = $result;
  228. }
  229. }
  230. return $this;
  231. }
  232. /**
  233. * 用回调函数过滤数据中的元素
  234. * @access public
  235. * @param callable|null $callback 回调函数
  236. * @return static
  237. */
  238. public function filter(callable $callback = null)
  239. {
  240. return new static(array_filter($this->items, $callback ?: null));
  241. }
  242. /**
  243. * 返回数据中指定的一列
  244. * @access public
  245. * @param mixed $columnKey 键名
  246. * @param null $indexKey 作为索引值的列
  247. * @return array
  248. */
  249. public function column($columnKey, $indexKey = null)
  250. {
  251. if (function_exists('array_column')) {
  252. return array_column($this->items, $columnKey, $indexKey);
  253. }
  254. $result = [];
  255. foreach ($this->items as $row) {
  256. $key = $value = null;
  257. $keySet = $valueSet = false;
  258. if (null !== $indexKey && array_key_exists($indexKey, $row)) {
  259. $key = (string) $row[$indexKey];
  260. $keySet = true;
  261. }
  262. if (null === $columnKey) {
  263. $valueSet = true;
  264. $value = $row;
  265. } elseif (is_array($row) && array_key_exists($columnKey, $row)) {
  266. $valueSet = true;
  267. $value = $row[$columnKey];
  268. }
  269. if ($valueSet) {
  270. if ($keySet) {
  271. $result[$key] = $value;
  272. } else {
  273. $result[] = $value;
  274. }
  275. }
  276. }
  277. return $result;
  278. }
  279. /**
  280. * 对数据排序,并返回排序后的数据组成的新 Collection 实例
  281. * @access public
  282. * @param callable|null $callback 回调函数
  283. * @return static
  284. */
  285. public function sort(callable $callback = null)
  286. {
  287. $items = $this->items;
  288. $callback = $callback ?: function ($a, $b) {
  289. return $a == $b ? 0 : (($a < $b) ? -1 : 1);
  290. };
  291. uasort($items, $callback);
  292. return new static($items);
  293. }
  294. /**
  295. * 将数据打乱后组成新的 Collection 实例
  296. * @access public
  297. * @return static
  298. */
  299. public function shuffle()
  300. {
  301. $items = $this->items;
  302. shuffle($items);
  303. return new static($items);
  304. }
  305. /**
  306. * 截取数据并返回新的 Collection 实例
  307. * @access public
  308. * @param int $offset 起始位置
  309. * @param int $length 截取长度
  310. * @param bool $preserveKeys 是否保持原先的键名
  311. * @return static
  312. */
  313. public function slice($offset, $length = null, $preserveKeys = false)
  314. {
  315. return new static(array_slice($this->items, $offset, $length, $preserveKeys));
  316. }
  317. /**
  318. * 指定的键是否存在
  319. * @access public
  320. * @param mixed $offset 键名
  321. * @return bool
  322. */
  323. public function offsetExists($offset)
  324. {
  325. return array_key_exists($offset, $this->items);
  326. }
  327. /**
  328. * 获取指定键对应的值
  329. * @access public
  330. * @param mixed $offset 键名
  331. * @return mixed
  332. */
  333. public function offsetGet($offset)
  334. {
  335. return $this->items[$offset];
  336. }
  337. /**
  338. * 设置键值
  339. * @access public
  340. * @param mixed $offset 键名
  341. * @param mixed $value 值
  342. * @return void
  343. */
  344. public function offsetSet($offset, $value)
  345. {
  346. if (is_null($offset)) {
  347. $this->items[] = $value;
  348. } else {
  349. $this->items[$offset] = $value;
  350. }
  351. }
  352. /**
  353. * 删除指定键值
  354. * @access public
  355. * @param mixed $offset 键名
  356. * @return void
  357. */
  358. public function offsetUnset($offset)
  359. {
  360. unset($this->items[$offset]);
  361. }
  362. /**
  363. * 统计数据的个数
  364. * @access public
  365. * @return int
  366. */
  367. public function count()
  368. {
  369. return count($this->items);
  370. }
  371. /**
  372. * 获取数据的迭代器
  373. * @access public
  374. * @return ArrayIterator
  375. */
  376. public function getIterator()
  377. {
  378. return new ArrayIterator($this->items);
  379. }
  380. /**
  381. * 将数据反序列化成数组
  382. * @access public
  383. * @return array
  384. */
  385. public function jsonSerialize()
  386. {
  387. return $this->toArray();
  388. }
  389. /**
  390. * 转换当前数据集为 JSON 字符串
  391. * @access public
  392. * @param integer $options json 参数
  393. * @return string
  394. */
  395. public function toJson($options = JSON_UNESCAPED_UNICODE)
  396. {
  397. return json_encode($this->toArray(), $options);
  398. }
  399. /**
  400. * 将数据转换成字符串
  401. * @access public
  402. * @return string
  403. */
  404. public function __toString()
  405. {
  406. return $this->toJson();
  407. }
  408. /**
  409. * 将数据转换成数组
  410. * @access protected
  411. * @param mixed $items 数据
  412. * @return array
  413. */
  414. protected function convertToArray($items)
  415. {
  416. return $items instanceof self ? $items->all() : (array) $items;
  417. }
  418. }