OilfieldToolsNet/shlink
4
0
Fork 0
mirror of https://github.com/shlinkio/shlink.git synced 2024-11-22 00:47:25 -06:00
Code Issues Packages Projects Releases Wiki Activity

Wrap visits tracking in transaction

Browse Source
This commit is contained in:
Alejandro Celaya 2024-03-27 19:14:41 +01:00
parent cef30c8e2d
commit 4a05c4be40
2 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
module/Core/src/Visit/VisitsTracker.php
View File

@ -12,12 +12,12 @@ use Shlinkio\Shlink\Core\ShortUrl\Entity\ShortUrl;
use Shlinkio\Shlink\Core\Visit\Entity\Visit;
use Shlinkio\Shlink\Core\Visit\Model\Visitor;
class VisitsTracker implements VisitsTrackerInterface
readonly class VisitsTracker implements VisitsTrackerInterface
{
public function __construct(
private readonly ORM\EntityManagerInterface $em,
private readonly EventDispatcherInterface $eventDispatcher,
private readonly TrackingOptions $options,
private ORM\EntityManagerInterface $em,
private EventDispatcherInterface $eventDispatcher,
private TrackingOptions $options,
) {
}
@ -71,10 +71,12 @@ class VisitsTracker implements VisitsTrackerInterface
return;
}
$visit = $createVisit($visitor->normalizeForTrackingOptions($this->options));
$this->em->persist($visit);
$this->em->flush();
$this->em->wrapInTransaction(function () use ($createVisit, $visitor): void {
$visit = $createVisit($visitor->normalizeForTrackingOptions($this->options));
$this->em->persist($visit);
$this->em->flush();
$this->eventDispatcher->dispatch(new UrlVisited($visit->getId(), $visitor->remoteAddress));
$this->eventDispatcher->dispatch(new UrlVisited($visit->getId(), $visitor->remoteAddress));
});
}
}

2
module/Core/test/Visit/VisitsTrackerTest.php
View File

@ -25,6 +25,8 @@ class VisitsTrackerTest extends TestCase
protected function setUp(): void
{
$this->em = $this->createMock(EntityManager::class);
$this->em->method('wrapInTransaction')->willReturnCallback(fn (callable $callback) => $callback());
$this->eventDispatcher = $this->createMock(EventDispatcherInterface::class);
}