Padosoft\SuperCacheInvalidate\Console\PruneCacheInvalidationDataCommand

Complexity

Total Complexity

5

Size/Duplication

Total Lines	82
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	pruneTable()	0	18	2
A	handle()	0	18	2
A	getPartitionsFromDb()	0	11	1

<?php

namespace Padosoft\SuperCacheInvalidate\Console;

use Illuminate\Console\Command;
use Illuminate\Support\Facades\DB;

class PruneCacheInvalidationDataCommand extends Command
{
    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'supercache:prune-invalidation-data
                            {--months=1 : The number of months to retain data}';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Prune old cache invalidation data by dropping partitions';

    /**
     * Prune partitions in a table older than the retention partition key.
     *
     * @param string $tableName             The name of the table
     * @param int    $retentionPartitionKey The partition key cutoff
     */
    protected function pruneTable(string $tableName, int $retentionPartitionKey): void
    {
        // Fetch partition names
        $partitions = $this->getPartitionsFromDb($tableName, $retentionPartitionKey);

        if (empty($partitions)) {
            $this->info("No partitions to prune for table {$tableName}.");

            return;
        }

        // Build DROP PARTITION statement
        $partitionNames = implode(', ', array_map(function ($partition) {
            return $partition->PARTITION_NAME;
        }, $partitions));

        DB::statement("ALTER TABLE `{$tableName}` DROP PARTITION {$partitionNames}");
        $this->info("Pruned partitions: {$partitionNames} from table {$tableName}.");
    }

    /**
     * Execute the console command.
     */
    public function handle(): void
    {
        $months = (int) $this->option('months');
        $retentionDate = now()->subMonths($months);
        $retentionPartitionKey = $retentionDate->year * 100 + $retentionDate->week();

        // Prune tables
        $tables = [
            'cache_invalidation_events',
            'cache_invalidation_timestamps',
            'cache_invalidation_event_associations',
        ];

        foreach ($tables as $tableName) {
            $this->pruneTable($tableName, $retentionPartitionKey);
        }

        $this->info('Old cache invalidation data has been pruned.');
    }

    /**
     * @param string $tableName
     * @param int $retentionPartitionKey
     * @return array
     */
    protected function getPartitionsFromDb(string $tableName, int $retentionPartitionKey): array
    {
        $partitions = DB::select('
            SELECT PARTITION_NAME, PARTITION_DESCRIPTION
            FROM INFORMATION_SCHEMA.PARTITIONS
            WHERE TABLE_SCHEMA = DATABASE()
            AND TABLE_NAME = ?
            AND PARTITION_NAME IS NOT NULL
            AND PARTITION_DESCRIPTION < ?
        ', [$tableName, $retentionPartitionKey]);
        return $partitions;
    }
}