nextcloud / server

lib/private/Files/Cache/Propagator.php

Indentation +179 added lines, -179 removed lines

@@ -22,183 +22,183 @@
 use Psr\Log\LoggerInterface;
 
 class Propagator implements IPropagator {
-	public const MAX_RETRIES = 3;
-
-	private bool $inBatch = false;
-	private array $batch = [];
-	private ClockInterface $clock;
-
-	public function __construct(
-		protected readonly IStorage $storage,
-		private readonly IDBConnection $connection,
-		private readonly array $ignore = [],
-	) {
-		$this->clock = Server::get(ClockInterface::class);
-	}
-
-	#[Override]
-	public function propagateChange(string $internalPath, int $time, int $sizeDifference = 0): void {
-		// Do not propagate changes in ignored paths
-		foreach ($this->ignore as $ignore) {
-			if (str_starts_with($internalPath, $ignore)) {
-				return;
-			}
-		}
-
-		$time = min($time, $this->clock->now()->getTimestamp());
-
-		$storageId = $this->storage->getCache()->getNumericStorageId();
-
-		$parents = $this->getParents($internalPath);
-
-		if ($this->inBatch) {
-			foreach ($parents as $parent) {
-				$this->addToBatch($parent, $time, $sizeDifference);
-			}
-			return;
-		}
-
-		$parentHashes = array_map('md5', $parents);
-		$etag = uniqid(); // since we give all folders the same etag we don't ask the storage for the etag
-
-		$builder = $this->connection->getQueryBuilder();
-		$hashParams = array_map(function ($hash) use ($builder) {
-			return $builder->expr()->literal($hash);
-		}, $parentHashes);
-
-		$builder->update('filecache')
-			->set('mtime', $builder->func()->greatest('mtime', $builder->createNamedParameter($time, IQueryBuilder::PARAM_INT)))
-			->where($builder->expr()->eq('storage', $builder->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
-			->andWhere($builder->expr()->in('path_hash', $hashParams));
-		if (!$this->storage->instanceOfStorage(IReliableEtagStorage::class)) {
-			$builder->set('etag', $builder->createNamedParameter($etag, IQueryBuilder::PARAM_STR));
-		}
-
-		if ($sizeDifference !== 0) {
-			$hasCalculatedSize = $builder->expr()->gt('size', $builder->expr()->literal(-1, IQUeryBuilder::PARAM_INT));
-			$sizeColumn = $builder->getColumnName('size');
-			$newSize = $builder->func()->greatest(
-				$builder->func()->add('size', $builder->createNamedParameter($sizeDifference)),
-				$builder->createNamedParameter(-1, IQueryBuilder::PARAM_INT)
-			);
-
-			// Only update if row had a previously calculated size
-			$builder->set('size', $builder->createFunction("CASE WHEN $hasCalculatedSize THEN $newSize ELSE $sizeColumn END"));
-
-			if ($this->storage->instanceOfStorage(Encryption::class)) {
-				// in case of encryption being enabled after some files are already uploaded, some entries will have an unencrypted_size of 0 and a non-zero size
-				$hasUnencryptedSize = $builder->expr()->neq('unencrypted_size', $builder->expr()->literal(0, IQueryBuilder::PARAM_INT));
-				$sizeColumn = $builder->getColumnName('size');
-				$unencryptedSizeColumn = $builder->getColumnName('unencrypted_size');
-				$newUnencryptedSize = $builder->func()->greatest(
-					$builder->func()->add(
-						$builder->createFunction("CASE WHEN $hasUnencryptedSize THEN $unencryptedSizeColumn ELSE $sizeColumn END"),
-						$builder->createNamedParameter($sizeDifference)
-					),
-					$builder->createNamedParameter(-1, IQueryBuilder::PARAM_INT)
-				);
-
-				// Only update if row had a previously calculated size
-				$builder->set('unencrypted_size', $builder->createFunction("CASE WHEN $hasCalculatedSize THEN $newUnencryptedSize ELSE $unencryptedSizeColumn END"));
-			}
-		}
-
-		for ($i = 0; $i < self::MAX_RETRIES; $i++) {
-			try {
-				$builder->executeStatement();
-				break;
-			} catch (DbalException $e) {
-				if (!$e->isRetryable()) {
-					throw $e;
-				}
-
-				/** @var LoggerInterface $loggerInterface */
-				$loggerInterface = \OCP\Server::get(LoggerInterface::class);
-				$loggerInterface->warning('Retrying propagation query after retryable exception.', [ 'exception' => $e ]);
-			}
-		}
-	}
-
-	/**
-	 * @return string[]
-	 */
-	protected function getParents(string $path): array {
-		$parts = explode('/', $path);
-		$parent = '';
-		$parents = [];
-		foreach ($parts as $part) {
-			$parents[] = $parent;
-			$parent = trim($parent . '/' . $part, '/');
-		}
-		return $parents;
-	}
-
-	#[Override]
-	public function beginBatch(): void {
-		$this->inBatch = true;
-	}
-
-	private function addToBatch(string $internalPath, int $time, int $sizeDifference): void {
-		if (!isset($this->batch[$internalPath])) {
-			$this->batch[$internalPath] = [
-				'hash' => md5($internalPath),
-				'time' => $time,
-				'size' => $sizeDifference,
-			];
-		} else {
-			$this->batch[$internalPath]['size'] += $sizeDifference;
-			if ($time > $this->batch[$internalPath]['time']) {
-				$this->batch[$internalPath]['time'] = $time;
-			}
-		}
-	}
-
-	#[Override]
-	public function commitBatch(): void {
-		if (!$this->inBatch) {
-			throw new \BadMethodCallException('Not in batch');
-		}
-		$this->inBatch = false;
-
-		try {
-			$this->connection->beginTransaction();
-
-			$query = $this->connection->getQueryBuilder();
-			$storageId = $this->storage->getCache()->getNumericStorageId();
-
-			$query->update('filecache')
-				->set('mtime', $query->func()->greatest('mtime', $query->createParameter('time')))
-				->set('etag', $query->expr()->literal(uniqid()))
-				->where($query->expr()->eq('storage', $query->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
-				->andWhere($query->expr()->eq('path_hash', $query->createParameter('hash')));
-
-			$sizeQuery = $this->connection->getQueryBuilder();
-			$sizeQuery->update('filecache')
-				->set('size', $sizeQuery->func()->add('size', $sizeQuery->createParameter('size')))
-				->where($query->expr()->eq('storage', $sizeQuery->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
-				->andWhere($query->expr()->eq('path_hash', $sizeQuery->createParameter('hash')))
-				->andWhere($sizeQuery->expr()->gt('size', $sizeQuery->createNamedParameter(-1, IQueryBuilder::PARAM_INT)));
-
-			foreach ($this->batch as $item) {
-				$query->setParameter('time', $item['time'], IQueryBuilder::PARAM_INT);
-				$query->setParameter('hash', $item['hash']);
-
-				$query->executeStatement();
-
-				if ($item['size']) {
-					$sizeQuery->setParameter('size', $item['size'], IQueryBuilder::PARAM_INT);
-					$sizeQuery->setParameter('hash', $item['hash']);
-
-					$sizeQuery->executeStatement();
-				}
-			}
-
-			$this->batch = [];
-
-			$this->connection->commit();
-		} catch (\Exception $e) {
-			$this->connection->rollback();
-			throw $e;
-		}
-	}
+    public const MAX_RETRIES = 3;
+
+    private bool $inBatch = false;
+    private array $batch = [];
+    private ClockInterface $clock;
+
+    public function __construct(
+        protected readonly IStorage $storage,
+        private readonly IDBConnection $connection,
+        private readonly array $ignore = [],
+    ) {
+        $this->clock = Server::get(ClockInterface::class);
+    }
+
+    #[Override]
+    public function propagateChange(string $internalPath, int $time, int $sizeDifference = 0): void {
+        // Do not propagate changes in ignored paths
+        foreach ($this->ignore as $ignore) {
+            if (str_starts_with($internalPath, $ignore)) {
+                return;
+            }
+        }
+
+        $time = min($time, $this->clock->now()->getTimestamp());
+
+        $storageId = $this->storage->getCache()->getNumericStorageId();
+
+        $parents = $this->getParents($internalPath);
+
+        if ($this->inBatch) {
+            foreach ($parents as $parent) {
+                $this->addToBatch($parent, $time, $sizeDifference);
+            }
+            return;
+        }
+
+        $parentHashes = array_map('md5', $parents);
+        $etag = uniqid(); // since we give all folders the same etag we don't ask the storage for the etag
+
+        $builder = $this->connection->getQueryBuilder();
+        $hashParams = array_map(function ($hash) use ($builder) {
+            return $builder->expr()->literal($hash);
+        }, $parentHashes);
+
+        $builder->update('filecache')
+            ->set('mtime', $builder->func()->greatest('mtime', $builder->createNamedParameter($time, IQueryBuilder::PARAM_INT)))
+            ->where($builder->expr()->eq('storage', $builder->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
+            ->andWhere($builder->expr()->in('path_hash', $hashParams));
+        if (!$this->storage->instanceOfStorage(IReliableEtagStorage::class)) {
+            $builder->set('etag', $builder->createNamedParameter($etag, IQueryBuilder::PARAM_STR));
+        }
+
+        if ($sizeDifference !== 0) {
+            $hasCalculatedSize = $builder->expr()->gt('size', $builder->expr()->literal(-1, IQUeryBuilder::PARAM_INT));
+            $sizeColumn = $builder->getColumnName('size');
+            $newSize = $builder->func()->greatest(
+                $builder->func()->add('size', $builder->createNamedParameter($sizeDifference)),
+                $builder->createNamedParameter(-1, IQueryBuilder::PARAM_INT)
+            );
+
+            // Only update if row had a previously calculated size
+            $builder->set('size', $builder->createFunction("CASE WHEN $hasCalculatedSize THEN $newSize ELSE $sizeColumn END"));
+
+            if ($this->storage->instanceOfStorage(Encryption::class)) {
+                // in case of encryption being enabled after some files are already uploaded, some entries will have an unencrypted_size of 0 and a non-zero size
+                $hasUnencryptedSize = $builder->expr()->neq('unencrypted_size', $builder->expr()->literal(0, IQueryBuilder::PARAM_INT));
+                $sizeColumn = $builder->getColumnName('size');
+                $unencryptedSizeColumn = $builder->getColumnName('unencrypted_size');
+                $newUnencryptedSize = $builder->func()->greatest(
+                    $builder->func()->add(
+                        $builder->createFunction("CASE WHEN $hasUnencryptedSize THEN $unencryptedSizeColumn ELSE $sizeColumn END"),
+                        $builder->createNamedParameter($sizeDifference)
+                    ),
+                    $builder->createNamedParameter(-1, IQueryBuilder::PARAM_INT)
+                );
+
+                // Only update if row had a previously calculated size
+                $builder->set('unencrypted_size', $builder->createFunction("CASE WHEN $hasCalculatedSize THEN $newUnencryptedSize ELSE $unencryptedSizeColumn END"));
+            }
+        }
+
+        for ($i = 0; $i < self::MAX_RETRIES; $i++) {
+            try {
+                $builder->executeStatement();
+                break;
+            } catch (DbalException $e) {
+                if (!$e->isRetryable()) {
+                    throw $e;
+                }
+
+                /** @var LoggerInterface $loggerInterface */
+                $loggerInterface = \OCP\Server::get(LoggerInterface::class);
+                $loggerInterface->warning('Retrying propagation query after retryable exception.', [ 'exception' => $e ]);
+            }
+        }
+    }
+
+    /**
+     * @return string[]
+     */
+    protected function getParents(string $path): array {
+        $parts = explode('/', $path);
+        $parent = '';
+        $parents = [];
+        foreach ($parts as $part) {
+            $parents[] = $parent;
+            $parent = trim($parent . '/' . $part, '/');
+        }
+        return $parents;
+    }
+
+    #[Override]
+    public function beginBatch(): void {
+        $this->inBatch = true;
+    }
+
+    private function addToBatch(string $internalPath, int $time, int $sizeDifference): void {
+        if (!isset($this->batch[$internalPath])) {
+            $this->batch[$internalPath] = [
+                'hash' => md5($internalPath),
+                'time' => $time,
+                'size' => $sizeDifference,
+            ];
+        } else {
+            $this->batch[$internalPath]['size'] += $sizeDifference;
+            if ($time > $this->batch[$internalPath]['time']) {
+                $this->batch[$internalPath]['time'] = $time;
+            }
+        }
+    }
+
+    #[Override]
+    public function commitBatch(): void {
+        if (!$this->inBatch) {
+            throw new \BadMethodCallException('Not in batch');
+        }
+        $this->inBatch = false;
+
+        try {
+            $this->connection->beginTransaction();
+
+            $query = $this->connection->getQueryBuilder();
+            $storageId = $this->storage->getCache()->getNumericStorageId();
+
+            $query->update('filecache')
+                ->set('mtime', $query->func()->greatest('mtime', $query->createParameter('time')))
+                ->set('etag', $query->expr()->literal(uniqid()))
+                ->where($query->expr()->eq('storage', $query->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
+                ->andWhere($query->expr()->eq('path_hash', $query->createParameter('hash')));
+
+            $sizeQuery = $this->connection->getQueryBuilder();
+            $sizeQuery->update('filecache')
+                ->set('size', $sizeQuery->func()->add('size', $sizeQuery->createParameter('size')))
+                ->where($query->expr()->eq('storage', $sizeQuery->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
+                ->andWhere($query->expr()->eq('path_hash', $sizeQuery->createParameter('hash')))
+                ->andWhere($sizeQuery->expr()->gt('size', $sizeQuery->createNamedParameter(-1, IQueryBuilder::PARAM_INT)));
+
+            foreach ($this->batch as $item) {
+                $query->setParameter('time', $item['time'], IQueryBuilder::PARAM_INT);
+                $query->setParameter('hash', $item['hash']);
+
+                $query->executeStatement();
+
+                if ($item['size']) {
+                    $sizeQuery->setParameter('size', $item['size'], IQueryBuilder::PARAM_INT);
+                    $sizeQuery->setParameter('hash', $item['hash']);
+
+                    $sizeQuery->executeStatement();
+                }
+            }
+
+            $this->batch = [];
+
+            $this->connection->commit();
+        } catch (\Exception $e) {
+            $this->connection->rollback();
+            throw $e;
+        }
+    }
 }