Module org.elasticsearch.compute

Package org.elasticsearch.compute.operator.fuse

Class LinearScoreEvalOperator

java.lang.Object

org.elasticsearch.compute.operator.CompleteInputCollectorOperator

org.elasticsearch.compute.operator.fuse.LinearScoreEvalOperator

All Implemented Interfaces:

Closeable, AutoCloseable, Operator, org.elasticsearch.core.Releasable

public class LinearScoreEvalOperator
extends CompleteInputCollectorOperator

Updates the score column in two stages: 1. Normalizes the scores using the normalization method specified in the config. Each row belongs to a result group which is specified by the discriminator column. Scores are normalized for each result group. 2. Multiplies the normalized score by the weight specified in the config. The config contains the weights that we need to apply for each result group.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final record	LinearScoreEvalOperator.Factory
static final record	LinearScoreEvalOperator.Status

Nested classes/interfaces inherited from interface org.elasticsearch.compute.operator.Operator

Operator.OperatorFactory

Field Summary

Fields inherited from class org.elasticsearch.compute.operator.CompleteInputCollectorOperator

finished, inputPages, pagesReceived, rowsReceived

Fields inherited from interface org.elasticsearch.compute.operator.Operator

MIN_TARGET_PAGE_SIZE, NOT_BLOCKED, TARGET_PAGE_SIZE

Constructor Summary

Constructors

Constructor	Description
LinearScoreEvalOperator(DriverContext driverContext, int discriminatorPosition, int scorePosition, LinearConfig config, WarningSourceLocation source)

Method Summary

Modifier and Type	Method	Description
boolean	canProduceMoreDataWithoutExtraInput()	Returns true if the operator can produce more output pages without requiring additional input pages.
protected boolean	isOperatorFinished()	Adds any additional checks to ensure the subclass is finished
protected void	onClose()	Additional implementation for closing out the subclass
protected void	onFinished()	Called when he base finish() call is completed
protected Page	onGetOutput()	Implementation to retrieve the output pages.
Operator.Status	status()	The status of the operator.
String	toString()

Methods inherited from class org.elasticsearch.compute.operator.CompleteInputCollectorOperator

addInput, close, finish, getOutput, isFinished, needsInput

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.elasticsearch.compute.operator.Operator

isBlocked

Constructor Details

LinearScoreEvalOperator

public LinearScoreEvalOperator(DriverContext driverContext,
 int discriminatorPosition,
 int scorePosition,
 LinearConfig config,
 WarningSourceLocation source)

Method Details

onFinished

protected void onFinished()

Description copied from class: CompleteInputCollectorOperator

Called when he base finish() call is completed

Specified by:

onFinished in class CompleteInputCollectorOperator

isOperatorFinished

protected boolean isOperatorFinished()

Description copied from class: CompleteInputCollectorOperator

Adds any additional checks to ensure the subclass is finished

Specified by:

isOperatorFinished in class CompleteInputCollectorOperator

Returns:

true if the subclass is finished

onGetOutput

protected Page onGetOutput()

Description copied from class: CompleteInputCollectorOperator

Implementation to retrieve the output pages. Will only run if `isFinished()` is true

Specified by:

onGetOutput in class CompleteInputCollectorOperator

Returns:

the output page (or null if not ready or none)

onClose

protected void onClose()

Description copied from class: CompleteInputCollectorOperator

Additional implementation for closing out the subclass

Specified by:

onClose in class CompleteInputCollectorOperator

canProduceMoreDataWithoutExtraInput

public boolean canProduceMoreDataWithoutExtraInput()

Description copied from interface: Operator

Returns true if the operator can produce more output pages without requiring additional input pages. This is useful for operators that buffer data or have internal state that can produce multiple output pages.

Operators that do not buffer data should return false - they cannot produce pages out of thin air. Examples of operators that may return true:

• Operators with internal buffers (e.g., AsyncOperator with pending results)
• Operators processing a single input page into multiple output pages
• Aggregation operators that buffer partial results

Returns:

true if the operator has buffered data that can produce output, false otherwise

toString

public String toString()

Overrides:

toString in class Object

status

public Operator.Status status()

Description copied from interface: Operator

The status of the operator.