Moved visits iteration logic from command to service to allow lazy loading of entries in resultset

module/Core/src/Exception/IpCannotBeLocatedException.php

@@ -0,0 +1,8 @@
+<?php
+declare(strict_types=1);
+
+namespace Shlinkio\Shlink\Core\Exception;
+
+class IpCannotBeLocatedException extends RuntimeException
+{
+}

module/Core/src/Repository/VisitRepository.php

@@ -10,15 +10,12 @@ use Shlinkio\Shlink\Core\Entity\Visit;
 
 class VisitRepository extends EntityRepository implements VisitRepositoryInterface
 {
-    /**
-     * @return Visit[]
-     */
-    public function findUnlocatedVisits(): array
+    public function findUnlocatedVisits(): iterable
     {
         $qb = $this->createQueryBuilder('v');
         $qb->where($qb->expr()->isNull('v.visitLocation'));
 
-        return $qb->getQuery()->getResult();
+        return $qb->getQuery()->iterate();
     }
 
     /**

module/Core/src/Repository/VisitRepositoryInterface.php

@@ -10,10 +10,7 @@ use Shlinkio\Shlink\Core\Entity\Visit;
 
 interface VisitRepositoryInterface extends ObjectRepository
 {
-    /**
-     * @return Visit[]
-     */
-    public function findUnlocatedVisits(): array;
+    public function findUnlocatedVisits(): iterable;
 
     /**
      * @param ShortUrl|int $shortUrl

module/Core/src/Service/VisitService.php

@@ -6,6 +6,7 @@ namespace Shlinkio\Shlink\Core\Service;
 use Doctrine\ORM\EntityManagerInterface;
 use Shlinkio\Shlink\Core\Entity\Visit;
 use Shlinkio\Shlink\Core\Entity\VisitLocation;
+use Shlinkio\Shlink\Core\Exception\IpCannotBeLocatedException;
 use Shlinkio\Shlink\Core\Repository\VisitRepository;
 
 class VisitService implements VisitServiceInterface
@@ -20,26 +21,36 @@ class VisitService implements VisitServiceInterface
         $this->em = $em;
     }
 
-    /**
-     * @return Visit[]
-     */
-    public function getUnlocatedVisits(): array
+    public function locateVisits(callable $getGeolocationData, ?callable $locatedVisit = null): void
     {
         /** @var VisitRepository $repo */
         $repo = $this->em->getRepository(Visit::class);
-        return $repo->findUnlocatedVisits();
+        $results = $repo->findUnlocatedVisits();
+
+        foreach ($results as [$visit]) {
+            try {
+                $locationData = $getGeolocationData($visit);
+            } catch (IpCannotBeLocatedException $e) {
+                // Skip if the visit's IP could not be located
+                continue;
+            }
+
+            $location = new VisitLocation($locationData);
+            $this->locateVisit($visit, $location, $locatedVisit);
+        }
     }
 
-    public function locateVisit(Visit $visit, VisitLocation $location, bool $clear = false): void
+    private function locateVisit(Visit $visit, VisitLocation $location, ?callable $locatedVisit): void
     {
         $visit->locate($location);
 
         $this->em->persist($visit);
         $this->em->flush();
 
-        if ($clear) {
-            $this->em->clear(VisitLocation::class);
-            $this->em->clear(Visit::class);
+        if ($locatedVisit !== null) {
+            $locatedVisit($location, $visit);
         }
+
+        $this->em->clear();
     }
 }

module/Core/src/Service/VisitServiceInterface.php

@@ -3,15 +3,7 @@ declare(strict_types=1);
 
 namespace Shlinkio\Shlink\Core\Service;
 
-use Shlinkio\Shlink\Core\Entity\Visit;
-use Shlinkio\Shlink\Core\Entity\VisitLocation;
-
 interface VisitServiceInterface
 {
-    /**
-     * @return Visit[]
-     */
-    public function getUnlocatedVisits(): array;
-
-    public function locateVisit(Visit $visit, VisitLocation $location, bool $clear = false): void;
+    public function locateVisits(callable $getGeolocationData, ?callable $locatedVisit = null): void;
 }