public function translate(AfterCommandInterface $command) { $result = $command->extractResult(); if (null === $result) { return null; } $class = $command->hasClass() ? $command->getClass() : EntityOperatedCommandInterface::DEFAULT_CLASS; if ($result instanceof Collection) { $items = clone $result; $items->map(function ($itemData) use($class) { $entity = $this->toEntity($itemData, $class); return $entity; }); return $items; } else { return $this->toEntity($result, $class); } }