]>
Commit | Line | Data |
---|---|---|
1 | <?php | |
2 | ||
3 | namespace Wallabag\CoreBundle\Repository; | |
4 | ||
5 | use Doctrine\ORM\EntityRepository; | |
6 | use Doctrine\ORM\QueryBuilder; | |
7 | use Wallabag\CoreBundle\Entity\Tag; | |
8 | ||
9 | class TagRepository extends EntityRepository | |
10 | { | |
11 | /** | |
12 | * Count all tags per user. | |
13 | * | |
14 | * @param int $userId | |
15 | * @param int $cacheLifeTime Duration of the cache for this query | |
16 | * | |
17 | * @return int | |
18 | */ | |
19 | public function countAllTags($userId, $cacheLifeTime = null) | |
20 | { | |
21 | $query = $this->createQueryBuilder('t') | |
22 | ->select('t.slug') | |
23 | ->leftJoin('t.entries', 'e') | |
24 | ->where('e.user = :userId')->setParameter('userId', $userId) | |
25 | ->groupBy('t.slug') | |
26 | ->getQuery(); | |
27 | ||
28 | if (null !== $cacheLifeTime) { | |
29 | $query->useQueryCache(true); | |
30 | $query->useResultCache(true); | |
31 | $query->setResultCacheLifetime($cacheLifeTime); | |
32 | } | |
33 | ||
34 | return \count($query->getArrayResult()); | |
35 | } | |
36 | ||
37 | /** | |
38 | * Find all tags per user. | |
39 | * Instead of just left joined on the Entry table, we select only id and group by id to avoid tag multiplication in results. | |
40 | * Once we have all tags id, we can safely request them one by one. | |
41 | * This'll still be fastest than the previous query. | |
42 | * | |
43 | * @param int $userId | |
44 | * | |
45 | * @return array | |
46 | */ | |
47 | public function findAllTags($userId) | |
48 | { | |
49 | $ids = $this->getQueryBuilderByUser($userId) | |
50 | ->select('t.id') | |
51 | ->getQuery() | |
52 | ->getArrayResult(); | |
53 | ||
54 | $tags = []; | |
55 | foreach ($ids as $id) { | |
56 | $tags[] = $this->find($id); | |
57 | } | |
58 | ||
59 | return $tags; | |
60 | } | |
61 | ||
62 | /** | |
63 | * Find all tags (flat) per user with nb entries. | |
64 | * | |
65 | * @param int $userId | |
66 | * | |
67 | * @return array | |
68 | */ | |
69 | public function findAllFlatTagsWithNbEntries($userId) | |
70 | { | |
71 | return $this->getQueryBuilderByUser($userId) | |
72 | ->select('t.id, t.label, t.slug, count(e.id) as nbEntries') | |
73 | ->distinct(true) | |
74 | ->getQuery() | |
75 | ->getArrayResult(); | |
76 | } | |
77 | ||
78 | public function findByLabelsAndUser($labels, $userId) | |
79 | { | |
80 | $qb = $this->getQueryBuilderByUser($userId) | |
81 | ->select('t.id'); | |
82 | ||
83 | $ids = $qb->andWhere($qb->expr()->in('t.label', $labels)) | |
84 | ->getQuery() | |
85 | ->getArrayResult(); | |
86 | ||
87 | $tags = []; | |
88 | foreach ($ids as $id) { | |
89 | $tags[] = $this->find($id); | |
90 | } | |
91 | ||
92 | return $tags; | |
93 | } | |
94 | ||
95 | /** | |
96 | * Used only in test case to get a tag for our entry. | |
97 | * | |
98 | * @return Tag | |
99 | */ | |
100 | public function findOneByEntryAndTagLabel($entry, $label) | |
101 | { | |
102 | return $this->createQueryBuilder('t') | |
103 | ->leftJoin('t.entries', 'e') | |
104 | ->where('e.id = :entryId')->setParameter('entryId', $entry->getId()) | |
105 | ->andWhere('t.label = :label')->setParameter('label', $label) | |
106 | ->setMaxResults(1) | |
107 | ->getQuery() | |
108 | ->getSingleResult(); | |
109 | } | |
110 | ||
111 | public function findForArchivedArticlesByUser($userId) | |
112 | { | |
113 | $ids = $this->getQueryBuilderByUser($userId) | |
114 | ->select('t.id') | |
115 | ->andWhere('e.isArchived = true') | |
116 | ->getQuery() | |
117 | ->getArrayResult(); | |
118 | ||
119 | $tags = []; | |
120 | foreach ($ids as $id) { | |
121 | $tags[] = $this->find($id); | |
122 | } | |
123 | ||
124 | return $tags; | |
125 | } | |
126 | ||
127 | /** | |
128 | * Retrieve a sorted list of tags used by a user. | |
129 | * | |
130 | * @param int $userId | |
131 | * | |
132 | * @return QueryBuilder | |
133 | */ | |
134 | private function getQueryBuilderByUser($userId) | |
135 | { | |
136 | return $this->createQueryBuilder('t') | |
137 | ->leftJoin('t.entries', 'e') | |
138 | ->where('e.user = :userId')->setParameter('userId', $userId) | |
139 | ->groupBy('t.id') | |
140 | ->orderBy('t.slug'); | |
141 | } | |
142 | } |