Example #1
0
 /**
  * Add attribute to sort
  *
  * @param string $attribute
  * @param string $dir
  * @return $this|\Magento\Catalog\Model\ResourceModel\Product\Collection
  */
 public function addAttributeToSort($attribute, $dir = self::SORT_ORDER_ASC)
 {
     if (in_array($attribute, ['review_cnt', 'last_created', 'avg_rating', 'avg_rating_approved'])) {
         $this->getSelect()->order($attribute . ' ' . $dir);
         return $this;
     }
     return parent::addAttributeToSort($attribute, $dir);
 }