. */ namespace Doctrine\DBAL\Tools\Cli\Tasks; use Doctrine\Common\Cli\Tasks\AbstractTask, Doctrine\Common\Cli\CliException, Doctrine\Common\Util\Debug, Doctrine\Common\Cli\Option, Doctrine\Common\Cli\OptionGroup; /** * Task for executing arbitrary SQL that can come from a file or directly from * the command line. * * @license http://www.opensource.org/licenses/lgpl-license.php LGPL * @link www.doctrine-project.org * @since 2.0 * @version $Revision$ * @author Guilherme Blanco * @author Jonathan Wage * @author Roman Borschel */ class RunSqlTask extends AbstractTask { /** * @inheritdoc */ public function buildDocumentation() { $dqlAndFile = new OptionGroup(OptionGroup::CARDINALITY_1_1, array( new Option( 'sql', '', 'The SQL to execute.' . PHP_EOL . 'If defined, --file can not be requested on same task.' ), new Option( 'file', '', 'The path to the file with the SQL to execute.' . PHP_EOL . 'If defined, --sql can not be requested on same task.' ) )); $depth = new OptionGroup(OptionGroup::CARDINALITY_0_1, array( new Option('depth', '', 'Dumping depth of Entities graph.') )); $doc = $this->getDocumentation(); $doc->setName('run-sql') ->setDescription('Executes arbitrary SQL from a file or directly from the command line.') ->getOptionGroup() ->addOption($dqlAndFile) ->addOption($depth); } /** * @inheritdoc */ public function validate() { $arguments = $this->getArguments(); $em = $this->getConfiguration()->getAttribute('em'); if ($em === null) { throw new CliException( "Attribute 'em' of CLI Configuration is not defined or it is not a valid EntityManager." ); } if ( ! (isset($arguments['sql']) ^ isset($arguments['file']))) { throw new CliException('One of --sql or --file required, and only one.'); } return true; } /** * Executes the task. */ public function run() { $arguments = $this->getArguments(); if (isset($arguments['file'])) { //TODO } else if (isset($arguments['sql'])) { $em = $this->getConfiguration()->getAttribute('em'); if (preg_match('/^select/i', $arguments['sql'])) { $stmt = $em->getConnection()->execute($arguments['sql']); $resultSet = $stmt->fetchAll(\Doctrine\DBAL\Connection::FETCH_ASSOC); } else { $resultSet = $em->getConnection()->executeUpdate($arguments['sql']); } $maxDepth = isset($arguments['depth']) ? $arguments['depth'] : 7; Debug::dump($resultSet, $maxDepth); } } }