A command to generate a migration from the sql queries executed when you load some data fixtures.
This commit is contained in:
parent
f54cedfe5e
commit
0f353c1411
@ -0,0 +1,117 @@
|
|||||||
|
<?php
|
||||||
|
|
||||||
|
/*
|
||||||
|
* This file is part of the Symfony package.
|
||||||
|
*
|
||||||
|
* (c) Fabien Potencier <fabien.potencier@symfony-project.com>
|
||||||
|
*
|
||||||
|
* For the full copyright and license information, please view the LICENSE
|
||||||
|
* file that was distributed with this source code.
|
||||||
|
*/
|
||||||
|
|
||||||
|
namespace Symfony\Bundle\DoctrineMigrationsBundle\Command;
|
||||||
|
|
||||||
|
use Symfony\Component\Console\Input\InputInterface;
|
||||||
|
use Symfony\Component\Console\Output\OutputInterface;
|
||||||
|
use Symfony\Component\Console\Input\InputOption;
|
||||||
|
use Doctrine\DBAL\Migrations\Tools\Console\Command\GenerateCommand;
|
||||||
|
use Symfony\Bundle\FrameworkBundle\Command\Command;
|
||||||
|
use Doctrine\Common\DataFixtures\Executor\ORMExecutor;
|
||||||
|
use Doctrine\Common\DataFixtures\Purger\ORMPurger;
|
||||||
|
use Symfony\Bundle\DoctrineAbstractBundle\Common\DataFixtures\Loader as DataFixturesLoader;
|
||||||
|
use Doctrine\DBAL\Tools\Console\Helper\ConnectionHelper;
|
||||||
|
use Doctrine\DBAL\Migrations\Configuration\Configuration;
|
||||||
|
use Symfony\Bundle\DoctrineMigrationsBundle\SQLLogger\FixturesToMigrationSQLLogger;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Command for generating a Doctrine database migration class from a set of fixtures.
|
||||||
|
*
|
||||||
|
* @author Jonathan H. Wage <jonwage@gmail.com>
|
||||||
|
*/
|
||||||
|
class MigrationsGenerateFromDataFixturesCommand extends GenerateCommand
|
||||||
|
{
|
||||||
|
protected function configure()
|
||||||
|
{
|
||||||
|
parent::configure();
|
||||||
|
|
||||||
|
$this
|
||||||
|
->setName('doctrine:migrations:generate:from-data-fixtures')
|
||||||
|
->addOption('fixtures', null, InputOption::VALUE_OPTIONAL | InputOption::VALUE_IS_ARRAY, 'The directory or file to load data fixtures from.')
|
||||||
|
->addOption('em', null, InputOption::VALUE_OPTIONAL, 'The entity manager to use for this command.')
|
||||||
|
;
|
||||||
|
}
|
||||||
|
|
||||||
|
public function execute(InputInterface $input, OutputInterface $output)
|
||||||
|
{
|
||||||
|
if ( ! class_exists('Doctrine\Common\DataFixtures\Loader')) {
|
||||||
|
throw new \Exception('You must have the Doctrine data fixtures extension installed in order to use this command.');
|
||||||
|
}
|
||||||
|
|
||||||
|
$sqlLogger = new FixturesToMigrationSQLLogger();
|
||||||
|
|
||||||
|
$container = $this->application->getKernel()->getContainer();
|
||||||
|
|
||||||
|
$emName = $input->getOption('em');
|
||||||
|
$emName = $emName ? $emName : 'default';
|
||||||
|
|
||||||
|
DoctrineCommand::setApplicationEntityManager($this->application, $emName);
|
||||||
|
|
||||||
|
$emServiceName = sprintf('doctrine.orm.%s_entity_manager', $emName);
|
||||||
|
$em = $container->get($emServiceName);
|
||||||
|
|
||||||
|
$em->getConnection()->getConfiguration()->setSQLLogger($sqlLogger);
|
||||||
|
|
||||||
|
$configuration = $this->getMigrationConfiguration($input, $output);
|
||||||
|
|
||||||
|
$dirOrFile = $input->getOption('fixtures');
|
||||||
|
if ($dirOrFile) {
|
||||||
|
$paths = is_array($dirOrFile) ? $dirOrFile : array($dirOrFile);
|
||||||
|
} else {
|
||||||
|
$paths = array();
|
||||||
|
foreach ($this->application->getKernel()->getBundles() as $bundle) {
|
||||||
|
$paths[] = $bundle->getPath().'/DataFixtures/ORM';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
$loader = new DataFixturesLoader($container);
|
||||||
|
foreach ($paths as $path) {
|
||||||
|
if (is_dir($path)) {
|
||||||
|
$loader->loadFromDirectory($path);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
$fixtures = $loader->getFixtures();
|
||||||
|
$purger = new ORMPurger($em);
|
||||||
|
$executor = new ORMExecutor($em, $purger);
|
||||||
|
$executor->setLogger(function($message) use ($output) {
|
||||||
|
$output->writeln(sprintf(' <comment>></comment> <info>%s</info>', $message));
|
||||||
|
});
|
||||||
|
$executor->execute($fixtures);
|
||||||
|
|
||||||
|
$queries = $sqlLogger->getQueries();
|
||||||
|
|
||||||
|
$output->writeln(sprintf(' <comment>></comment> <info>%s queries logged</info>', count($queries)));
|
||||||
|
foreach ($queries as $query) {
|
||||||
|
$output->writeln(sprintf(' <comment>-</comment> <info>%s (%s)</info>', $query[0], implode(',', $query[1])));
|
||||||
|
}
|
||||||
|
|
||||||
|
$version = date('YmdHis');
|
||||||
|
|
||||||
|
$up = $this->buildCodeFromSql($configuration, $queries);
|
||||||
|
$down = 'throw new \Doctrine\DBAL\Migrations\IrreversibleMigrationException();';
|
||||||
|
$path = $this->generateMigration($configuration, $input, $version, $up, $down);
|
||||||
|
|
||||||
|
$output->writeln(sprintf(' <comment>></comment> <info>Generated new migration class to %s</info>', $path));
|
||||||
|
}
|
||||||
|
|
||||||
|
private function buildCodeFromSql(Configuration $configuration, array $queries)
|
||||||
|
{
|
||||||
|
$code = array();
|
||||||
|
foreach ($queries as $query) {
|
||||||
|
if (strpos($query[0], $configuration->getMigrationsTableName()) !== false) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
$code[] = sprintf("\$this->_addSql(\"%s\", %s);", $query[0], var_export($query[1], true));
|
||||||
|
}
|
||||||
|
return implode("\n", $code);
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,22 @@
|
|||||||
|
<?php
|
||||||
|
|
||||||
|
namespace Symfony\Bundle\DoctrineMigrationsBundle\SQLLogger;
|
||||||
|
|
||||||
|
class FixturesToMigrationSQLLogger implements \Doctrine\DBAL\Logging\SQLLogger
|
||||||
|
{
|
||||||
|
private $queries = array();
|
||||||
|
|
||||||
|
public function startQuery($sql, array $params = null, array $types = null)
|
||||||
|
{
|
||||||
|
$this->queries[] = array($sql, $params, $types);
|
||||||
|
}
|
||||||
|
|
||||||
|
public function getQueries()
|
||||||
|
{
|
||||||
|
return $this->queries;
|
||||||
|
}
|
||||||
|
|
||||||
|
public function stopQuery()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user