AbstractDatabase.php

Go to the documentation of this file.

<?php
namespace Magento\MediaStorage\Model\File\Storage\Database;

abstract class AbstractDatabase extends \Magento\Framework\Model\AbstractModel
{
    const CONNECTION_DEFAULT = 'default_setup';

    protected $_coreFileStorageDb = null;

    protected $_date;

    protected $_configuration;

    private $connectionName = self::CONNECTION_DEFAULT;

    public function __construct(
        \Magento\Framework\Model\Context $context,
        \Magento\Framework\Registry $registry,
        \Magento\MediaStorage\Helper\File\Storage\Database $coreFileStorageDb,
        \Magento\Framework\Stdlib\DateTime\DateTime $dateModel,
        \Magento\Framework\App\Config\ScopeConfigInterface $configuration,
        \Magento\Framework\Model\ResourceModel\AbstractResource $resource = null,
        \Magento\Framework\Data\Collection\AbstractDb $resourceCollection = null,
        $connectionName = null,
        array $data = []
    ) {
        parent::__construct($context, $registry, $resource, $resourceCollection, $data);
        $this->_configuration = $configuration;
        $this->_coreFileStorageDb = $coreFileStorageDb;
        $this->_date = $dateModel;
        if (!$connectionName) {
            $connectionName = $this->getConfigConnectionName();
        }
        $this->setConnectionName($connectionName);
    }

    public function getConfigConnectionName()
    {
        $connectionName = $this->_configuration
            ->getValue(
                \Magento\MediaStorage\Model\File\Storage::XML_PATH_STORAGE_MEDIA_DATABASE,
                'default'
            );
        if (empty($connectionName)) {
            $connectionName = self::CONNECTION_DEFAULT;
        }
        return $connectionName;
    }

    protected function _getResource()
    {
        $resource = parent::_getResource();
        $resource->setConnectionName($this->getConnectionName());

        return $resource;
    }

    public function prepareStorage()
    {
        $this->_getResource()->createDatabaseScheme();

        return $this;
    }

    public function setConnectionName($connectionName)
    {
        if (!empty($connectionName)) {
            $this->connectionName = $connectionName;
            $this->_getResource()->setConnectionName($this->connectionName);
        }

        return $this;
    }

    public function getConnectionName()
    {
        return $this->connectionName;
    }
}