Mirrors/shlink
1
0
Fork 0
mirror of https://github.com/shlinkio/shlink.git synced 2026-03-06 23:33:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improved how existing imported short URLs are checked by tracking its original short code

Browse Source
This commit is contained in:
Alejandro Celaya
2020-10-25 11:57:26 +01:00
parent 786e4f642b
commit 7c343f42c1
7 changed files with 49 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
module/Core/src/Entity/ShortUrl.php
View File

@@ -36,6 +36,7 @@ class ShortUrl extends AbstractEntity
private bool $customSlugWasProvided;
private int $shortCodeLength;
private ?string $importSource = null;
private ?string $importOriginalShortCode = null;
public function __construct(
string $longUrl,
@@ -72,6 +73,7 @@ class ShortUrl extends AbstractEntity
$instance = new self($url->longUrl(), ShortUrlMeta::fromRawData($meta), $domainResolver);
$instance->importSource = $url->source();
$instance->importOriginalShortCode = $url->shortCode();
$instance->dateCreated = Chronos::instance($url->createdAt());
return $instance;

8
module/Core/src/Importer/ImportedLinksProcessor.php
View File

@@ -50,7 +50,7 @@ class ImportedLinksProcessor implements ImportedLinksProcessorInterface
$longUrl = $url->longUrl();
// Skip already imported URLs
if ($shortUrlRepo->importedUrlExists($url, $importShortCodes)) {
if ($shortUrlRepo->importedUrlExists($url)) {
$io->text(sprintf('%s: <comment>Skipped</comment>', $longUrl));
continue;
}
@@ -58,7 +58,7 @@ class ImportedLinksProcessor implements ImportedLinksProcessorInterface
$shortUrl = ShortUrl::fromImport($url, $importShortCodes, $this->domainResolver);
$shortUrl->setTags($this->tagNamesToEntities($this->em, $url->tags()));
if (! $this->handleShortcodeUniqueness($url, $shortUrl, $io, $importShortCodes)) {
if (! $this->handleShortCodeUniqueness($url, $shortUrl, $io, $importShortCodes)) {
continue;
}
@@ -67,7 +67,7 @@ class ImportedLinksProcessor implements ImportedLinksProcessorInterface
}
}
private function handleShortcodeUniqueness(
private function handleShortCodeUniqueness(
ImportedShlinkUrl $url,
ShortUrl $shortUrl,
StyleInterface $io,
@@ -90,6 +90,6 @@ class ImportedLinksProcessor implements ImportedLinksProcessorInterface
return false;
}
return $this->handleShortcodeUniqueness($url, $shortUrl, $io, false);
return $this->handleShortCodeUniqueness($url, $shortUrl, $io, false);
}
}

40
module/Core/src/Repository/ShortUrlRepository.php
View File

@@ -190,13 +190,7 @@ DQL;
->setParameter('slug', $slug)
->setMaxResults(1);
if ($domain !== null) {
$qb->join('s.domain', 'd')
->andWhere($qb->expr()->eq('d.authority', ':authority'))
->setParameter('authority', $domain);
} else {
$qb->andWhere($qb->expr()->isNull('s.domain'));
}
$this->whereDomainIs($qb, $domain);
return $qb;
}
@@ -256,15 +250,31 @@ DQL;
return $qb->getQuery()->getOneOrNullResult();
}
public function importedUrlExists(ImportedShlinkUrl $url, bool $importShortCodes): bool
public function importedUrlExists(ImportedShlinkUrl $url): bool
{
$findConditions = ['importSource' => $url->source()];
if ($importShortCodes) {
$findConditions['shortCode'] = $url->shortCode();
} else {
$findConditions['longUrl'] = $url->longUrl();
}
$qb = $this->getEntityManager()->createQueryBuilder();
$qb->select('COUNT(DISTINCT s.id)')
->from(ShortUrl::class, 's')
->andWhere($qb->expr()->eq('s.importOriginalShortCode', ':shortCode'))
->setParameter('shortCode', $url->shortCode())
->andWhere($qb->expr()->eq('s.importSource', ':importSource'))
->setParameter('importSource', $url->source())
->setMaxResults(1);
return $this->count($findConditions) > 0;
$this->whereDomainIs($qb, $url->domain());
$result = (int) $qb->getQuery()->getSingleScalarResult();
return $result > 0;
}
private function whereDomainIs(QueryBuilder $qb, ?string $domain): void
{
if ($domain !== null) {
$qb->join('s.domain', 'd')
->andWhere($qb->expr()->eq('d.authority', ':authority'))
->setParameter('authority', $domain);
} else {
$qb->andWhere($qb->expr()->isNull('s.domain'));
}
}
}

2
module/Core/src/Repository/ShortUrlRepositoryInterface.php
View File

@@ -32,5 +32,5 @@ interface ShortUrlRepositoryInterface extends ObjectRepository
public function findOneMatching(string $url, array $tags, ShortUrlMeta $meta): ?ShortUrl;
public function importedUrlExists(ImportedShlinkUrl $url, bool $importShortCodes): bool;
public function importedUrlExists(ImportedShlinkUrl $url): bool;
}