Log and continue on Dav reader failure
Signed-off-by: John Molakvoæ (skjnldsv) <skjnldsv@protonmail.com>
This commit is contained in:
parent
5e44699139
commit
2be63bcb6a
3 changed files with 32 additions and 12 deletions
|
@ -147,7 +147,7 @@ class Repair implements IOutput {
|
||||||
new AddPreviewBackgroundCleanupJob(\OC::$server->getJobList()),
|
new AddPreviewBackgroundCleanupJob(\OC::$server->getJobList()),
|
||||||
new AddCleanupUpdaterBackupsJob(\OC::$server->getJobList()),
|
new AddCleanupUpdaterBackupsJob(\OC::$server->getJobList()),
|
||||||
new RepairPendingCronJobs(\OC::$server->getDatabaseConnection(), \OC::$server->getConfig()),
|
new RepairPendingCronJobs(\OC::$server->getDatabaseConnection(), \OC::$server->getConfig()),
|
||||||
new SetVcardDatabaseUID(\OC::$server->getDatabaseConnection(), \OC::$server->getConfig())
|
new SetVcardDatabaseUID(\OC::$server->getDatabaseConnection(), \OC::$server->getConfig(), \OC::$server->getLogger())
|
||||||
];
|
];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -25,9 +25,11 @@ namespace OC\Repair\NC15;
|
||||||
|
|
||||||
use OCP\IConfig;
|
use OCP\IConfig;
|
||||||
use OCP\IDBConnection;
|
use OCP\IDBConnection;
|
||||||
|
use OCP\ILogger;
|
||||||
use OCP\Migration\IOutput;
|
use OCP\Migration\IOutput;
|
||||||
use OCP\Migration\IRepairStep;
|
use OCP\Migration\IRepairStep;
|
||||||
use Sabre\VObject\Reader;
|
use Sabre\VObject\Reader;
|
||||||
|
use Sabre\VObject\ParseException;
|
||||||
|
|
||||||
class SetVcardDatabaseUID implements IRepairStep {
|
class SetVcardDatabaseUID implements IRepairStep {
|
||||||
const MAX_ROWS = 1000;
|
const MAX_ROWS = 1000;
|
||||||
|
@ -38,11 +40,15 @@ class SetVcardDatabaseUID implements IRepairStep {
|
||||||
/** @var IConfig */
|
/** @var IConfig */
|
||||||
private $config;
|
private $config;
|
||||||
|
|
||||||
|
/** @var ILogger */
|
||||||
|
private $logger;
|
||||||
|
|
||||||
private $updateQuery;
|
private $updateQuery;
|
||||||
|
|
||||||
public function __construct(IDBConnection $connection, IConfig $config) {
|
public function __construct(IDBConnection $connection, IConfig $config, ILogger $logger) {
|
||||||
$this->connection = $connection;
|
$this->connection = $connection;
|
||||||
$this->config = $config;
|
$this->config = $config;
|
||||||
|
$this->logger = $logger;
|
||||||
}
|
}
|
||||||
|
|
||||||
public function getName() {
|
public function getName() {
|
||||||
|
@ -75,14 +81,21 @@ class SetVcardDatabaseUID implements IRepairStep {
|
||||||
* Extract UID from vcard
|
* Extract UID from vcard
|
||||||
*
|
*
|
||||||
* @param string $cardData the vcard raw data
|
* @param string $cardData the vcard raw data
|
||||||
|
* @param IOutput $output the output logger
|
||||||
* @return string the uid or empty if none
|
* @return string the uid or empty if none
|
||||||
*/
|
*/
|
||||||
private function getUID(string $cardData): string {
|
private function getUID(string $cardData, IOutput $output): string {
|
||||||
|
try {
|
||||||
$vCard = Reader::read($cardData);
|
$vCard = Reader::read($cardData);
|
||||||
if ($vCard->UID) {
|
if ($vCard->UID) {
|
||||||
$uid = $vCard->UID->getValue();
|
$uid = $vCard->UID->getValue();
|
||||||
|
|
||||||
return $uid;
|
return $uid;
|
||||||
}
|
}
|
||||||
|
} catch (ParseException $e) {
|
||||||
|
$output->warning('One vCard is broken. We logged the exception and will continue the repair.');
|
||||||
|
$this->logger->logException($e);
|
||||||
|
}
|
||||||
|
|
||||||
return '';
|
return '';
|
||||||
}
|
}
|
||||||
|
@ -106,7 +119,7 @@ class SetVcardDatabaseUID implements IRepairStep {
|
||||||
$this->updateQuery->execute();
|
$this->updateQuery->execute();
|
||||||
}
|
}
|
||||||
|
|
||||||
private function repair(): int {
|
private function repair(IOutput $output): int {
|
||||||
$this->connection->beginTransaction();
|
$this->connection->beginTransaction();
|
||||||
$entries = $this->getInvalidEntries();
|
$entries = $this->getInvalidEntries();
|
||||||
$count = 0;
|
$count = 0;
|
||||||
|
@ -116,7 +129,7 @@ class SetVcardDatabaseUID implements IRepairStep {
|
||||||
if (is_resource($cardData)) {
|
if (is_resource($cardData)) {
|
||||||
$cardData = stream_get_contents($cardData);
|
$cardData = stream_get_contents($cardData);
|
||||||
}
|
}
|
||||||
$uid = $this->getUID($cardData);
|
$uid = $this->getUID($cardData, $output);
|
||||||
$this->update($entry['id'], $uid);
|
$this->update($entry['id'], $uid);
|
||||||
}
|
}
|
||||||
$this->connection->commit();
|
$this->connection->commit();
|
||||||
|
@ -133,7 +146,7 @@ class SetVcardDatabaseUID implements IRepairStep {
|
||||||
|
|
||||||
public function run(IOutput $output) {
|
public function run(IOutput $output) {
|
||||||
if ($this->shouldRun()) {
|
if ($this->shouldRun()) {
|
||||||
$count = $this->repair();
|
$count = $this->repair($output);
|
||||||
|
|
||||||
$output->info('Fixed ' . $count . ' vcards');
|
$output->info('Fixed ' . $count . ' vcards');
|
||||||
}
|
}
|
||||||
|
|
|
@ -24,6 +24,8 @@
|
||||||
namespace Test\Repair;
|
namespace Test\Repair;
|
||||||
|
|
||||||
use OCP\IConfig;
|
use OCP\IConfig;
|
||||||
|
use OCP\ILogger;
|
||||||
|
use OCP\Migration\IOutput;
|
||||||
use OC\Repair\NC15\SetVcardDatabaseUID;
|
use OC\Repair\NC15\SetVcardDatabaseUID;
|
||||||
use Test\TestCase;
|
use Test\TestCase;
|
||||||
|
|
||||||
|
@ -38,11 +40,15 @@ class SetVcardDatabaseUIDTest extends TestCase {
|
||||||
/** @var IConfig */
|
/** @var IConfig */
|
||||||
private $config;
|
private $config;
|
||||||
|
|
||||||
|
/** @var Ilogger */
|
||||||
|
private $logger;
|
||||||
|
|
||||||
protected function setUp() {
|
protected function setUp() {
|
||||||
parent::setUp();
|
parent::setUp();
|
||||||
|
|
||||||
$this->config = $this->createMock(IConfig::class);
|
$this->config = $this->createMock(IConfig::class);
|
||||||
$this->repair = new SetVcardDatabaseUID(\OC::$server->getDatabaseConnection(), $this->config);
|
$this->logger = $this->createMock(Ilogger::class);
|
||||||
|
$this->repair = new SetVcardDatabaseUID(\OC::$server->getDatabaseConnection(), $this->config, $this->logger);
|
||||||
}
|
}
|
||||||
|
|
||||||
protected function tearDown() {
|
protected function tearDown() {
|
||||||
|
@ -86,7 +92,8 @@ class SetVcardDatabaseUIDTest extends TestCase {
|
||||||
* @param string|boolean $expected
|
* @param string|boolean $expected
|
||||||
*/
|
*/
|
||||||
public function testExtractUIDFromVcard($from, $expected) {
|
public function testExtractUIDFromVcard($from, $expected) {
|
||||||
$uid = $this->invokePrivate($this->repair, 'getUid', ['carddata' => $from]);
|
$output = $this->createMock(IOutput::class);
|
||||||
|
$uid = $this->invokePrivate($this->repair, 'getUid', ['carddata' => $from, 'output' => $output]);
|
||||||
$this->assertEquals($expected, $uid);
|
$this->assertEquals($expected, $uid);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue