public function getRoot(ManagedRepositoryInterface $repository) { $className = $repository->getClassName(); if (!isset($this->nodes[$className]) || !isset($this->nodes[$className]['_root'])) { throw new \InvalidArgumentException(sprintf('No root node exists for repository %s', $repository->getName())); } return $this->nodes[$className]['_root']; }