count followers/following/posts on cache update

Signed-off-by: Maxence Lange <maxence@artificial-owl.com>

5 files changed, 112 insertions, 0 deletions

diff --git a/lib/Db/CoreRequestBuilder.php b/lib/Db/CoreRequestBuilder.php
index f6622803..130dedd7 100644
--- a/lib/Db/CoreRequestBuilder.php
+++ b/lib/Db/CoreRequestBuilder.php
@@ -196,6 +196,18 @@ class CoreRequestBuilder {
 
 
 	/**
+	 * Limit the request to the FollowId
+	 *
+	 * @param IQueryBuilder $qb
+	 * @param bool $accepted
+	 */
+	protected function limitToAccepted(IQueryBuilder &$qb, bool $accepted) {
+		$this->limitToDBField($qb, 'accepted', ($accepted) ? '1' : '0');
+
+	}
+
+
+	/**
 	 * Limit the request to the ServiceId
 	 *
 	 * @param IQueryBuilder $qb
@@ -273,6 +285,17 @@ class CoreRequestBuilder {
 
 
 	/**
+	 * Limit the request to the url
+	 *
+	 * @param IQueryBuilder $qb
+	 * @param string $actorId
+	 */
+	protected function limitToAttributedTo(IQueryBuilder &$qb, string $actorId) {
+		$this->limitToDBField($qb, 'attributed_to', $actorId, false);
+	}
+
+
+	/**
 	 * Limit the request to the status
 	 *
 	 * @param IQueryBuilder $qb
diff --git a/lib/Db/FollowsRequest.php b/lib/Db/FollowsRequest.php
index b4197b17..5b024414 100644
--- a/lib/Db/FollowsRequest.php
+++ b/lib/Db/FollowsRequest.php
@@ -31,6 +31,7 @@ declare(strict_types=1);
 namespace OCA\Social\Db;
 
 
+use daita\MySmallPhpTools\Traits\TArrayTools;
 use OCA\Social\Exceptions\FollowDoesNotExistException;
 use OCA\Social\Model\ActivityPub\Follow;
 
@@ -43,6 +44,9 @@ use OCA\Social\Model\ActivityPub\Follow;
 class FollowsRequest extends FollowsRequestBuilder {
 
 
+	use TArrayTools;
+
+
 	/**
 	 * Insert a new Note in the database.
 	 *
@@ -97,6 +101,42 @@ class FollowsRequest extends FollowsRequestBuilder {
 
 
 	/**
+	 * @param string $actorId
+	 *
+	 * @return int
+	 */
+	public function countFollowers(string $actorId): int {
+		$qb = $this->countFollowsSelectSql();
+		$this->limitToObjectId($qb, $actorId);
+		$this->limitToAccepted($qb, true);
+
+		$cursor = $qb->execute();
+		$data = $cursor->fetch();
+		$cursor->closeCursor();
+
+		return $this->getInt('count', $data, 0);
+	}
+
+
+	/**
+	 * @param string $actorId
+	 *
+	 * @return int
+	 */
+	public function countFollowing(string $actorId): int {
+		$qb = $this->countFollowsSelectSql();
+		$this->limitToActorId($qb, $actorId);
+		$this->limitToAccepted($qb, true);
+
+		$cursor = $qb->execute();
+		$data = $cursor->fetch();
+		$cursor->closeCursor();
+
+		return $this->getInt('count', $data, 0);
+	}
+
+
+	/**
 	 * @param string $followId
 	 *
 	 * @return Follow[]
diff --git a/lib/Db/FollowsRequestBuilder.php b/lib/Db/FollowsRequestBuilder.php
index 1947dd4c..7a6882c4 100644
--- a/lib/Db/FollowsRequestBuilder.php
+++ b/lib/Db/FollowsRequestBuilder.php
@@ -93,6 +93,22 @@ class FollowsRequestBuilder extends CoreRequestBuilder {
 
 
 	/**
+	 * Base of the Sql Select request for Shares
+	 *
+	 * @return IQueryBuilder
+	 */
+	protected function countFollowsSelectSql(): IQueryBuilder {
+		$qb = $this->dbConnection->getQueryBuilder();
+		$qb->selectAlias($qb->createFunction('COUNT(*)'), 'count')
+		   ->from(self::TABLE_SERVER_FOLLOWS, 'f');
+
+		$this->defaultSelectAlias = 'f';
+
+		return $qb;
+	}
+
+
+	/**
 	 * Base of the Sql Delete request
 	 *
 	 * @return IQueryBuilder
diff --git a/lib/Db/NotesRequest.php b/lib/Db/NotesRequest.php
index e2c7d9c8..f4bf6842 100644
--- a/lib/Db/NotesRequest.php
+++ b/lib/Db/NotesRequest.php
@@ -139,6 +139,23 @@ class NotesRequest extends NotesRequestBuilder {
 
 	/**
 	 * @param string $actorId
+	 *
+	 * @return int
+	 */
+	public function countNotesFromActorId(string $actorId): int {
+		$qb = $this->countNotesSelectSql();
+		$this->limitToAttributedTo($qb, $actorId);
+
+		$cursor = $qb->execute();
+		$data = $cursor->fetch();
+		$cursor->closeCursor();
+
+		return $this->getInt('count', $data, 0);
+	}
+
+
+	/**
+	 * @param string $actorId
 	 * @param int $since
 	 * @param int $limit
 	 *
diff --git a/lib/Db/NotesRequestBuilder.php b/lib/Db/NotesRequestBuilder.php
index 90ff24e1..e6f2cb18 100644
--- a/lib/Db/NotesRequestBuilder.php
+++ b/lib/Db/NotesRequestBuilder.php
@@ -93,6 +93,22 @@ class NotesRequestBuilder extends CoreRequestBuilder {
 
 
 	/**
+	 * Base of the Sql Select request for Shares
+	 *
+	 * @return IQueryBuilder
+	 */
+	protected function countNotesSelectSql(): IQueryBuilder {
+		$qb = $this->dbConnection->getQueryBuilder();
+		$qb->selectAlias($qb->createFunction('COUNT(*)'), 'count')
+		   ->from(self::TABLE_SERVER_NOTES, 'sn');
+
+		$this->defaultSelectAlias = 'sn';
+
+		return $qb;
+	}
+
+
+	/**
 	 * Base of the Sql Delete request
 	 *
 	 * @return IQueryBuilder