mirror of
https://github.com/silverstripe/silverstripe-fulltextsearch
synced 2024-10-22 14:05:29 +02:00
114 lines
3.8 KiB
PHP
114 lines
3.8 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\FullTextSearch\Solr\Reindex\Handlers;
|
|
|
|
use Psr\Log\LoggerInterface;
|
|
use SilverStripe\FullTextSearch\Solr\SolrIndex;
|
|
use SilverStripe\ORM\DataObject;
|
|
use SilverStripe\ORM\DB;
|
|
use SilverStripe\Core\Convert;
|
|
use SilverStripe\Core\Injector\Injector;
|
|
use SilverStripe\FullTextSearch\Solr\Reindex\Jobs\SolrReindexQueuedJob;
|
|
use SilverStripe\FullTextSearch\Solr\Reindex\Jobs\SolrReindexGroupQueuedJob;
|
|
use SilverStripe\FullTextSearch\Search\Processors\SearchUpdateCommitJobProcessor;
|
|
|
|
if (!interface_exists('SilverStripe\QueuedJobs\Services\QueuedJob')) {
|
|
return;
|
|
}
|
|
|
|
use SilverStripe\QueuedJobs\Services\QueuedJob;
|
|
use SilverStripe\QueuedJobs\Services\QueuedJobService;
|
|
use SilverStripe\QueuedJobs\DataObjects\QueuedJobDescriptor;
|
|
|
|
/**
|
|
* Represents a queued task to start the reindex job
|
|
*/
|
|
class SolrReindexQueuedHandler extends SolrReindexBase
|
|
{
|
|
/**
|
|
* @return QueuedJobService
|
|
*/
|
|
protected function getQueuedJobService()
|
|
{
|
|
return singleton(QueuedJobService::class);
|
|
}
|
|
|
|
/**
|
|
* Cancel any cancellable jobs
|
|
*
|
|
* @param string $type Type of job to cancel
|
|
* @return int Number of jobs cleared
|
|
*/
|
|
protected function cancelExistingJobs($type)
|
|
{
|
|
$clearable = array(
|
|
// Paused jobs need to be discarded
|
|
QueuedJob::STATUS_PAUSED,
|
|
|
|
// These types would be automatically started
|
|
QueuedJob::STATUS_NEW,
|
|
QueuedJob::STATUS_WAIT,
|
|
|
|
// Cancel any in-progress job
|
|
QueuedJob::STATUS_INIT,
|
|
QueuedJob::STATUS_RUN
|
|
);
|
|
DB::query(sprintf(
|
|
'UPDATE "%s" '
|
|
. ' SET "JobStatus" = \'%s\''
|
|
. ' WHERE "JobStatus" IN (\'%s\')'
|
|
. ' AND "Implementation" = \'%s\'',
|
|
Convert::raw2sql(DataObject::getSchema()->tableName(QueuedJobDescriptor::class)),
|
|
Convert::raw2sql(QueuedJob::STATUS_CANCELLED),
|
|
implode("','", Convert::raw2sql($clearable)),
|
|
Convert::raw2sql($type)
|
|
));
|
|
return DB::affected_rows();
|
|
}
|
|
|
|
public function triggerReindex(LoggerInterface $logger, $batchSize, $taskName, $classes = null)
|
|
{
|
|
// Cancel existing jobs
|
|
$queues = $this->cancelExistingJobs(SolrReindexQueuedJob::class);
|
|
$groups = $this->cancelExistingJobs(SolrReindexGroupQueuedJob::class);
|
|
$logger->info("Cancelled {$queues} re-index tasks and {$groups} re-index groups");
|
|
|
|
// Although this class is used as a service (singleton) it may also be instantiated
|
|
// as a queuedjob
|
|
$job = Injector::inst()->create(SolrReindexQueuedJob::class, $batchSize, $taskName, $classes);
|
|
$this
|
|
->getQueuedJobService()
|
|
->queueJob($job);
|
|
|
|
$title = $job->getTitle();
|
|
$logger->info("Queued {$title}");
|
|
}
|
|
|
|
protected function processGroup(
|
|
LoggerInterface $logger, SolrIndex $indexInstance, $state, $class, $groups, $group, $taskName
|
|
) {
|
|
// Trigger another job for this group
|
|
$job = Injector::inst()->create(
|
|
SolrReindexGroupQueuedJob::class,
|
|
get_class($indexInstance), $state, $class, $groups, $group
|
|
);
|
|
$this
|
|
->getQueuedJobService()
|
|
->queueJob($job);
|
|
|
|
$title = $job->getTitle();
|
|
$logger->info("Queued {$title}");
|
|
}
|
|
|
|
public function runGroup(
|
|
LoggerInterface $logger, SolrIndex $indexInstance, $state, $class, $groups, $group
|
|
) {
|
|
parent::runGroup($logger, $indexInstance, $state, $class, $groups, $group);
|
|
|
|
// After any changes have been made, mark all indexes as dirty for commit
|
|
// see http://stackoverflow.com/questions/7512945/how-to-fix-exceeded-limit-of-maxwarmingsearchers
|
|
$logger->info("Queuing commit on all changes");
|
|
SearchUpdateCommitJobProcessor::queue();
|
|
}
|
|
}
|