wikimedia/mediawiki-extensions-CirrusSearch

<?php

namespace CirrusSearch\Search\Rescore;

use CirrusSearch\SearchConfig;
use Elastica\Query\FunctionScore;

/**
 * Builds a function using a custom numeric field and
 * parameters attached to a profile.
 * Uses the function field_value_factor
 */
class CustomFieldFunctionScoreBuilder extends FunctionScoreBuilder {
    /**
     * @var array the field_value_factor profile
     */
    private $profile;

    /**
     * @param SearchConfig $config
     * @param float $weight
     * @param array $profile
     */
    public function __construct( SearchConfig $config, $weight, $profile ) {
        parent::__construct( $config, $weight );
        if ( isset( $profile['factor'] ) ) {
            $profile['factor'] = $this->getOverriddenFactor( $profile['factor'] );
        }
        $this->profile = $profile;
    }

    public function append( FunctionScore $functionScore ) {
        if ( isset( $this->profile['factor'] ) && $this->profile['factor'] === 0.0 ) {
            // If factor is 0 this function score will have no impact.
            return;
        }
        $functionScore->addFunction( 'field_value_factor', $this->profile, null, $this->weight );
    }
}