perf(comments): Also cache the comments count

Since we now have an easy way to fetch the comments count.

Signed-off-by: Carl Schwan <carl.schwan@nextclound.com>

Author: Carl Schwan

apps/dav/lib/Connector/Sabre/CommentPropertiesPlugin.php

@@ -22,6 +22,7 @@ class CommentPropertiesPlugin extends ServerPlugin {
 
 	protected ?Server $server = null;
 	private array $cachedUnreadCount = [];
+	private array $cachedCount = [];
 	private array $cachedDirectories = [];
 
 	public function __construct(
@@ -48,7 +49,11 @@ public function initialize(\Sabre\DAV\Server $server) {
 		$this->server->on('propFind', [$this, 'handleGetProperties']);
 	}
 
-	private function cacheDirectory(Directory $directory): void {
+	private function cacheDirectory(Directory $directory, PropFind $propFind): void {
+		if (is_null($propFind->getStatus(self::PROPERTY_NAME_UNREAD)) && is_null($propFind->getStatus(self::PROPERTY_NAME_COUNT))) {
+			return;
+		}
+
 		$children = $directory->getChildren();
 
 		$ids = [];
@@ -66,15 +71,30 @@ private function cacheDirectory(Directory $directory): void {
 		}
 
 		$ids[] = (string)$directory->getId();
-		$unread = $this->commentsManager->getNumberOfUnreadCommentsForObjects('files', $ids, $this->userSession->getUser());
+		if (!is_null($propFind->getStatus(self::PROPERTY_NAME_UNREAD))) {
+			$user = $this->userSession->getUser();
+			if ($user) {
+				$unread = $this->commentsManager->getNumberOfUnreadCommentsForObjects('files', $ids, $user);
+				foreach ($unread as $id => $count) {
+					$this->cachedUnreadCount[(int)$id] = $count;
+				}
+			} else {
+				foreach ($ids as $id) {
+					$this->cachedUnreadCount[(int)$id] = null;
+				}
+			}
+		}
 
-		foreach ($unread as $id => $count) {
-			$this->cachedUnreadCount[(int)$id] = $count;
+		if (!is_null($propFind->getStatus(self::PROPERTY_NAME_COUNT))) {
+			$commentCounts = $this->commentsManager->getNumberOfCommentsForObjects('files', $ids);
+			foreach ($commentCounts as $id => $count) {
+				$this->cachedCount[(int)$id] = $count;
+			}
 		}
+
 	}
 
-	private function preloadCollection(PropFind $propFind, ICollection $collection):
-	void {
+	private function preloadCollection(PropFind $propFind, ICollection $collection): void {
 		if (!($collection instanceof Directory)) {
 			return;
 		}
@@ -83,7 +103,7 @@ private function preloadCollection(PropFind $propFind, ICollection $collection):
 		if (!isset($this->cachedDirectories[$collectionPath]) && $propFind->getStatus(
 			self::PROPERTY_NAME_UNREAD
 		) !== null) {
-			$this->cacheDirectory($collection);
+			$this->cacheDirectory($collection, $propFind);
 			$this->cachedDirectories[$collectionPath] = true;
 		}
 	}
@@ -105,7 +125,7 @@ public function handleGetProperties(
 		}
 
 		$propFind->handle(self::PROPERTY_NAME_COUNT, function () use ($node): int {
-			return $this->commentsManager->getNumberOfCommentsForObject('files', (string)$node->getId());
+			return $this->cachedCount[$node->getId()] ?? $this->commentsManager->getNumberOfCommentsForObject('files', (string)$node->getId());
 		});
 
 		$propFind->handle(self::PROPERTY_NAME_HREF, function () use ($node): ?string {
@@ -141,8 +161,7 @@ public function getUnreadCount(Node $node): ?int {
 			return null;
 		}
 
-		$lastRead = $this->commentsManager->getReadMark('files', (string)$node->getId(), $user);
-
-		return $this->commentsManager->getNumberOfCommentsForObject('files', (string)$node->getId(), $lastRead);
+		$objectId = (string)$node->getId();
+		return $this->commentsManager->getNumberOfUnreadCommentsForObjects('files', [$objectId], $user)[$objectId];
 	}
 }

apps/dav/tests/unit/Connector/Sabre/CommentsPropertiesPluginTest.php

@@ -104,8 +104,8 @@ public function testGetUnreadCount(?string $user): void {
 			->willReturn($user);
 
 		$this->commentsManager->expects($this->any())
-			->method('getNumberOfCommentsForObject')
-			->willReturn(42);
+			->method('getNumberOfUnreadCommentsForObjects')
+			->willReturn(['4567' => 42]);
 
 		$unread = $this->plugin->getUnreadCount($node);
 		if (is_null($user)) {

lib/private/Comments/Manager.php

@@ -632,7 +632,7 @@ public function getNumberOfCommentsForObject($objectType, $objectId, ?\DateTime
 	/** @inheritDoc */
 	public function getNumberOfCommentsForObjects(string $objectType, array $objectIds, ?\DateTime $notOlderThan = null, string $verb = ''): array {
 		$qb = $this->dbConn->getQueryBuilder();
-		$query = $qb->select($qb->func()->count('id'), 'object_id')
+		$query = $qb->select($qb->func()->count('id', 'num_comments'), 'object_id')
 			->from('comments')
 			->where($qb->expr()->eq('object_type', $qb->createNamedParameter($objectType, IQueryBuilder::PARAM_STR)))
 			->andWhere($qb->expr()->in('object_id', $qb->createNamedParameter($objectIds, IQueryBuilder::PARAM_STR_ARRAY)));
@@ -651,7 +651,7 @@ public function getNumberOfCommentsForObjects(string $objectType, array $objectI
 		$comments = array_fill_keys($objectIds, 0);
 		$resultStatement = $query->execute();
 		while ($data = $resultStatement->fetch()) {
-			$comments[$data[1]] = (int)$data[0];
+			$comments[$data['object_id']] = (int)$data['num_comments'];
 		}
 		$resultStatement->closeCursor();
 

lib/public/Comments/ICommentsManager.php

@@ -191,7 +191,7 @@ public function getNumberOfCommentsForObject($objectType, $objectId, ?\DateTime
 	 * @param $objectIds string[] the ids of the object
 	 * @param \DateTime|null $notOlderThan optional, timestamp of the oldest comments
 	 *                                     that may be returned
-	 * @param string $verb Limit the verb of the comment - Added in 14.0.0
+	 * @param string $verb Limit the verb of the comment
 	 * @return array<string, int>
 	 * @since 32.0.0
 	 */