Adds get ownership urls count to url repository

YorCreative · Aug 29, 2024 · 3c7f331 · 3c7f331
1 parent 9b6f9f5
commit 3c7f331
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/Repositories/UrlRepository.php b/src/Repositories/UrlRepository.php
@@ -50,6 +50,25 @@ public static function findOwnershipUrls(string $owner_type, int $owner_id): Col
         }
     }
 
+    /**
+     * @throws UrlRepositoryException
+     */
+    public static function getOwnershipUrlsCount(string $owner_type, int $owner_id): int
+    {
+        try {
+            return ShortUrl::whereIn('id', function ($query) use ($owner_type, $owner_id) {
+                $query->from('short_url_ownerships');
+                $query->where([
+                    'ownerable_type' => $owner_type,
+                    'ownerable_id' => $owner_id,
+                ]);
+                $query->select('short_url_id');
+            })->count();
+        } catch (Exception $exception) {
+            throw new UrlRepositoryException($exception->getMessage());
+        }
+    }
+
     /**
      * @return mixed
      *

diff --git a/src/Services/UrlService.php b/src/Services/UrlService.php
@@ -97,4 +97,12 @@ public static function getAllByOwner(string $ownerable_type, string $ownerable_i
     {
         return UrlRepository::findOwnershipUrls($ownerable_type, $ownerable_id);
     }
+
+    /**
+     * @throws UrlRepositoryException
+     */
+    public static function getAllByOwnerCount(string $ownerable_type, string $ownerable_id): int
+    {
+        return UrlRepository::getOwnershipUrlsCount($ownerable_type,$ownerable_id);
+    }
 }