forked from tomispepe/magento2-1
/
Collection.php
executable file
·85 lines (76 loc) · 1.9 KB
/
Collection.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
<?php
/**
* Copyright © 2015 Magento. All rights reserved.
* See COPYING.txt for license details.
*/
/**
* Report Reviews collection
*
* @author Magento Core Team <core@magentocommerce.com>
*/
namespace Magento\Reports\Model\Resource\Review;
class Collection extends \Magento\Review\Model\Resource\Review\Collection
{
/**
* Resource initialization
*
* @return void
*/
protected function _construct()
{
$this->_init('Magento\Review\Model\Review', 'Magento\Review\Model\Resource\Review');
}
/**
* Add product filter
*
* @param string $productId
* @return $this
*/
public function addProductFilter($productId)
{
$this->addFieldToFilter('entity_pk_value', ['eq' => (int)$productId]);
return $this;
}
/**
* Reset select
*
* @return $this
*/
public function resetSelect()
{
parent::resetSelect();
$this->_joinFields();
return $this;
}
/**
* Get select count sql
*
* @return string
*/
public function getSelectCountSql()
{
$countSelect = clone $this->_select;
$countSelect->reset(\Zend_Db_Select::ORDER);
$countSelect->reset(\Zend_Db_Select::LIMIT_COUNT);
$countSelect->reset(\Zend_Db_Select::LIMIT_OFFSET);
$countSelect->reset(\Zend_Db_Select::COLUMNS);
$countSelect->columns("COUNT(main_table.review_id)");
return $countSelect;
}
/**
* Set order
*
* @param string $attribute
* @param string $dir
* @return $this
*/
public function setOrder($attribute, $dir = self::SORT_ORDER_DESC)
{
if (in_array($attribute, ['nickname', 'title', 'detail', 'created_at'])) {
$this->_select->order($attribute . ' ' . $dir);
} else {
parent::setOrder($attribute, $dir);
}
return $this;
}
}