Package io.substrait.extension

Class ExtensionCollector

java.lang.Object

io.substrait.extension.AbstractExtensionLookup

io.substrait.extension.ExtensionCollector

All Implemented Interfaces:

ExtensionLookup

public class ExtensionCollector
extends AbstractExtensionLookup

Maintains a mapping between function/type anchors and function/type references. Generates references for new anchors as they are requested.

Used to replace instances of function and types in the POJOs with references when converting from Plan to Plan

Field Summary

Fields inherited from class io.substrait.extension.AbstractExtensionLookup

functionAnchorMap, typeAnchorMap

Constructor Summary

Constructors

Constructor	Description
ExtensionCollector()

Method Summary

Modifier and Type	Method	Description
void	addExtensionsToExtendedExpression(ExtendedExpression.Builder builder)
void	addExtensionsToPlan(Plan.Builder builder)
int	getFunctionReference(SimpleExtension.Function declaration)
int	getTypeReference(SimpleExtension.TypeAnchor typeAnchor)

Methods inherited from class io.substrait.extension.AbstractExtensionLookup

getAggregateFunction, getScalarFunction, getType, getWindowFunction

Methods inherited from class java.lang.Object

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

Constructor Details

ExtensionCollector

public ExtensionCollector()

Method Details

getFunctionReference

public int getFunctionReference(SimpleExtension.Function declaration)

getTypeReference

public int getTypeReference(SimpleExtension.TypeAnchor typeAnchor)

addExtensionsToPlan

public void addExtensionsToPlan(Plan.Builder builder)

addExtensionsToExtendedExpression

public void addExtensionsToExtendedExpression(ExtendedExpression.Builder builder)