/*

* Licensed to the Apache Software Foundation (ASF) under one or more

* contributor license agreements. See the NOTICE file distributed with

* this work for additional information regarding copyright ownership.

* The ASF licenses this file to you under the Apache License, Version 2.0

* (the "License"); you may not use this file except in compliance with

* the License. You may obtain a copy of the License at

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*/

package org.apache.calcite.test;

import org.apache.calcite.adapter.enumerable.CallImplementor;

import org.apache.calcite.adapter.java.ReflectiveSchema;

import org.apache.calcite.jdbc.CalciteConnection;

import org.apache.calcite.linq4j.Ord;

import org.apache.calcite.linq4j.function.SemiStrict;

import org.apache.calcite.linq4j.tree.Expressions;

import org.apache.calcite.linq4j.tree.Types;

import org.apache.calcite.rel.type.RelDataType;

import org.apache.calcite.rel.type.RelDataTypeFactory;

import org.apache.calcite.rel.type.RelProtoDataType;

import org.apache.calcite.schema.FunctionParameter;

import org.apache.calcite.schema.ImplementableFunction;

import org.apache.calcite.schema.ScalarFunction;

import org.apache.calcite.schema.SchemaPlus;

import org.apache.calcite.schema.impl.AbstractSchema;

import org.apache.calcite.schema.impl.ScalarFunctionImpl;

import org.apache.calcite.schema.impl.ViewTable;

import org.apache.calcite.sql.type.SqlTypeName;

import org.apache.calcite.sql.validate.SqlConformanceEnum;

import org.apache.calcite.test.schemata.hr.HrSchema;

import org.apache.calcite.util.Smalls;

import com.google.common.collect.ImmutableList;

import org.junit.jupiter.api.Disabled;

import org.junit.jupiter.api.Test;

import java.lang.reflect.Method;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.Statement;

import java.util.ArrayList;

import java.util.List;

import java.util.concurrent.atomic.AtomicInteger;

import static org.hamcrest.CoreMatchers.is;

import static org.hamcrest.MatcherAssert.assertThat;

/**

* Tests for user-defined functions;

* includes user-defined aggregate functions

* but user-defined table functions are in {@link TableFunctionTest}.

*

* @see Smalls

*/

class UdfTest {

private CalciteAssert.AssertThat withUdf() {

final String model = "{\n"

+ " version: '1.0',\n"

+ " schemas: [\n"

+ " {\n"

+ " name: 'adhoc',\n"

+ " tables: [\n"

+ " {\n"

+ " name: 'EMPLOYEES',\n"

+ " type: 'custom',\n"

+ " factory: '"

+ JdbcTest.EmpDeptTableFactory.class.getName()

+ "',\n"

+ " operand: {'foo': true, 'bar': 345}\n"

+ " }\n"

+ " ],\n"

+ " functions: [\n"

+ " {\n"

+ " name: 'MY_PLUS',\n"

+ " className: '"

+ Smalls.MyPlusFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_DET_PLUS',\n"

+ " className: '"

+ Smalls.MyDeterministicPlusFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_LEFT',\n"

+ " className: '"

+ Smalls.MyLeftFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'ABCDE',\n"

+ " className: '"

+ Smalls.MyAbcdeFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_STR',\n"

+ " className: '"

+ Smalls.MyToStringFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_DOUBLE',\n"

+ " className: '"

+ Smalls.MyDoubleFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_EXCEPTION',\n"

+ " className: '"

+ Smalls.MyExceptionFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_NILADIC_PARENTHESES',\n"

+ " className: '"

+ Smalls.MyNiladicParenthesesFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'COUNT_ARGS',\n"

+ " className: '"

+ Smalls.CountArgs0Function.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'COUNT_ARGS',\n"

+ " className: '"

+ Smalls.CountArgs1Function.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'COUNT_ARGS',\n"

+ " className: '"

+ Smalls.CountArgs1NullableFunction.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'COUNT_ARGS',\n"

+ " className: '"

+ Smalls.CountArgs2Function.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'MY_ABS',\n"

+ " className: '"

+ java.lang.Math.class.getName()

+ "',\n"

+ " methodName: 'abs'\n"

+ " },\n"

+ " {\n"

+ " name: 'NULL4',\n"

+ " className: '"

+ Smalls.Null4Function.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'NULL8',\n"

+ " className: '"

+ Smalls.Null8Function.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " className: '"

+ Smalls.MultipleFunction.class.getName()

+ "',\n"

+ " methodName: '*'\n"

+ " },\n"

+ " {\n"

+ " className: '"

+ Smalls.AllTypesFunction.class.getName()

+ "',\n"

+ " methodName: '*'\n"

+ " },\n"

+ " {\n"

+ " name: 'UNBASE64',\n"

+ " className: '"

+ Smalls.MyUnbase64Function.class.getName()

+ "'\n"

+ " },\n"

+ " {\n"

+ " name: 'CHARACTERARRAY',\n"

+ " className: '"

+ Smalls.CharacterArrayFunction.class.getName()

+ "'\n"

+ " }\n"

+ " ]\n"

+ " }\n"

+ " ]\n"

+ "}";

return CalciteAssert.model(model);

}

/** Tests a user-defined function that is defined in terms of a class with

* non-static methods. */

@Disabled("[CALCITE-1561] Intermittent test failures")

@Test void testUserDefinedFunction() {

final String sql = "select \"adhoc\".my_plus(\"deptno\", 100) as p\n"

+ "from \"adhoc\".EMPLOYEES";

final AtomicInteger c = Smalls.MyPlusFunction.INSTANCE_COUNT.get();

final int before = c.get();

withUdf().query(sql).returnsUnordered("P=110",

"P=120",

"P=110",

"P=110");

final int after = c.get();

assertThat(after, is(before + 4));

}

/** As {@link #testUserDefinedFunction()}, but checks that the class is

* instantiated exactly once, per

* <a href="/?originalUrl=https%3A%2F%2Fgithub.com%2F%26quot%3Bhttps%3A%2F%2Fissues.apache.org%2Fjira%2Fbrowse%2FCALCITE-1548%26quot%3B%26gt%3B%5BCALCITE-1548%5D%253C%2Fdiv">