wikimedia/mediawiki-extensions-CirrusSearch

<?php

namespace CirrusSearch\Maintenance\Validators;

use CirrusSearch\Maintenance\Printer;
use Elastica\Index;
use MediaWiki\Status\Status;

class ReplicaRangeValidator extends Validator {
    /**
     * @var Index
     */
    private $index;

    /**
     * @var string
     */
    protected $replicaCount;

    /**
     * @param Index $index
     * @param string $replicaCount
     * @param Printer|null $out
     */
    public function __construct( Index $index, $replicaCount, Printer $out = null ) {
        parent::__construct( $out );

        $this->index = $index;
        $this->replicaCount = $replicaCount;
    }

    /**
     * @return Status
     */
    public function validate() {
        $this->outputIndented( "\tValidating replica range..." );
        $settings = $this->index->getSettings()->get();
        $actualReplicaCount = $settings['auto_expand_replicas'] ?? 'false';
        if ( $actualReplicaCount == $this->replicaCount ) {
            $this->output( "ok\n" );
        } else {
            $this->output( "is $actualReplicaCount but should be " . $this->replicaCount . '...' );
            $this->index->getSettings()->set( [ 'auto_expand_replicas' => $this->replicaCount ] );
            $this->output( "corrected\n" );
        }

        return Status::newGood();
    }
}