. */ 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'])) { $em = $this->getConfiguration()->getAttribute('em'); $conn = $em->getConnection(); $printer = $this->getPrinter(); $fileNames = (array) $arguments['file']; foreach ($fileNames as $fileName) { if ( ! file_exists($fileName)) { throw new CliException(sprintf('The SQL file [%s] does not exist.', $fileName)); } else if ( ! is_readable($fileName)) { throw new CliException(sprintf('The SQL file [%s] does not have read permissions.', $fileName)); } $printer->write('Processing file [' . $fileName . ']... '); $sql = file_get_contents($fileName); if ($conn instanceof \Doctrine\DBAL\Driver\PDOConnection) { // PDO Drivers try { $lines = 0; $stmt = $conn->prepare($sql); $stmt->execute(); do { // Required due to "MySQL has gone away!" issue $stmt->fetch(); $stmt->closeCursor(); $lines++; } while ($stmt->nextRowset()); $printer->writeln(sprintf('%d statements executed!', $lines)); } catch (\PDOException $e) { $printer->writeln('error!') ->writeln($e->getMessage()); } } else { // Non-PDO Drivers (ie. OCI8 driver) $stmt = $conn->prepare($sql); $rs = $stmt->execute(); if ($rs) { $printer->writeln('OK!'); } else { $error = $stmt->errorInfo(); $printer->writeln('error!') ->writeln($error['message']); } $stmt->closeCursor(); } } } 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); } } }