Add transaction around mass mounts operations

Signed-off-by: Joas Schilling <coding@schilljs.com>
nextcloud · Jul 8, 2022 · e4e8033 · e4e8033
1 parent 3dfb5c9
commit e4e8033
Showing 1 changed file with 19 additions and 12 deletions.
diff --git a/lib/private/Files/Config/UserMountCache.php b/lib/private/Files/Config/UserMountCache.php
@@ -116,18 +116,25 @@ public function registerMounts(IUser $user, array $mounts, array $mountProviderC
 
 		$changedMounts = $this->findChangedMounts($newMounts, $cachedMounts);
 
-		foreach ($addedMounts as $mount) {
-			$this->addToCache($mount);
-			/** @psalm-suppress InvalidArgument */
-			$this->mountsForUsers[$user->getUID()][] = $mount;
-		}
-		foreach ($removedMounts as $mount) {
-			$this->removeFromCache($mount);
-			$index = array_search($mount, $this->mountsForUsers[$user->getUID()]);
-			unset($this->mountsForUsers[$user->getUID()][$index]);
-		}
-		foreach ($changedMounts as $mount) {
-			$this->updateCachedMount($mount);
+		$this->connection->beginTransaction();
+		try {
+			foreach ($addedMounts as $mount) {
+				$this->addToCache($mount);
+				/** @psalm-suppress InvalidArgument */
+				$this->mountsForUsers[$user->getUID()][] = $mount;
+			}
+			foreach ($removedMounts as $mount) {
+				$this->removeFromCache($mount);
+				$index = array_search($mount, $this->mountsForUsers[$user->getUID()]);
+				unset($this->mountsForUsers[$user->getUID()][$index]);
+			}
+			foreach ($changedMounts as $mount) {
+				$this->updateCachedMount($mount);
+			}
+			$this->connection->commit();
+		} catch (\Throwable $e) {
+			$this->connection->rollBack();
+			throw $e;
 		}
 	}