Chris@17: <?php
Chris@17: 
Chris@17: namespace Drupal\workspaces\EntityQuery;
Chris@17: 
Chris@17: use Drupal\Core\Entity\Query\Sql\QueryAggregate as BaseQueryAggregate;
Chris@17: 
Chris@17: /**
Chris@17:  * Alters aggregate entity queries to use a workspace revision if possible.
Chris@17:  */
Chris@17: class QueryAggregate extends BaseQueryAggregate {
Chris@17: 
Chris@17:   use QueryTrait {
Chris@17:     prepare as traitPrepare;
Chris@17:   }
Chris@17: 
Chris@17:   /**
Chris@17:    * {@inheritdoc}
Chris@17:    */
Chris@17:   public function prepare() {
Chris@17:     // Aggregate entity queries do not return an array of entity IDs keyed by
Chris@17:     // revision IDs, they only return the values of the aggregated fields, so we
Chris@17:     // don't need to add any expressions like we do in
Chris@17:     // \Drupal\workspaces\EntityQuery\Query::prepare().
Chris@17:     $this->traitPrepare();
Chris@17: 
Chris@17:     // Throw away the ID fields.
Chris@17:     $this->sqlFields = [];
Chris@17:     return $this;
Chris@17:   }
Chris@17: 
Chris@17: }