mirror of
https://github.com/shlinkio/shlink.git
synced 2024-12-24 08:00:13 -06:00
59 lines
2.0 KiB
PHP
59 lines
2.0 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace ShlinkMigrations;
|
|
|
|
use Doctrine\DBAL\Schema\Schema;
|
|
use Doctrine\Migrations\AbstractMigration;
|
|
|
|
/**
|
|
* Create initial entries in the short_url_visits_counts table for existing visits
|
|
*/
|
|
final class Version20240318084804 extends AbstractMigration
|
|
{
|
|
public function up(Schema $schema): void
|
|
{
|
|
$qb = $this->connection->createQueryBuilder();
|
|
$result = $qb->select('id')
|
|
->from('short_urls')
|
|
->executeQuery();
|
|
|
|
while ($shortUrlId = $result->fetchOne()) {
|
|
$visitsQb = $this->connection->createQueryBuilder();
|
|
$visitsQb->select('COUNT(id)')
|
|
->from('visits')
|
|
->where($visitsQb->expr()->eq('short_url_id', ':short_url_id'))
|
|
->andWhere($visitsQb->expr()->eq('potential_bot', ':potential_bot'))
|
|
->setParameter('short_url_id', $shortUrlId);
|
|
|
|
$botsCount = $visitsQb->setParameter('potential_bot', '1')->executeQuery()->fetchOne();
|
|
$nonBotsCount = $visitsQb->setParameter('potential_bot', '0')->executeQuery()->fetchOne();
|
|
|
|
if ($botsCount > 0) {
|
|
$this->insertCount($shortUrlId, $botsCount, potentialBot: true);
|
|
}
|
|
if ($nonBotsCount > 0) {
|
|
$this->insertCount($shortUrlId, $nonBotsCount, potentialBot: false);
|
|
}
|
|
}
|
|
}
|
|
|
|
private function insertCount(string $shortUrlId, int $count, bool $potentialBot): void
|
|
{
|
|
$this->connection->createQueryBuilder()
|
|
->insert('short_url_visits_counts')
|
|
->values([
|
|
'short_url_id' => ':short_url_id',
|
|
'count' => ':count',
|
|
'potential_bot' => ':potential_bot',
|
|
])
|
|
->setParameters([
|
|
'short_url_id' => $shortUrlId,
|
|
'count' => $count,
|
|
'potential_bot' => $potentialBot ? '1' : '0',
|
|
])
|
|
->executeStatement();
|
|
}
|
|
}
|