iCom\SolrClient\Query\UpdateQuery

Complexity

Total Complexity

21

Size/Duplication

Total Lines	102
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	1

10 Methods

Rating	Name	Duplication	Size	Complexity
A	optimize()	0	13	4
A	add()	0	13	4
A	__construct()	0	5	1
A	commit()	0	13	5
A	create()	0	3	1
A	withCommand()	0	6	1
A	toJson()	0	8	2
A	deleteByIds()	0	3	1
A	rollback()	0	3	1
A	deleteByQuery()	0	3	1

<?php

declare(strict_types=1);

/*
 * This file is part of Solr Client Symfony package.
 *
 * (c) ingatlan.com Zrt. <fejlesztes@ingatlan.com>
 *
 * This source file is subject to the MIT license that is bundled
 * with this source code in the file LICENSE.
 */

namespace iCom\SolrClient\Query;

use iCom\SolrClient\JsonQuery;
use iCom\SolrClient\Query\Command\Add;
use iCom\SolrClient\Query\Command\Commit;
use iCom\SolrClient\Query\Command\Delete;
use iCom\SolrClient\Query\Command\Optimize;
use iCom\SolrClient\Query\Command\Rollback;

/**
 * Creates a JSON formatted update query.
 *
 * Note that: Solr has his own JSON syntax which allows to have multiple keys (multiple commands).
 *
 * @see https://lucene.apache.org/solr/guide/8_3/uploading-data-with-index-handlers.html#sending-json-update-commands
 *
 * @psalm-immutable
 */
final class UpdateQuery implements JsonQuery
{
    /**
     * @var Command[]
     * @psaml-var list<Command>
     */
    private array $commands = [];

    /**
     * @psalm-param list<Command> $commands
     */
    public function __construct(iterable $commands = [])
    {
        $this->commands = (static function (Command ...$commands): array {
            return $commands;
        })(...$commands);
    }

    /**
     * @psalm-param list<Command> $commands
     */
    public static function create(iterable $commands = []): self
    {
        return new self($commands);
    }

    public function add(array $document, ?int $commitWithin = null, ?bool $overwrite = null): self
    {
        $add = Add::create($document);

        if (null !== $overwrite) {
            $add = $overwrite ? $add->enableOverWrite() : $add->disableOverWrite();
        }

        if (null !== $commitWithin) {
            $add = $add->commitWithin($commitWithin);
        }

        return $this->withCommand($add);
    }

    public function commit(?bool $waitSearcher = null, ?bool $expungeDeletes = null): self
    {
        $commit = Commit::create();

        if (null !== $waitSearcher) {
            $commit = $waitSearcher ? $commit->enableWaitSearcher() : $commit->disableWaitSearcher();
        }

        if (null !== $expungeDeletes) {
            $commit = $expungeDeletes ? $commit->enableExpungeDeletes() : $commit->disableExpungeDeletes();
        }

        return $this->withCommand($commit);
    }

    public function optimize(?bool $waitSearcher = null, ?int $maxSegments = null): self
    {
        $optimize = Optimize::create();

        if (null !== $waitSearcher) {
            $optimize = $waitSearcher ? $optimize->enableWaitSearcher() : $optimize->disableWaitSearcher();
        }

        if (null !== $maxSegments) {
            $optimize = $optimize->maxSegments($maxSegments);
        }

        return $this->withCommand($optimize);
    }

    public function rollback(): self
    {
        return $this->withCommand(Rollback::create());
    }

    public function deleteByIds(array $ids): self
    {
        return $this->withCommand(Delete::fromIds($ids));
    }

    public function deleteByQuery(SelectQuery $query): self
    {
        return $this->withCommand(Delete::fromQuery($query));
    }

    public function toJson(): string
    {
        $commands = [];
        foreach ($this->commands as $command) {
            $commands[] = sprintf('"%s":%s', $command->getName(), $command->toJson());
        }

        return sprintf('{%s}', implode(',', $commands));
    }

    private function withCommand(Command $command): self
    {
        $self = clone $this;
        $self->commands[] = $command;

        return $self;
    }
}