Skip to content

Commit

Permalink
fix(userstatus): catch unique constraint violation
Browse files Browse the repository at this point in the history
Signed-off-by: Anna Larch <anna@nextcloud.com>
  • Loading branch information
miaulalala committed Jan 16, 2024
1 parent 58b31a9 commit 020c8bf
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 2 deletions.
19 changes: 17 additions & 2 deletions apps/user_status/lib/Listener/UserLiveStatusListener.php
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,12 @@
use OCA\UserStatus\Service\StatusService;
use OCP\AppFramework\Db\DoesNotExistException;
use OCP\AppFramework\Utility\ITimeFactory;
use OCP\DB\Exception;
use OCP\EventDispatcher\Event;
use OCP\EventDispatcher\IEventListener;
use OCP\User\Events\UserLiveStatusEvent;
use OCP\UserStatus\IUserStatus;
use Psr\Log\LoggerInterface;

/**
* Class UserDeletedListener
Expand All @@ -50,7 +52,8 @@ class UserLiveStatusListener implements IEventListener {
public function __construct(UserStatusMapper $mapper,
StatusService $statusService,
ITimeFactory $timeFactory,
private CalendarStatusService $calendarStatusService) {
private CalendarStatusService $calendarStatusService,
private LoggerInterface $logger) {
$this->mapper = $mapper;
$this->statusService = $statusService;
$this->timeFactory = $timeFactory;
Expand Down Expand Up @@ -110,7 +113,19 @@ public function handle(Event $event): void {
$userStatus->setIsUserDefined(false);

if ($userStatus->getId() === null) {
$this->mapper->insert($userStatus);
try {
$this->mapper->insert($userStatus);
} catch (Exception $e) {
if ($e->getReason() === Exception::REASON_UNIQUE_CONSTRAINT_VIOLATION) {
// A different process might have written another status
// update to the DB while we're processing our stuff.
// We can safely ignore it as we're only changing between AWAY and ONLINE
// and not doing anything with the message or icon.
$this->logger->debug('Unique constraint violation for live user status', ['exception' => $e]);
return;
}
throw $e;
}
} else {
$this->mapper->update($userStatus);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@
use OCP\IUser;
use OCP\User\Events\UserLiveStatusEvent;
use PHPUnit\Framework\MockObject\MockObject;
use Psr\Log\LoggerInterface;
use Test\TestCase;

class UserLiveStatusListenerTest extends TestCase {
Expand All @@ -54,19 +55,23 @@ class UserLiveStatusListenerTest extends TestCase {

private CalendarStatusService|MockObject $calendarStatusService;

private LoggerInterface|MockObject $logger;

protected function setUp(): void {
parent::setUp();

$this->mapper = $this->createMock(UserStatusMapper::class);
$this->statusService = $this->createMock(StatusService::class);
$this->timeFactory = $this->createMock(ITimeFactory::class);
$this->calendarStatusService = $this->createMock(CalendarStatusService::class);
$this->logger = $this->createMock(LoggerInterface::class);

$this->listener = new UserLiveStatusListener(
$this->mapper,
$this->statusService,
$this->timeFactory,
$this->calendarStatusService,
$this->logger,
);
}

Expand Down

0 comments on commit 020c8bf

Please sign in to comment.