12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- <?php
- namespace App\Commands;
- use App\AppService;
- use Symfony\Component\Console\Helper\Table;
- use Symfony\Component\Console\Command\Command;
- use Symfony\Component\Console\Input\InputArgument;
- use Symfony\Component\Console\Input\InputInterface;
- use Symfony\Component\Console\Output\OutputInterface;
- use YdbPlatform\Ydb\Session;
- use YdbPlatform\Ydb\Ydb;
- class SelectCommand extends Command
- {
- /**
- * @var string
- */
- protected static $defaultName = 'select';
- /**
- * @var AppService
- */
- protected $appService;
- public function __construct()
- {
- $this->appService = new AppService;
- parent::__construct();
- }
- protected function configure()
- {
- $this->setDescription('Select rows form a table.');
- $this->addArgument('table', InputArgument::REQUIRED, 'The table name.');
- }
- /**
- * @param InputInterface $input
- * @param OutputInterface $output
- * @return int
- * @throws \YandexCloud\Ydb\Exception
- */
- protected function execute(InputInterface $input, OutputInterface $output)
- {
- $table_name = $input->getArgument('table') ?: '';
- $ydb = $this->appService->initYdb();
- $result = $ydb->table()->retrySession(function (Session $session) use ($table_name, $output) {
- return $session->query('select * from `' . $table_name . '` limit 10;');
- }, true);
- $output->writeln('Column count: ' . $result->columnCount());
- $output->writeln('Row count: ' . $result->rowCount());
- $t = new Table($output);
- $t
- ->setHeaders(array_map(function($column) {
- return $column['name'];
- }, $result->columns()))
- ->setRows($result->rows())
- ;
- $t->render();
- return Command::SUCCESS;
- }
- }
|