Example 1 with VARCHAR_TO_BIGINT_RETURN_VALUE
use of io.trino.metadata.TestPolymorphicScalarFunction.TestMethods.VARCHAR_TO_BIGINT_RETURN_VALUE in project trino by trinodb.
the class TestPolymorphicScalarFunction method testSelectsMethodBasedOnReturnType.
@Test
public void testSelectsMethodBasedOnReturnType() throws Throwable {
SqlScalarFunction function = new PolymorphicScalarFunctionBuilder(TestMethods.class).signature(SIGNATURE).deterministic(true).choice(choice -> choice.implementation(methodsGroup -> methodsGroup.methods("varcharToVarcharCreateSliceWithExtraParameterLength")).implementation(methodsGroup -> methodsGroup.methods("varcharToBigintReturnExtraParameter").withExtraParameters(context -> ImmutableList.of(42)))).build();
ChoicesScalarFunctionImplementation functionImplementation = (ChoicesScalarFunctionImplementation) function.specialize(BOUND_SIGNATURE, new FunctionDependencies(FUNCTION_MANAGER::getScalarFunctionInvoker, ImmutableMap.of(), ImmutableSet.of()));
assertEquals(functionImplementation.getChoices().get(0).getMethodHandle().invoke(INPUT_SLICE), VARCHAR_TO_BIGINT_RETURN_VALUE);
}
Also used :
VARCHAR_TO_VARCHAR_RETURN_VALUE(io.trino.metadata.TestPolymorphicScalarFunction.TestMethods.VARCHAR_TO_VARCHAR_RETURN_VALUE)
BLOCK_POSITION(io.trino.spi.function.InvocationConvention.InvocationArgumentConvention.BLOCK_POSITION)
Slice(io.airlift.slice.Slice)
FAIL_ON_NULL(io.trino.spi.function.InvocationConvention.InvocationReturnConvention.FAIL_ON_NULL)
MAX_SHORT_PRECISION(io.trino.spi.type.Decimals.MAX_SHORT_PRECISION)
TypeSignatureParameter.typeVariable(io.trino.spi.type.TypeSignatureParameter.typeVariable)
BOOLEAN(io.trino.spi.type.BooleanType.BOOLEAN)
Assert.assertEquals(org.testng.Assert.assertEquals)
ChoicesScalarFunctionImplementation(io.trino.operator.scalar.ChoicesScalarFunctionImplementation)
Test(org.testng.annotations.Test)
IS_DISTINCT_FROM(io.trino.spi.function.OperatorType.IS_DISTINCT_FROM)
FunctionManager.createTestingFunctionManager(io.trino.metadata.FunctionManager.createTestingFunctionManager)
VARCHAR(io.trino.spi.type.VarcharType.VARCHAR)
ImmutableList(com.google.common.collect.ImmutableList)
NULL_FLAG(io.trino.spi.function.InvocationConvention.InvocationArgumentConvention.NULL_FLAG)
Assertions.assertThatThrownBy(org.assertj.core.api.Assertions.assertThatThrownBy)
Block(io.trino.spi.block.Block)
Arrays.asList(java.util.Arrays.asList)
Slices(io.airlift.slice.Slices)
LongArrayBlock(io.trino.spi.block.LongArrayBlock)
Assert.assertFalse(org.testng.Assert.assertFalse)
TypeSignature(io.trino.spi.type.TypeSignature)
Int128(io.trino.spi.type.Int128)
ImmutableSet(com.google.common.collect.ImmutableSet)
ImmutableMap(com.google.common.collect.ImmutableMap)
Signature.comparableWithVariadicBound(io.trino.metadata.Signature.comparableWithVariadicBound)
VARCHAR_TO_BIGINT_RETURN_VALUE(io.trino.metadata.TestPolymorphicScalarFunction.TestMethods.VARCHAR_TO_BIGINT_RETURN_VALUE)
InvocationConvention(io.trino.spi.function.InvocationConvention)
ADD(io.trino.spi.function.OperatorType.ADD)
BIGINT(io.trino.spi.type.BigintType.BIGINT)
Optional(java.util.Optional)
Assert.assertTrue(org.testng.Assert.assertTrue)
VarcharType.createVarcharType(io.trino.spi.type.VarcharType.createVarcharType)
DecimalType(io.trino.spi.type.DecimalType)
ChoicesScalarFunctionImplementation(io.trino.operator.scalar.ChoicesScalarFunctionImplementation)
Test(org.testng.annotations.Test)
Aggregations
ImmutableList (com.google.common.collect.ImmutableList)1
ImmutableMap (com.google.common.collect.ImmutableMap)1
ImmutableSet (com.google.common.collect.ImmutableSet)1
Slice (io.airlift.slice.Slice)1
Slices (io.airlift.slice.Slices)1
FunctionManager.createTestingFunctionManager (io.trino.metadata.FunctionManager.createTestingFunctionManager)1
Signature.comparableWithVariadicBound (io.trino.metadata.Signature.comparableWithVariadicBound)1
VARCHAR_TO_BIGINT_RETURN_VALUE (io.trino.metadata.TestPolymorphicScalarFunction.TestMethods.VARCHAR_TO_BIGINT_RETURN_VALUE)1
VARCHAR_TO_VARCHAR_RETURN_VALUE (io.trino.metadata.TestPolymorphicScalarFunction.TestMethods.VARCHAR_TO_VARCHAR_RETURN_VALUE)1
ChoicesScalarFunctionImplementation (io.trino.operator.scalar.ChoicesScalarFunctionImplementation)1
Block (io.trino.spi.block.Block)1
LongArrayBlock (io.trino.spi.block.LongArrayBlock)1
InvocationConvention (io.trino.spi.function.InvocationConvention)1
BLOCK_POSITION (io.trino.spi.function.InvocationConvention.InvocationArgumentConvention.BLOCK_POSITION)1
NULL_FLAG (io.trino.spi.function.InvocationConvention.InvocationArgumentConvention.NULL_FLAG)1
FAIL_ON_NULL (io.trino.spi.function.InvocationConvention.InvocationReturnConvention.FAIL_ON_NULL)1
ADD (io.trino.spi.function.OperatorType.ADD)1
IS_DISTINCT_FROM (io.trino.spi.function.OperatorType.IS_DISTINCT_FROM)1
BIGINT (io.trino.spi.type.BigintType.BIGINT)1
BOOLEAN (io.trino.spi.type.BooleanType.BOOLEAN)1
