Added before and after query callbacks

AshleyDawson · Jan 25, 2018 · de3bb0c · de3bb0c
1 parent ca3bf92
commit de3bb0c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/AshleyDawson/SimplePagination/Paginator.php b/lib/AshleyDawson/SimplePagination/Paginator.php
@@ -200,35 +200,37 @@ public function getItemTotalCallback()
     }
 
     /**
-     * @return \Closure
+     * {@inheritdoc}
      */
     public function getBeforeQueryCallback()
     {
         return $this->beforeQueryCallback;
     }
 
     /**
-     * @param \Closure $beforeQueryCallback
+     * {@inheritdoc}
      */
     public function setBeforeQueryCallback($beforeQueryCallback)
     {
         $this->beforeQueryCallback = $beforeQueryCallback;
+        return $this;
     }
 
     /**
-     * @return \Closure
+     * {@inheritdoc}
      */
     public function getAfterQueryCallback()
     {
         return $this->afterQueryCallback;
     }
 
     /**
-     * @param \Closure $afterQueryCallback
+     * {@inheritdoc}
      */
     public function setAfterQueryCallback($afterQueryCallback)
     {
         $this->afterQueryCallback = $afterQueryCallback;
+        return $this;
     }
 
     /**