spreed/lib/Model/ThreadMapper.php
Anna Larch 1f35a31caa fix(threads): chunk thread query and shortcut mapper
call when param is empty

Signed-off-by: Anna Larch <anna@nextcloud.com>
2025-10-16 13:41:21 +02:00

126 lines
3.2 KiB
PHP

<?php
declare(strict_types=1);
/**
* SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
namespace OCA\Talk\Model;
use OCP\AppFramework\Db\DoesNotExistException;
use OCP\AppFramework\Db\QBMapper;
use OCP\DB\QueryBuilder\IQueryBuilder;
use OCP\IDBConnection;
/**
* @method Thread findEntity(IQueryBuilder $query)
* @method list<Thread> findEntities(IQueryBuilder $query)
* @template-extends QBMapper<Thread>
*/
class ThreadMapper extends QBMapper {
public function __construct(IDBConnection $db) {
parent::__construct($db, 'talk_threads', Thread::class);
}
/**
* @param non-negative-int $roomId
* @param non-negative-int $threadId
* @throws DoesNotExistException
*/
public function findById(int $roomId, int $threadId): Thread {
$query = $this->db->getQueryBuilder();
$query->select('*')
->from($this->getTableName())
->where($query->expr()->eq(
'id',
$query->createNamedParameter($threadId, IQueryBuilder::PARAM_INT),
IQueryBuilder::PARAM_INT,
))
->andWhere($query->expr()->eq(
'room_id',
$query->createNamedParameter($roomId, IQueryBuilder::PARAM_INT),
IQueryBuilder::PARAM_INT,
));
return $this->findEntity($query);
}
/**
* @param non-negative-int $roomId
* @param list<non-negative-int> $threadIds
* @return list<Thread>
*/
public function findByIds(int $roomId, array $threadIds): array {
$query = $this->db->getQueryBuilder();
$query->select('*')
->from($this->getTableName())
->where($query->expr()->in(
'id',
$query->createNamedParameter($threadIds, IQueryBuilder::PARAM_INT_ARRAY),
IQueryBuilder::PARAM_INT,
))
->andWhere($query->expr()->eq(
'room_id',
$query->createNamedParameter($roomId, IQueryBuilder::PARAM_INT),
IQueryBuilder::PARAM_INT,
));
return $this->findEntities($query);
}
/**
* @param list<non-negative-int> $threadIds
* @return list<Thread>
*/
public function getForIds(array $threadIds): array {
$threads = [];
$query = $this->db->getQueryBuilder();
$query->select('*')
->from($this->getTableName());
foreach (array_chunk($threadIds, 1000) as $ids) {
$query->where($query->expr()->in(
'id',
$query->createNamedParameter($ids, IQueryBuilder::PARAM_INT_ARRAY),
IQueryBuilder::PARAM_INT,
));
$threads[] = $this->findEntities($query);
}
return array_merge(...$threads);
}
/**
* @param int<1, 50> $limit
* @return list<Thread>
*/
public function getRecentByRoomId(int $roomId, int $limit): array {
$query = $this->db->getQueryBuilder();
$query->select('*')
->from($this->getTableName())
->where($query->expr()->eq(
'room_id',
$query->createNamedParameter($roomId, IQueryBuilder::PARAM_INT),
IQueryBuilder::PARAM_INT,
))
->orderBy('last_activity', 'DESC')
->setMaxResults($limit);
return $this->findEntities($query);
}
public function deleteByRoomId(int $roomId): int {
$query = $this->db->getQueryBuilder();
$query->delete($this->getTableName())
->where($query->expr()->eq(
'room_id',
$query->createNamedParameter($roomId, IQueryBuilder::PARAM_INT),
IQueryBuilder::PARAM_INT,
));
return $query->executeStatement();
}
}