| @@ -4,9 +4,9 @@ | ||
| 4 | 4 | |
| 5 | 5 | class MapperData |
| 6 | 6 | { |
| 7 | 7 | /** |
| 8 | - * @var PostMapper | |
| 8 | + * @var MapperInterface | |
| 9 | 9 | */ |
| 10 | 10 | private $_mapper; |
| 11 | 11 | |
| 12 | 12 | private $template_data; |
| @@ -11,9 +11,9 @@ | ||
| 11 | 11 | |
| 12 | 12 | private $template_data; |
| 13 | 13 | |
| 14 | 14 | /** |
| 15 | - * @param PostMapper $mapper | |
| 15 | + * @param MapperInterface $mapper | |
| 16 | 16 | * @param int $i |
| 17 | 17 | */ |
| 18 | 18 | public function __construct($mapper, $i) |
| 19 | 19 | { |
| @@ -25,11 +25,16 @@ | ||
| 25 | 25 | { |
| 26 | 26 | return $this->_mapper->filter(); |
| 27 | 27 | } |
| 28 | 28 | |
| 29 | - public function data($args) | |
| 29 | + public function get_total_records() | |
| 30 | 30 | { |
| 31 | - $record = $this->_mapper->record(); | |
| 31 | + return count($this->_mapper->records); | |
| 32 | + } | |
| 33 | + | |
| 34 | + public function data($args, $index = 0) | |
| 35 | + { | |
| 36 | + $record = $this->_mapper->record($index); | |
| 32 | 37 | |
| 33 | 38 | if (empty($args)) { |
| 34 | 39 | return [$record]; |
| 35 | 40 | } |