/**
* Returns the MongoDB instance for a class.
*
* @param string $className The class name.
* @return \Doctrine\MongoDB\Database
*/
public function getDocumentDatabase($className)
{
if (isset($this->documentDatabases[$className])) {
return $this->documentDatabases[$className];
}
$metadata = $this->metadataFactory->getMetadataFor($className);
$db = $metadata->getDatabase();
$db = $db ? $db : $this->config->getDefaultDB();
$db = $db ? $db : 'doctrine';
$this->documentDatabases[$className] = $this->connection->selectDatabase($db);
return $this->documentDatabases[$className];
}