Code Coverage |
||||||||||
Classes and Traits |
Functions and Methods |
Lines |
||||||||
| Total | |
0.00% |
0 / 1 |
|
0.00% |
0 / 4 |
CRAP | |
0.00% |
0 / 48 |
| ShardAllocationValidator | |
0.00% |
0 / 1 |
|
0.00% |
0 / 4 |
110 | |
0.00% |
0 / 48 |
| __construct | |
0.00% |
0 / 1 |
2 | |
0.00% |
0 / 4 |
|||
| validate | |
0.00% |
0 / 1 |
56 | |
0.00% |
0 / 33 |
|||
| fetchActualAllocation | |
0.00% |
0 / 1 |
2 | |
0.00% |
0 / 3 |
|||
| set | |
0.00% |
0 / 1 |
2 | |
0.00% |
0 / 8 |
|||
| <?php | |
| namespace CirrusSearch\Maintenance\Validators; | |
| use CirrusSearch\Maintenance\Printer; | |
| use Elastica\Index; | |
| use Status; | |
| class ShardAllocationValidator extends Validator { | |
| /** | |
| * @var Index | |
| */ | |
| private $index; | |
| /** | |
| * @var array | |
| */ | |
| private $indexAllocation; | |
| /** | |
| * @param Index $index | |
| * @param array $indexAllocation | |
| * @param Printer|null $out | |
| */ | |
| public function __construct( Index $index, array $indexAllocation, Printer $out = null ) { | |
| parent::__construct( $out ); | |
| $this->index = $index; | |
| $this->indexAllocation = $indexAllocation; | |
| } | |
| /** | |
| * @return Status | |
| */ | |
| public function validate() { | |
| $this->outputIndented( "\tValidating shard allocation settings..." ); | |
| $actual = $this->fetchActualAllocation(); | |
| $changed = false; | |
| foreach ( [ 'include', 'exclude', 'require' ] as $type ) { | |
| $desired = $this->indexAllocation[$type]; | |
| if ( $desired ) { | |
| $this->output( "\n" ); | |
| $this->outputIndented( "\t\tUpdating '$type' allocations..." ); | |
| $this->set( $type, $desired ); | |
| $this->output( "done" ); | |
| $changed = true; | |
| } | |
| if ( isset( $actual[$type] ) ) { | |
| $undesired = array_filter( array_keys( $actual[$type] ), | |
| static function ( $key ) use ( $actual, $type, $desired ) { | |
| return $actual[$type][$key] !== '' && !isset( $desired[$key] ); | |
| } | |
| ); | |
| if ( $undesired ) { | |
| $this->output( "\n" ); | |
| $this->outputIndented( "\t\tClearing '$type' allocations..." ); | |
| $this->set( $type, array_fill_keys( $undesired, '' ) ); | |
| $this->output( "done" ); | |
| $changed = true; | |
| } | |
| } | |
| } | |
| if ( $changed ) { | |
| $this->output( "\n" ); | |
| } else { | |
| $this->output( "done\n" ); | |
| } | |
| return Status::newGood(); | |
| } | |
| /** | |
| * @return array | |
| */ | |
| private function fetchActualAllocation() { | |
| $settings = $this->index->getSettings()->get(); | |
| return $settings['routing']['allocation'] ?? []; | |
| } | |
| /** | |
| * @param string $type | |
| * @param array $allocation | |
| */ | |
| private function set( $type, $allocation ) { | |
| $this->index->getSettings()->set( [ | |
| 'routing' => [ | |
| 'allocation' => [ | |
| $type => $allocation, | |
| ] | |
| ] | |
| ] ); | |
| } | |
| } |