/**
* Returns the Riak instance for a class.
*
* @param string $className The class name.
* @return \CosmoW\Riak\Database
*/
public function getDocumentDatabase($className)
{
$className = ltrim($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];
}
