LimitCallback support

Author: michallohnisky

README.md

@@ -50,6 +50,11 @@ $dataSource = $this->queryObjectDataSourceFactory->create($qo, "id")
 					$queryObject->{'by' . $field}($value);
 			}
 		}
+	})
+	->setLimitCallback(function ($offset, $limit, $defaultCallback) use ($query, $itemRepository) {
+		// This callback is not necessary, but you can do your stuff with $offset and $limit here.
+
+		$defaultCallback();	// Run the default action
 	});
 
 $grid->setDataSource($queryObjectDataSource);

src/QueryObjectDataSource.php

@@ -29,6 +29,9 @@ class QueryObjectDataSource implements \Ublaboo\DataGrid\DataSource\IDataSource
 	/** @var callable */
 	public $sortCallback;
 
+	/** @var callable */
+	public $limitCallback;
+
 	/**
 	 * @param \Kdyby\Doctrine\QueryObject $queryObject
 	 * @param \Kdyby\Doctrine\EntityRepository $repo
@@ -65,6 +68,18 @@ public function setSortCallback($callback) {
 		return $this;
 	}
 
+	/**
+	 * Pro nastavení limitu lze specifikovat callback $callback. Pokud není zadán, je použit defaultní callback $defaultCallback.
+	 * Callback $callback dostane mezi parametry i $defaultCallback, aby ho mohl případně zavolat.
+	 *
+	 * @param callable $callback Dostane parametry $offset, $limit, $defaultCallback
+	 * @return self
+	 */
+	public function setLimitCallback($callback) {
+		$this->limitCallback = $callback;
+		return $this;
+	}
+
 	protected function getResultSet() {
 		if (!$this->resultSet) {
 			$this->resultSet = $this->repo
@@ -144,7 +159,16 @@ public function filterOne(array $filter) {
 	 * @return static
 	 */
 	public function limit($offset, $limit) {
-		$this->getResultSet()->applyPaging($offset, $limit);
+		$defaultCallback = function () use ($offset, $limit) {
+			$this->getResultSet()->applyPaging($offset, $limit);
+		};
+
+		if (is_callable($this->limitCallback)) {
+			$this->limitCallback($offset, $limit, $defaultCallback);
+
+		} else {
+			$defaultCallback();
+		}
 
 		return $this;
 	}