kiri-core/Database/Pagination.php

<?php
declare(strict_types=1);

namespace Database;


use Snowflake\Abstracts\Component;
use Closure;
use Exception;
use Snowflake\Event;
use Snowflake\Snowflake;
use Swoole\Coroutine;

/**
 * Class Pagination
 * @package Database
 */
class Pagination extends Component
{

	/** @var ActiveQuery */
	private ActiveQuery $activeQuery;

	/** @var int 从第几个开始查 */
	private int $_offset = 0;

	/** @var int 每页数量 */
	private int $_limit = 100;

	/** @var int 最大查询数量 */
	private int $_max = 0;

	/** @var int 当前已查询数量 */
	private int $_length = 0;

	/** @var Closure */
	private Closure $_callback;

	/** @var Coroutine\WaitGroup */
	private Coroutine\WaitGroup $_group;

	/**
	 * PaginationIteration constructor.
	 * @param ActiveQuery $activeQuery
	 * @param array $config
	 */
	public function __construct(ActiveQuery $activeQuery, array $config = [])
	{
		parent::__construct($config);
		$this->activeQuery = $activeQuery;
	}


	/**
	 * @param array|Closure $callback
	 * @throws Exception
	 */
	public function setCallback(array|Closure $callback)
	{
		if (!is_callable($callback, true)) {
			throw new Exception('非法回调函数~');
		}
		$this->_callback = $callback;
	}


	/**
	 * @param int $number
	 * @return Pagination
	 */
	public function setOffset(int $number): static
	{
		if ($number < 0) {
			$number = 0;
		}
		$this->_offset = $number;
		return $this;
	}


	/**
	 * @param int $number
	 * @return Pagination
	 */
	public function setLimit(int $number): static
	{
		if ($number < 1) {
			$number = 100;
		} else if ($number > 5000) {
			$number = 5000;
		}
		$this->_limit = $number;
		return $this;
	}


	/**
	 * @param int $number
	 * @return Pagination
	 */
	public function setMax(int $number): static
	{
		if ($number < 0) {
			return $this;
		}
		$this->_max = $number;
		return $this;
	}


	/**
	 * @param array $param
	 * @return void
	 */
	public function plunk($param = [])
	{
		$this->_group = new Coroutine\WaitGroup();
		Coroutine::create([$this, 'loop'], $param);
		$this->_group->wait();
	}


	/**
	 * 轮训
	 * @param $param
	 * @return array
	 */
	public function loop($param): array
	{
		if ($this->_max > 0 && $this->_length >= $this->_max) {
			return $this->output();
		}
		[$length, $data] = $this->get();

		$this->runner($data, $param);

		unset($data);
		if ($length < $this->_limit) {
			return $this->output();
		}
		return $this->loop($param);
	}


	/**
	 * @return array
	 */
	public function output(): array
	{
		return [];
	}


	/**
	 * @param $data
	 * @param $param
	 */
	private function runner($data, $param)
	{
		if (Snowflake::inCoroutine()) {
			$this->executed($this->_callback, $data, $param);
		} else {
			call_user_func($this->_callback, $data, $param);
		}
	}


	/**
	 * @param $callback
	 * @param $data
	 * @param $param
	 * 解释器
	 * @return mixed
	 */
	private function executed($callback, $data, $param): mixed
	{
		$this->_group->add(1);
		return Coroutine::create(function ($callback, $data, $param): void {
			try {
				call_user_func($callback, $data, $param);
			} catch (\Throwable $exception) {
				$this->addError($exception);
			} finally {
				$event = Snowflake::app()->getEvent();
				$event->trigger(Event::SYSTEM_RESOURCE_RELEASES);
				$this->_group->done();
			}
		}, $callback, $data, $param);
	}


	/**
	 * @return array|Collection
	 */
	private function get(): Collection|array
	{
		if ($this->_length + $this->_limit > $this->_max) {
			$this->_limit = $this->_length + $this->_limit - $this->_max;
		}
		$data = $this->activeQuery->limit($this->_offset, $this->_limit)->get();
		$this->_offset += $this->_limit;
		$this->_length += $data->size();
		return [$data->size(), $data];
	}

}