/**
* Returns the MongoDB instance for a class.
*
* @param string $className The class name.
* @return Doctrine\MongoDB\Database
*/
public function getDocumentDatabase($className)
{
$metadata = $this->metadataFactory->getMetadataFor($className);
$db = $metadata->getDatabase();
$db = $db ? $db : $this->config->getDefaultDB();
$db = $db ? $db : 'doctrine';
$db = sprintf('%s%s', $this->config->getEnvironmentPrefix(), $db);
$database = $this->connection->selectDatabase($db);
return $database;
}
/**
* Returns the MongoDB instance for a class.
*
* @param string $className The class name.
* @return Doctrine\ODM\MongoDB\MongoDB
*/
public function getDocumentDB($className)
{
$db = $this->_metadataFactory->getMetadataFor($className)->getDB();
$db = $db ? $db : $this->_config->getDefaultDB();
$db = $db ? $db : 'doctrine';
$db = sprintf('%s%s', $this->_config->getEnvironmentPrefix(), $db);
if ($db && !isset($this->_documentDBs[$db])) {
$database = $this->_mongo->selectDB($db);
$this->_documentDBs[$db] = new MongoDB($database);
}
if (!isset($this->_documentDBs[$db])) {
throw MongoDBException::documentNotMappedToDB($className);
}
return $this->_documentDBs[$db];
}
