koel/app/Models/Concerns/SupportsDeleteWhereValueNotIn.php

55 lines
1.6 KiB
PHP
Raw Normal View History

2022-08-01 10:42:33 +00:00
<?php
namespace App\Models\Concerns;
2022-08-01 10:42:33 +00:00
use Illuminate\Database\Eloquent\Builder;
use Illuminate\Support\Facades\DB;
/**
* With reference to GitHub issue #463.
* MySQL and PostgresSQL seem to have a limit of 2^16-1 (65535) elements in an IN statement.
* This trait provides a method as a workaround to this limitation.
*
* @method static Builder query()
2022-08-01 10:42:33 +00:00
*/
trait SupportsDeleteWhereValueNotIn
{
/**
* Deletes all records whose certain value is not in an array.
*/
public static function deleteWhereValueNotIn(array $values, ?string $field = null): void
2022-08-01 10:42:33 +00:00
{
$field ??= (new static())->getKeyName();
$maxChunkSize = DB::getDriverName() === 'sqlite' ? 999 : 65_535;
2022-08-01 10:42:33 +00:00
if (count($values) <= $maxChunkSize) {
static::query()->whereNotIn($field, $values)->delete();
2022-08-01 10:42:33 +00:00
return;
}
$allIds = static::query()->select($field)->get()->pluck($field)->all();
$deletableIds = array_diff($allIds, $values);
2022-08-01 10:42:33 +00:00
if (count($deletableIds) < $maxChunkSize) {
static::query()->whereIn($field, $deletableIds)->delete();
2022-08-01 10:42:33 +00:00
return;
}
static::deleteByChunk($deletableIds, $maxChunkSize, $field);
2022-08-01 10:42:33 +00:00
}
public static function deleteByChunk(array $values, int $chunkSize = 65_535, ?string $field = null): void
2022-08-01 10:42:33 +00:00
{
$field ??= (new static())->getKeyName();
2022-08-01 10:42:33 +00:00
DB::transaction(static function () use ($values, $field, $chunkSize): void {
foreach (array_chunk($values, $chunkSize) as $chunk) {
static::query()->whereIn($field, $chunk)->delete();
2022-08-01 10:42:33 +00:00
}
});
}
}