blob: 53e6deba1cadaa35f68cf7ea36719077e5c95166 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
<?php
declare(strict_types=1);
/**
* SPDX-FileCopyrightText: 2021 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
namespace OCA\DAV\Migration;
use OCP\DB\Exception;
use OCP\IDBConnection;
use OCP\IUserManager;
use OCP\Migration\IOutput;
use OCP\Migration\IRepairStep;
class RemoveDeletedUsersCalendarSubscriptions implements IRepairStep {
/** @var IDBConnection */
private $connection;
/** @var IUserManager */
private $userManager;
/** @var int */
private $progress = 0;
/** @var int[] */
private $orphanSubscriptionIds = [];
private const SUBSCRIPTIONS_CHUNK_SIZE = 1000;
public function __construct(IDBConnection $connection, IUserManager $userManager) {
$this->connection = $connection;
$this->userManager = $userManager;
}
/**
* @inheritdoc
*/
public function getName(): string {
return 'Clean up old calendar subscriptions from deleted users that were not cleaned-up';
}
/**
* @inheritdoc
*/
public function run(IOutput $output) {
$nbSubscriptions = $this->countSubscriptions();
$output->startProgress($nbSubscriptions);
while ($this->progress < $nbSubscriptions) {
$this->checkSubscriptions();
$this->progress += self::SUBSCRIPTIONS_CHUNK_SIZE;
$output->advance(min(self::SUBSCRIPTIONS_CHUNK_SIZE, $nbSubscriptions));
}
$output->finishProgress();
$this->deleteOrphanSubscriptions();
$output->info(sprintf('%d calendar subscriptions without an user have been cleaned up', count($this->orphanSubscriptionIds)));
}
/**
* @throws Exception
*/
private function countSubscriptions(): int {
$qb = $this->connection->getQueryBuilder();
$query = $qb->select($qb->func()->count('*'))
->from('calendarsubscriptions');
$result = $query->execute();
$count = $result->fetchOne();
$result->closeCursor();
if ($count !== false) {
$count = (int)$count;
} else {
$count = 0;
}
return $count;
}
/**
* @throws Exception
*/
private function checkSubscriptions(): void {
$qb = $this->connection->getQueryBuilder();
$query = $qb->selectDistinct(['id', 'principaluri'])
->from('calendarsubscriptions')
->setMaxResults(self::SUBSCRIPTIONS_CHUNK_SIZE)
->setFirstResult($this->progress);
$result = $query->execute();
while ($row = $result->fetch()) {
$username = $this->getPrincipal($row['principaluri']);
if (!$this->userManager->userExists($username)) {
$this->orphanSubscriptionIds[] = (int) $row['id'];
}
}
$result->closeCursor();
}
/**
* @throws Exception
*/
private function deleteOrphanSubscriptions(): void {
foreach ($this->orphanSubscriptionIds as $orphanSubscriptionID) {
$this->deleteOrphanSubscription($orphanSubscriptionID);
}
}
/**
* @throws Exception
*/
private function deleteOrphanSubscription(int $orphanSubscriptionID): void {
$qb = $this->connection->getQueryBuilder();
$qb->delete('calendarsubscriptions')
->where($qb->expr()->eq('id', $qb->createNamedParameter($orphanSubscriptionID)))
->executeStatement();
}
private function getPrincipal(string $principalUri): string {
$uri = explode('/', $principalUri);
return array_pop($uri);
}
}
|
