CreateViewsBackup.php 2.95 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\Backup\Model\ResourceModel\View;

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\Backup\Db\BackupInterface;
use Magento\Framework\DB\Adapter\AdapterInterface;

/**
 * Creates backup of Views in the database.
 */
class CreateViewsBackup
{
    /**
     * @var GetListViews
     */
    private $getListViews;

    /**
     * @var ResourceConnection
     */
    private $resourceConnection;

    /**
     * @var AdapterInterface
     */
    private $connection;

    /**
     * @param GetListViews $getListViews
     * @param ResourceConnection $resourceConnection
     */
    public function __construct(
        GetListViews $getListViews,
        ResourceConnection $resourceConnection
    ) {
        $this->getListViews = $getListViews;
        $this->resourceConnection = $resourceConnection;
    }

    /**
     * Write backup data to backup file.
     *
     * @param BackupInterface $backup
     */
    public function execute(BackupInterface $backup): void
    {
        $views = $this->getListViews->execute();

        foreach ($views as $view) {
            $backup->write($this->getViewHeader($view));
            $backup->write($this->getDropViewSql($view));
            $backup->write($this->getCreateView($view));
        }
    }

    /**
     * Retrieve Database connection for Backup.
     *
     * @return AdapterInterface
     */
    private function getConnection(): AdapterInterface
    {
        if (!$this->connection) {
            $this->connection = $this->resourceConnection->getConnection('backup');
        }

        return $this->connection;
    }

    /**
     * Get CREATE VIEW query for the specific view.
     *
     * @param string $viewName
     * @return string
     */
    private function getCreateView(string $viewName): string
    {
        $quotedViewName = $this->getConnection()->quoteIdentifier($viewName);
        $query = 'SHOW CREATE VIEW ' . $quotedViewName;
        $row = $this->getConnection()->fetchRow($query);
        $regExp = '/\sDEFINER\=\`([^`]*)\`\@\`([^`]*)\`/';
        $sql = preg_replace($regExp, '', $row['Create View']);

        return $sql . ';' . "\n";
    }

    /**
     * Prepare a header for View being dumped.
     *
     * @param string $viewName
     * @return string
     */
    public function getViewHeader(string $viewName): string
    {
        $quotedViewName = $this->getConnection()->quoteIdentifier($viewName);
        return "\n--\n" . "-- Structure for view {$quotedViewName}\n" . "--\n\n";
    }

    /**
     * Make sure that View being created is deleted if already exists.
     *
     * @param string $viewName
     * @return string
     */
    public function getDropViewSql(string $viewName): string
    {
        $quotedViewName = $this->getConnection()->quoteIdentifier($viewName);
        return sprintf('DROP VIEW IF EXISTS %s;\n', $quotedViewName);
    }
}