. */ namespace Doctrine\ORM\Tools\Cli\Tasks; use Doctrine\Common\DoctrineException, 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() { $args = $this->getArguments(); $printer = $this->getPrinter(); $isSql = isset($args['sql']); $isFile = isset($args['file']); if ( ! ($isSql ^ $isFile)) { $printer->writeln("One of --sql or --file required, and only one.", 'ERROR'); return false; } return true; } /** * Executes the task. */ public function run() { $args = $this->getArguments(); try { if (isset($args['file'])) { //TODO } else if (isset($args['sql'])) { $conn = $this->getEntityManager()->getConnection(); if (preg_match('/^select/i', $args['sql'])) { $stmt = $conn->execute($args['sql']); $resultSet = $stmt->fetchAll(\Doctrine\DBAL\Connection::FETCH_ASSOC); } else { $resultSet = $conn->executeUpdate($args['sql']); } $maxDepth = isset($args['depth']) ? $args['depth'] : 7; Debug::dump($resultSet, $maxDepth); } } catch (\Exception $ex) { throw new DoctrineException($ex); } } }