Firesphere\SolrSearch\Tasks\ClearDirtyClassesTask

Complexity

Total Complexity

7

Size/Duplication

Total Lines	70
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	run()	0	26	5
A	createDeleteList()	0	10	2

<?php


namespace Firesphere\SolrSearch\Tasks;

use Exception;
use Firesphere\SolrSearch\Extensions\DataObjectExtension;
use Firesphere\SolrSearch\Helpers\SolrLogger;
use Firesphere\SolrSearch\Models\DirtyClass;
use Firesphere\SolrSearch\Services\SolrCoreService;
use Firesphere\SolrSearch\Traits\LoggerTrait;
use GuzzleHttp\Exception\GuzzleException;
use SilverStripe\Control\HTTPRequest;
use SilverStripe\Dev\BuildTask;
use SilverStripe\ORM\ArrayList;
use SilverStripe\ORM\DataList;
use SilverStripe\ORM\ValidationException;

/**
 * Class ClearDirtyClasses
 * Clear out classes that were not succesfully updated or deleted in Solr.
 *
 * Any classes that failed to index properly or be removed properly need to be cleaned out regularly
 * This task takes care of doing this. It can be run directly via /dev/tasks, or via a queued job
 *
 * @package Firesphere\SolrSearch\Tasks
 */
class ClearDirtyClassesTask extends BuildTask
{
    use LoggerTrait;
    /**
     * @var string URLSegment
     */
    private static $segment = 'SolrClearDirtyClasses';

The private property $segment is not used, and could be removed.

    /**
     * @var string Title
     */
    protected $title = 'Fix broken items in the Solr cores';
    /**
     * @var string Description
     */
    protected $description = 'Clear out classes that are marked as dirty on Solr.';

    /**
     * Clean up Dirty Classes in the index
     *
     * @param HTTPRequest $request
     * @return void
     * @throws GuzzleException
     * @throws \ReflectionException
     * @throws ValidationException
     */
    public function run($request)
    {
        /** @var DataList|DirtyClass $classes */
        $classes = DirtyClass::get();
        /** @var SolrCoreService $service */
        $service = new SolrCoreService();
        $solrLogger = new SolrLogger();
        foreach ($classes as $class) {
            $dirtyClass = $class->Class;
            $ids = json_decode($class->IDs);
            try {
                if ($class->Type === DataObjectExtension::WRITE) {
                    $dirtyClasses = $dirtyClass::get()->byIDs($ids);
                    $service->updateItems($dirtyClasses, SolrCoreService::UPDATE_TYPE);
                }
                if ($class->Type === DataObjectExtension::DELETE) {
                    $deletions = $this->createDeleteList($ids, $dirtyClass);
                    $service->updateItems($deletions, SolrCoreService::DELETE_TYPE);
                }
                $class->delete();
            } catch (Exception $exception) {
                $this->getLogger()->error($exception->getMessage());
                continue;
            }
        }
        $solrLogger->saveSolrLog('Index');
    }

    /**
     * Create an ArrayList of the dirty items to be deleted from Solr
     *
     * @param array $items
     * @param string $dirtyClass
     * @return ArrayList
     */
    public function createDeleteList($items, $dirtyClass): ArrayList
    {
        /** @var ArrayList $deletions */
        $deletions = ArrayList::create();
        foreach ($items as $item) {
            $dirtItem = $dirtyClass::create(['ID' => $item]);
            $deletions->push($dirtItem);
        }

        return $deletions;
    }
}