Examples with FunctionInfo io.confluent.ksql.rest.entity.FunctionInfo used on opensource projects

Example 1 with FunctionInfo

use of io.confluent.ksql.rest.entity.FunctionInfo in project ksql by confluentinc.

the class DescribeFunctionExecutor method describeNonAggregateFunction.

private static FunctionDescriptionList describeNonAggregateFunction(final KsqlExecutionContext executionContext, final FunctionName functionName, final String statementText) {
    final UdfFactory udfFactory = executionContext.getMetaStore().getUdfFactory(functionName);
    final ImmutableList.Builder<FunctionInfo> listBuilder = ImmutableList.builder();
    udfFactory.eachFunction(func -> listBuilder.add(getFunctionInfo(func.parameterInfo(), func.declaredReturnType(), func.getDescription(), func.isVariadic())));
    return createFunctionDescriptionList(statementText, udfFactory.getMetadata(), listBuilder.build(), FunctionType.SCALAR);
}

Example 2 with FunctionInfo

use of io.confluent.ksql.rest.entity.FunctionInfo in project ksql by confluentinc.

the class DescribeFunctionExecutor method getFunctionInfo.

private static FunctionInfo getFunctionInfo(final List<ParameterInfo> argTypes, final ParamType returnTypeSchema, final String description, final boolean variadic) {
    final List<ArgumentInfo> args = new ArrayList<>();
    for (int i = 0; i < argTypes.size(); i++) {
        final ParameterInfo param = argTypes.get(i);
        final boolean isVariadic = variadic && i == (argTypes.size() - 1);
        final String type = isVariadic ? ((ArrayType) param.type()).element().toString() : param.type().toString();
        args.add(new ArgumentInfo(param.name(), type, param.description(), isVariadic));
    }
    return new FunctionInfo(args, returnTypeSchema.toString(), description);
}

Example 3 with FunctionInfo

use of io.confluent.ksql.rest.entity.FunctionInfo in project ksql by confluentinc.

the class DescribeFunctionExecutor method describeAggregateFunction.

private static FunctionDescriptionList describeAggregateFunction(final KsqlExecutionContext ksqlEngine, final FunctionName functionName, final String statementText) {
    final AggregateFunctionFactory aggregateFactory = ksqlEngine.getMetaStore().getAggregateFactory(functionName);
    final ImmutableList.Builder<FunctionInfo> listBuilder = ImmutableList.builder();
    aggregateFactory.eachFunction(func -> listBuilder.add(getFunctionInfo(func.parameterInfo(), func.declaredReturnType(), func.getDescription(), false)));
    return createFunctionDescriptionList(statementText, aggregateFactory.getMetadata(), listBuilder.build(), FunctionType.AGGREGATE);
}

Example 4 with FunctionInfo

use of io.confluent.ksql.rest.entity.FunctionInfo in project ksql by confluentinc.

the class DescribeFunctionExecutorTest method shouldDescribeUDTF.

@Test
public void shouldDescribeUDTF() {
    // When:
    final FunctionDescriptionList functionList = (FunctionDescriptionList) CUSTOM_EXECUTORS.describeFunction().execute((ConfiguredStatement<DescribeFunction>) engine.configure("DESCRIBE FUNCTION TEST_UDTF1;"), mock(SessionProperties.class), engine.getEngine(), engine.getServiceContext()).getEntity().orElseThrow(IllegalStateException::new);
    // Then:
    assertThat(functionList, new TypeSafeMatcher<FunctionDescriptionList>() {

        @Override
        protected boolean matchesSafely(final FunctionDescriptionList item) {
            return item.getName().equals("TEST_UDTF1") && item.getType().equals(FunctionType.TABLE);
        }

        @Override
        public void describeTo(final Description description) {
            description.appendText(functionList.getName());
        }
    });
    assertThat(functionList.getFunctions(), hasSize(2));
    final FunctionInfo expected1 = new FunctionInfo(Arrays.asList(new ArgumentInfo("foo", "INT", "", false)), "INT", "test_udtf1 int");
    assertTrue(functionList.getFunctions().contains(expected1));
    final FunctionInfo expected2 = new FunctionInfo(Arrays.asList(new ArgumentInfo("foo", "DOUBLE", "", false)), "DOUBLE", "test_udtf1 double");
    assertTrue(functionList.getFunctions().contains(expected2));
}

Example 5 with FunctionInfo

use of io.confluent.ksql.rest.entity.FunctionInfo in project ksql by confluentinc.

the class DescribeFunctionExecutor method describeTableFunction.

private static FunctionDescriptionList describeTableFunction(final KsqlExecutionContext executionContext, final FunctionName functionName, final String statementText) {
    final TableFunctionFactory tableFunctionFactory = executionContext.getMetaStore().getTableFunctionFactory(functionName);
    final ImmutableList.Builder<FunctionInfo> listBuilder = ImmutableList.builder();
    tableFunctionFactory.eachFunction(func -> listBuilder.add(getFunctionInfo(func.parameterInfo(), func.declaredReturnType(), func.getDescription(), func.isVariadic())));
    return createFunctionDescriptionList(statementText, tableFunctionFactory.getMetadata(), listBuilder.build(), FunctionType.TABLE);
}