Graze\DataDb\Export\Query\RedshiftQueryExporter

Complexity

Total Complexity

5

Size/Duplication

Total Lines	57
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	8

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	7	2
A	validateFile()	0	6	1
A	export()	0	16	2

<?php

namespace Graze\DataDb\Export\Query;

use Graze\DataDb\Helper\RedshiftHelper;
use Graze\DataDb\QueryNodeInterface;
use Graze\DataFile\Format\FormatAwareInterface;
use Graze\DataFile\Helper\Builder\BuilderAwareInterface;
use Graze\DataFile\Helper\Builder\BuilderTrait;
use Graze\DataFile\Helper\OptionalLoggerTrait;
use Graze\DataFile\Node\FileNodeInterface;
use InvalidArgumentException;
use League\Flysystem\AwsS3v3\AwsS3Adapter;
use Psr\Log\LoggerAwareInterface;
use Psr\Log\LogLevel;

class RedshiftQueryExporter implements QueryExporterInterface, BuilderAwareInterface, LoggerAwareInterface
{
    use OptionalLoggerTrait;
    use BuilderTrait;

    /** @var FileNodeInterface */
    private $file;

    /**
     * RedshiftQueryExporter constructor.
     *
     * @param FileNodeInterface $file
     */
    public function __construct(FileNodeInterface $file)
    {
        if (!$this->validateFile($file)) {
            throw new InvalidArgumentException("The supplied file: $file should be a s3 file");
        }
        $this->file = $file;
    }

    /**
     * @param FileNodeInterface $file
     *
     * @return bool
     */
    private function validateFile(FileNodeInterface $file)
    {
        $filesystem = $file->getFilesystem();
        $adapter = $filesystem->getAdapter();
        return ($adapter instanceof AwsS3Adapter);
    }

    /**
     * Export a table to somethingMysqlTableExporter
     *
     * @param QueryNodeInterface $query
     *
     * @return FileNodeInterface
     */
    public function export(QueryNodeInterface $query)
    {
        $helper = $this->getBuilder()->build(RedshiftHelper::class, $query->getAdapter()->getDialect());
        $format = $helper->getDefaultExportFormat();

        $this->log(LogLevel::INFO, "Exporting redshift query to file: {file}", ['file' => $this->file]);
        /** @var RedshiftExportQuery $exportQuery */
        $exportQuery = $this->getBuilder()->build(RedshiftExportQuery::class, $query, $this->file, $format);

        $exportQuery->query();

        if ($this->file instanceof FormatAwareInterface) {
            $this->file->setFormat($format);
        }
        return $this->file;
    }
}