From 510647a883fd2877768ecc58e9ff4bebc99e3886 Mon Sep 17 00:00:00 2001
From: Jeff Brown <thebrillpuff@gmail.com>
Date: Thu, 20 Aug 2020 23:26:44 -0400
Subject: [PATCH] Improve performance querying group_folders_manage

Signed-off-by: Jeff Brown <thebrillpuff@gmail.com>
Co-authored-by: Robin Appelman <robin@icewind.nl>
---
 lib/Folder/FolderManager.php | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/lib/Folder/FolderManager.php b/lib/Folder/FolderManager.php
index ed05f18b2..2ed924034 100644
--- a/lib/Folder/FolderManager.php
+++ b/lib/Folder/FolderManager.php
@@ -110,9 +110,12 @@ public function getAllFoldersWithSize($rootStorageId) {
 
 		$rows = $query->execute()->fetchAll();
 
+		$folderMappings = $this->getAllFolderMappings();
+
 		$folderMap = [];
 		foreach ($rows as $row) {
 			$id = (int)$row['folder_id'];
+			$mappings = $folderMappings[$id] ?? [];
 			$folderMap[$id] = [
 				'id' => $id,
 				'mount_point' => $row['mount_point'],
@@ -120,19 +123,35 @@ public function getAllFoldersWithSize($rootStorageId) {
 				'quota' => $row['quota'],
 				'size' => $row['size'] ? $row['size'] : 0,
 				'acl' => (bool)$row['acl'],
-				'manage' => $this->getManageAcl($id)
+				'manage' => $this->getManageAcl($mappings)
 			];
 		}
 
 		return $folderMap;
 	}
 
-	private function getManageAcl($folderId) {
+	private function getAllFolderMappings() {
 		$query = $this->connection->getQueryBuilder();
 		$query->select('*')
-			->from('group_folders_manage')
-			->where($query->expr()->eq('folder_id', $query->createNamedParameter($folderId)));
-		$result =  $query->execute()->fetchAll();
+			->from('group_folders_manage');
+		$rows = $query->execute()->fetchAll();
+
+		$folderMap = [];
+		foreach ($rows as $row) {
+			$id = (int)$row['folder_id'];
+			
+			if (!isset($folderMap[$id])) {
+				$folderMap[$id] = [$row];
+			}
+			else {
+				$folderMap[$id][] = $row;
+			}
+		}
+
+		return $folderMap;
+	}
+
+	private function getManageAcl($mappings) {
 		return array_filter(array_map(function ($entry) {
 			if ($entry['mapping_type'] === 'user') {
 				$user = \OC::$server->getUserManager()->get($entry['mapping_id']);
@@ -154,7 +173,7 @@ private function getManageAcl($folderId) {
 				'id' => $group->getGID(),
 				'displayname' => $group->getDisplayName()
 			];
-		}, $result), function($element) { return $element !== null; });
+		}, $mappings), function($element) { return $element !== null; });
 	}
 
 	public function getFolder($id, $rootStorageId) {