Allow running dbal:run-sql using arbitrary connection

src/Tools/Console/Command/RunSqlCommand.php

@@ -2,6 +2,7 @@
 
 namespace Doctrine\DBAL\Tools\Console\Command;
 
+use Doctrine\DBAL\Tools\Console\Helper\ConnectionHelper;
 use Doctrine\DBAL\Tools\Dumper;
 use LogicException;
 use RuntimeException;
@@ -14,6 +15,7 @@
 use function is_bool;
 use function is_numeric;
 use function is_string;
+use function sprintf;
 use function stripos;
 
 /**
@@ -30,6 +32,7 @@ protected function configure()
         ->setDescription('Executes arbitrary SQL directly from the command line.')
         ->setDefinition([
             new InputArgument('sql', InputArgument::REQUIRED, 'The SQL statement to execute.'),
+            new InputOption('connection', null, InputOption::VALUE_REQUIRED, 'The database connection on which to execute the SQL statement', 'db'),
             new InputOption('depth', null, InputOption::VALUE_REQUIRED, 'Dumping depth of result set.', 7),
             new InputOption('force-fetch', null, InputOption::VALUE_NONE, 'Forces fetching the result.'),
         ])
@@ -44,7 +47,23 @@ protected function configure()
      */
     protected function execute(InputInterface $input, OutputInterface $output)
     {
-        $conn = $this->getHelper('db')->getConnection();
+        $connection = $input->getOption('connection');
+
+        if (! is_string($connection)) {
+            throw new LogicException("Option 'connection' must contain a string value");
+        }
+
+        $connHelper = $this->getHelper($connection);
+
+        if (! $connHelper instanceof ConnectionHelper) {
+            throw new LogicException(sprintf(
+                "Helper '%s' must be a valid '%s' object.",
+                $connection,
+                ConnectionHelper::class
+            ));
+        }
+
+        $conn = $connHelper->getConnection();
 
         $sql = $input->getArgument('sql');