Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\aggregator;
Chris@0: 
Chris@0: use Drupal\Core\Entity\Query\QueryInterface;
Chris@0: use Drupal\Core\Entity\Sql\SqlContentEntityStorage;
Chris@0: 
Chris@0: /**
Chris@0:  * Controller class for aggregators items.
Chris@0:  *
Chris@0:  * This extends the Drupal\Core\Entity\Sql\SqlContentEntityStorage class, adding
Chris@0:  * required special handling for feed item entities.
Chris@0:  */
Chris@0: class ItemStorage extends SqlContentEntityStorage implements ItemStorageInterface {
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function getItemCount(FeedInterface $feed) {
Chris@0:     $query = \Drupal::entityQuery('aggregator_item')
Chris@0:       ->condition('fid', $feed->id())
Chris@0:       ->count();
Chris@0: 
Chris@0:     return $query->execute();
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function loadAll($limit = NULL) {
Chris@0:     $query = \Drupal::entityQuery('aggregator_item');
Chris@0:     return $this->executeFeedItemQuery($query, $limit);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function loadByFeed($fid, $limit = NULL) {
Chris@0:     $query = \Drupal::entityQuery('aggregator_item')
Chris@0:       ->condition('fid', $fid);
Chris@0:     return $this->executeFeedItemQuery($query, $limit);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Helper method to execute an item query.
Chris@0:    *
Chris@0:    * @param \Drupal\Core\Entity\Query\QueryInterface $query
Chris@0:    *   The query to execute.
Chris@0:    * @param int $limit
Chris@0:    *   (optional) The number of items to return.
Chris@0:    *
Chris@0:    * @return \Drupal\aggregator\ItemInterface[]
Chris@0:    *   An array of the feed items.
Chris@0:    */
Chris@0:   protected function executeFeedItemQuery(QueryInterface $query, $limit) {
Chris@0:     $query->sort('timestamp', 'DESC')
Chris@0:       ->sort('iid', 'DESC');
Chris@0:     if (!empty($limit)) {
Chris@0:       $query->pager($limit);
Chris@0:     }
Chris@0: 
Chris@0:     return $this->loadMultiple($query->execute());
Chris@0:   }
Chris@0: 
Chris@0: }