EC-CUBE / ec-cube

src/Eccube/Doctrine/Common/CsvDataFixtures/CsvFixture.php

<?php

namespace Eccube\Doctrine\Common\CsvDataFixtures;

use Doctrine\Common\DataFixtures\FixtureInterface;
use Doctrine\Common\Persistence\ObjectManager;
use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Schema\Table;

/**
 * CSVファイルを扱うためのフィクスチャ.
 *
 * @see https://github.com/doctrine/data-fixtures/blob/master/lib/Doctrine/Common/DataFixtures/FixtureInterface.php
 */
class CsvFixture implements FixtureInterface
{
    /** @var \SplFileObject $file */
    protected $file;

    public function __construct(\SplFileObject $file)

Missing function doc comment

    {
        $this->file = $file;
    }

    /**
     * {@inheritdoc}
     */
    public function load(ObjectManager $manager)
    {
        // CSV Reader に設定
        $this->file->setFlags(\SplFileObject::READ_CSV | \SplFileObject::READ_AHEAD | \SplFileObject::SKIP_EMPTY);

        // ヘッダ行を取得
        $headers = $this->file->current();
        $this->file->next();

        // ファイル名からテーブル名を取得
        $table_name = str_replace('.'.$this->file->getExtension(), '', $this->file->getFilename());
        $sql = $this->getSql($table_name, $headers);

It seems like $headers defined by $this->file->current() on line 34 can also be of type string; however, Eccube\Doctrine\Common\CsvDataFixtures\CsvFixture::getSql() does only seem to accept array, maybe add an additional type check?

        /** @var Connection $Connection */
        $Connection = $manager->getConnection();

It seems like you code against a concrete implementation and not the interface Doctrine\Common\Persistence\ObjectManager as the method getConnection() does only exist in the following implementations of said interface: Doctrine\ORM\Decorator\EntityManagerDecorator, Doctrine\ORM\EntityManager.

        $Connection->beginTransaction();

        // mysqlの場合はNO_AUTO_VALUE_ON_ZEROを設定
        if ('mysql' === $Connection->getDatabasePlatform()->getName()) {
            $Connection->exec("SET SESSION sql_mode='NO_AUTO_VALUE_ON_ZERO';");
        }

        // TODO エラーハンドリング
        $prepare = $Connection->prepare($sql);
        while (!$this->file->eof()) {
            $rows = $this->file->current();
            $index = 1;
            // データ行をバインド
            foreach ($rows as $col) {

The expression $rows of type string|array is not guaranteed to be traversable. How about adding an additional type check?

                $col = $col === '' ? null : $col;
                $prepare->bindValue($index, $col);
                $index++;
            }
            // batch insert
            $result = $prepare->execute();

$result is not used, you could remove the assignment.

            $this->file->next();
        }
        $Connection->commit();

        // postgresqlの場合はシーケンスを振り直す
        if ('postgresql' === $Connection->getDatabasePlatform()->getName()) {
            // テーブル情報を取得
            $sm = $Connection->getSchemaManager();
            $table = $sm->listTableDetails($table_name);

            // 主キーがないテーブルはスキップ
            if (!$table->hasPrimaryKey()) {
                return;
            }

            // 複合主キーのテーブルはスキップ
            $pkColumns = $table->getPrimaryKey()->getColumns();
            if (count($pkColumns) != 1) {
                return;
            }

            // シーケンス名を取得
            $pk_name = $pkColumns[0];
            $sequence_name = sprintf('%s_%s_seq', $table_name, $pk_name);

            // シーケンスの存在チェック
            $sql = 'SELECT COUNT(*) FROM information_schema.sequences WHERE sequence_name = ?';
            $count = $Connection->fetchColumn($sql, [$sequence_name]);
            if ($count < 1) {
                return;
            }

            // シーケンスを更新
            $sql = sprintf('SELECT MAX(%s) FROM %s', $pk_name, $table_name);
            $max = $Connection->fetchColumn($sql);
            if (is_null($max)) {
                // レコードが無い場合は1を初期値に設定
                $sql = sprintf("SELECT SETVAL('%s', 1, false)", $sequence_name);
            } else {
                // レコードがある場合は最大値を設定
                $sql = sprintf("SELECT SETVAL('%s', %s)", $sequence_name, $max);
            }
            $Connection->executeQuery($sql);
        }
    }

    /**
     * INSERT を生成する.
     *
     * @param string $table_name テーブル名
     * @param array $headers カラム名の配列

Expected 2 spaces after parameter type; 1 found

Expected 4 spaces after parameter name; 1 found

     * @return string INSERT 文
     */
    public function getSql($table_name, array $headers)
    {
        return 'INSERT INTO '.$table_name.' ('.implode(', ', $headers).') VALUES ('.implode(', ', array_fill(0, count($headers), '?')).')';
    }

    /**
     * 保持している \SplFileObject を返す.
     *
     * @return \SplFileObject
     */
    public function getFile()
    {
        return $this->file;
    }
}