* @copyright 2019 Akeneo SAS (http://www.akeneo.com) * @license http://opensource.org/licenses/osl-3.0.php Open Software License (OSL 3.0) */ class UpdateMappingIndexCommand extends Command { protected static $defaultName = 'akeneo:elasticsearch:update-mapping'; /** @var ClientRegistry */ private $esClientsRegistry; /** @var array */ private $hosts; public function __construct(ClientRegistry $clientRegistry, $hosts) { $this->esClientsRegistry = $clientRegistry; $this->hosts = is_string($hosts) ? [$hosts] : $hosts; parent::__construct(self::$defaultName); } public function configure() { $this ->addArgument( 'indices', InputArgument::IS_ARRAY, 'Elasticsearch indices name to reindex, separated by spaces' ); $this->addOption('all', 'a', InputOption::VALUE_NONE, "Use --all if you want to update all mappings of all indices", null); } public function execute(InputInterface $input, OutputInterface $output): int { $indices = $input->getOption('all') ? [] : $input->getArgument('indices'); $io = new SymfonyStyle($input, $output); $warning = <<warning($warning); if (!$io->confirm('Are you sure to continue?', true)) { $output->writeln("You decided to abort your Elasticearch mapping update"); return 0; } $clients = $this->esClients($indices); $names = array_map(function (Client $client): string { return $client->getIndexName(); }, $clients); $io->writeln("You will migrate those indices (if it misses one you gave, it means that you didn't write it correctly) : "); $io->listing($names); $updateIndexMapping = new UpdateIndexMapping(); foreach ($clients as $client) { $io->note("Starting to migrate: {$client->getIndexName()}"); $nativeClient = $this->buildNativeClient($client); $updateIndexMapping->updateIndexMapping($nativeClient['client'], $nativeClient['name'], $nativeClient['configuration']); $io->note("Finished to migrate: {$client->getIndexName()}"); } $io->success("All the indices listed above have been migrated"); return 0; } private function buildNativeClient(Client $client): array { $clientBuilder = new ClientBuilder(); $clientBuilder->setHosts($this->hosts); $nativeClient = $clientBuilder->build(); return [ 'client' => $nativeClient, 'configuration' => $client->getConfigurationLoader(), 'name' => $client->getIndexName() ]; } private function esClients(array $indexNames = []): array { return array_filter($this->esClientsRegistry->getClients(), function (Client $client) use ($indexNames) { if ($indexNames === []) { return true; } return in_array($client->getIndexName(), $indexNames); }); } }