Edyan\Neuralyzer\Helper\DB\MySQL

Complexity

Total Complexity

5

Size/Duplication

Total Lines	61
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	getDriverOptions()	0	4	1
A	registerCustomTypes()	0	16	2
A	loadData()	0	16	2

<?php
/**
 * neuralyzer : Data Anonymization Library and CLI Tool
 *
 * PHP Version 7.1
 *
 * @author Emmanuel Dyan
 * @copyright 2018 Emmanuel Dyan
 *
 * @package edyan/neuralyzer
 *
 * @license GNU General Public License v2.0
 *
 * @link https://github.com/edyan/neuralyzer
 */

namespace Edyan\Neuralyzer\Helper\DB;

/**
 * Various methods related to MySQL
 */
class MySQL extends AbstractDBHelper
{
    /**
     * Send options to be able to load dataset
     * @return array
     */
    public static function getDriverOptions(): array
    {
        return [
            \PDO::MYSQL_ATTR_LOCAL_INFILE => true
        ];
    }


    /**
     * Add a custom enum type
     * @return void
     */
    public function registerCustomTypes(): void
    {
        // already registered
        if (\Doctrine\DBAL\Types\Type::hasType('neuralyzer_enum')) {
            return;
        }

        // Else register
        // Manage specific types such as enum
        \Doctrine\DBAL\Types\Type::addType(
            'neuralyzer_enum',
            'Edyan\Neuralyzer\Doctrine\Type\Enum'
        );
        $platform = $this->conn->getDatabasePlatform();
        $platform->registerDoctrineTypeMapping('enum', 'neuralyzer_enum');
        $platform->registerDoctrineTypeMapping('bit', 'boolean');
    }


    /**
     * Load Data from a CSV
     * @param  string  $table
     * @param  string  $filename
     * @param  array   $fields
     * @param  string  $mode  Not in used here
     * @return string
     */
    public function loadData(
        string $table,
        string $filename,
        array $fields,
        string $mode
    ): string {
        $sql ="LOAD DATA LOCAL INFILE '{$filename}'
     REPLACE INTO TABLE {$table}
     FIELDS TERMINATED BY '|' ENCLOSED BY '\"' LINES TERMINATED BY '" . PHP_EOL . "'
     (`" . implode("`, `", $fields) . "`)";
        // Run the query if asked
        if ($this->pretend === false) {
            $this->conn->query($sql);
        }

        return $sql;
    }
}