/** * @param DocumentCollection $documents * @return DocumentCollection */ private function decorateDocuments(DocumentCollection $documents) { $newCollection = new DocumentCollection(); $newCollection->setTotalCount($documents->getTotalCount()); foreach ($documents as $document) { $newCollection->append($this->checkTypeAndDecorate($document)); } return $newCollection; }