Merge pull request #23444 from nextcloud/notify-dry-run

add dry-run option to files_external:notify
This commit is contained in:
Roeland Jago Douma 2021-03-30 21:46:27 +02:00 committed by GitHub
commit 0a6416f84f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 52 additions and 32 deletions

View File

@ -99,6 +99,11 @@ class Notify extends Base {
'', '',
InputOption::VALUE_NONE, InputOption::VALUE_NONE,
'Disable self check on startup' 'Disable self check on startup'
)->addOption(
'dry-run',
'',
InputOption::VALUE_NONE,
'Don\'t make any changes, only log detected changes'
); );
parent::configure(); parent::configure();
} }
@ -181,23 +186,24 @@ class Notify extends Base {
return 1; return 1;
} }
$verbose = $input->getOption('verbose'); $dryRun = $input->getOption('dry-run');
if ($dryRun && $output->getVerbosity() < OutputInterface::VERBOSITY_VERBOSE) {
$output->setVerbosity(OutputInterface::VERBOSITY_VERBOSE);
}
$path = trim($input->getOption('path'), '/'); $path = trim($input->getOption('path'), '/');
$notifyHandler = $storage->notify($path); $notifyHandler = $storage->notify($path);
if (!$input->getOption('no-self-check')) { if (!$input->getOption('no-self-check')) {
$this->selfTest($storage, $notifyHandler, $verbose, $output); $this->selfTest($storage, $notifyHandler, $output);
} }
$notifyHandler->listen(function (IChange $change) use ($mount, $verbose, $output) { $notifyHandler->listen(function (IChange $change) use ($mount, $output, $dryRun) {
if ($verbose) { $this->logUpdate($change, $output);
$this->logUpdate($change, $output);
}
if ($change instanceof IRenameChange) { if ($change instanceof IRenameChange) {
$this->markParentAsOutdated($mount->getId(), $change->getTargetPath(), $output); $this->markParentAsOutdated($mount->getId(), $change->getTargetPath(), $output, $dryRun);
} }
$this->markParentAsOutdated($mount->getId(), $change->getPath(), $output); $this->markParentAsOutdated($mount->getId(), $change->getPath(), $output, $dryRun);
}); });
return 0; return 0;
} }
@ -207,29 +213,44 @@ class Notify extends Base {
return new $class($mount->getBackendOptions()); return new $class($mount->getBackendOptions());
} }
private function markParentAsOutdated($mountId, $path, OutputInterface $output) { private function markParentAsOutdated($mountId, $path, OutputInterface $output, bool $dryRun) {
$parent = ltrim(dirname($path), '/'); $parent = ltrim(dirname($path), '/');
if ($parent === '.') { if ($parent === '.') {
$parent = ''; $parent = '';
} }
try { try {
$storageIds = $this->getStorageIds($mountId); $storages = $this->getStorageIds($mountId, $parent);
} catch (DriverException $ex) { } catch (DriverException $ex) {
$this->logger->logException($ex, ['message' => 'Error while trying to find correct storage ids.', 'level' => ILogger::WARN]); $this->logger->logException($ex, ['message' => 'Error while trying to find correct storage ids.', 'level' => ILogger::WARN]);
$this->connection = $this->reconnectToDatabase($this->connection, $output); $this->connection = $this->reconnectToDatabase($this->connection, $output);
$output->writeln('<info>Needed to reconnect to the database</info>'); $output->writeln('<info>Needed to reconnect to the database</info>');
$storageIds = $this->getStorageIds($mountId); $storages = $this->getStorageIds($mountId, $path);
} }
if (count($storageIds) === 0) { if (count($storages) === 0) {
throw new StorageNotAvailableException('No storages found by mount ID ' . $mountId); $output->writeln(" no users found with access to '$parent', skipping", OutputInterface::VERBOSITY_VERBOSE);
return;
} }
$storageIds = array_map('intval', $storageIds);
$result = $this->updateParent($storageIds, $parent); $users = array_map(function (array $storage) {
if ($result === 0) { return $storage['user_id'];
//TODO: Find existing parent further up the tree in the database and register that folder instead. }, $storages);
$this->logger->info('Failed updating parent for "' . $path . '" while trying to register change. It may not exist in the filecache.');
$output->writeln(" marking '$parent' as outdated for " . implode(', ', $users), OutputInterface::VERBOSITY_VERBOSE);
$storageIds = array_map(function (array $storage) {
return intval($storage['storage_id']);
}, $storages);
$storageIds = array_values(array_unique($storageIds));
if ($dryRun) {
$output->writeln(" dry-run: skipping database write");
} else {
$result = $this->updateParent($storageIds, $parent);
if ($result === 0) {
//TODO: Find existing parent further up the tree in the database and register that folder instead.
$this->logger->info('Failed updating parent for "' . $path . '" while trying to register change. It may not exist in the filecache.');
}
} }
} }
@ -256,21 +277,20 @@ class Notify extends Base {
$text .= ' to ' . $change->getTargetPath(); $text .= ' to ' . $change->getTargetPath();
} }
$output->writeln($text); $output->writeln($text, OutputInterface::VERBOSITY_VERBOSE);
} }
/** private function getStorageIds(int $mountId, string $path): array {
* @param int $mountId $pathHash = md5(trim((string)\OC_Util::normalizeUnicode($path), '/'));
* @return array
*/
private function getStorageIds($mountId) {
$qb = $this->connection->getQueryBuilder(); $qb = $this->connection->getQueryBuilder();
return $qb return $qb
->select('storage_id') ->select('storage_id', 'user_id')
->from('mounts') ->from('mounts', 'm')
->innerJoin('m', 'filecache', 'f', $qb->expr()->eq('m.storage_id', 'f.storage'))
->where($qb->expr()->eq('mount_id', $qb->createNamedParameter($mountId, IQueryBuilder::PARAM_INT))) ->where($qb->expr()->eq('mount_id', $qb->createNamedParameter($mountId, IQueryBuilder::PARAM_INT)))
->andWhere($qb->expr()->eq('path_hash', $qb->createNamedParameter($pathHash, IQueryBuilder::PARAM_STR)))
->execute() ->execute()
->fetchAll(\PDO::FETCH_COLUMN); ->fetchAll();
} }
/** /**
@ -312,7 +332,7 @@ class Notify extends Base {
} }
private function selfTest(IStorage $storage, INotifyHandler $notifyHandler, $verbose, OutputInterface $output) { private function selfTest(IStorage $storage, INotifyHandler $notifyHandler, OutputInterface $output) {
usleep(100 * 1000); //give time for the notify to start usleep(100 * 1000); //give time for the notify to start
$storage->file_put_contents('/.nc_test_file.txt', 'test content'); $storage->file_put_contents('/.nc_test_file.txt', 'test content');
$storage->mkdir('/.nc_test_folder'); $storage->mkdir('/.nc_test_folder');
@ -339,11 +359,11 @@ class Notify extends Base {
} }
} }
if ($foundRootChange && $foundSubfolderChange && $verbose) { if ($foundRootChange && $foundSubfolderChange) {
$output->writeln('<info>Self-test successful</info>'); $output->writeln('<info>Self-test successful</info>', OutputInterface::VERBOSITY_VERBOSE);
} elseif ($foundRootChange && !$foundSubfolderChange) { } elseif ($foundRootChange) {
$output->writeln('<error>Error while running self-test, change is subfolder not detected</error>'); $output->writeln('<error>Error while running self-test, change is subfolder not detected</error>');
} elseif (!$foundRootChange) { } else {
$output->writeln('<error>Error while running self-test, no changes detected</error>'); $output->writeln('<error>Error while running self-test, no changes detected</error>');
} }
} }