/* Copyright (c) 2001-2010, The HSQL Development Group
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* Neither the name of the HSQL Development Group nor the names of its
* contributors may be used to endorse or promote products derived from this
* software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG,
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.hsqldb;
import org.hsqldb.error.Error;
import org.hsqldb.error.ErrorCode;
import org.hsqldb.lib.ArrayListIdentity;
import org.hsqldb.lib.HsqlList;
import org.hsqldb.lib.Set;
import org.hsqldb.result.Result;
import org.hsqldb.store.ValuePool;
import org.hsqldb.types.Type;
/**
* Implementation of SQL-invoked user-defined function calls - PSM and JRT
*
* @author Fred Toussi (fredt@users dot sourceforge.net)
* @version 1.9.0
* @since 1.9.0
*/
public class FunctionSQLInvoked extends Expression {
RoutineSchema routineSchema;
Routine routine;
FunctionSQLInvoked(RoutineSchema routineSchema) {
super(routineSchema.isAggregate() ? OpTypes.USER_AGGREGATE
: OpTypes.FUNCTION);
this.routineSchema = routineSchema;
}
public void setArguments(Expression[] newNodes) {
this.nodes = newNodes;
}
public HsqlList resolveColumnReferences(RangeVariable[] rangeVarArray,
int rangeCount, HsqlList unresolvedSet, boolean acceptsSequences) {
if (isSelfAggregate()) {
if (unresolvedSet == null) {
unresolvedSet = new ArrayListIdentity();
}
unresolvedSet.add(this);
return unresolvedSet;
} else {
return super.resolveColumnReferences(rangeVarArray, rangeCount,
unresolvedSet,
acceptsSequences);
}
}
public void resolveTypes(Session session, Expression parent) {
Type[] types = new Type[nodes.length];
for (int i = 0; i < nodes.length; i++) {
Expression e = nodes[i];
e.resolveTypes(session, this);
types[i] = e.dataType;
}
routine = routineSchema.getSpecificRoutine(types);
for (int i = 0; i < nodes.length; i++) {
if (nodes[i].dataType == null) {
nodes[i].dataType = routine.getParameterTypes()[i];
}
}
dataType = routine.getReturnType();
}
private Object getValueInternal(Session session, Object[] aggregateData) {
boolean isValue = false;
int variableCount = routine.getVariableCount();
Result result;
int extraArg = routine.javaMethodWithConnection ? 1
: 0;
Object[] data = ValuePool.emptyObjectArray;
boolean push = true;
if (extraArg + nodes.length > 0) {
if (opType == OpTypes.USER_AGGREGATE) {
data = new Object[routine.getParameterCount()];
for (int i = 0; i < aggregateData.length; i++) {
data[i + 1] = aggregateData[i];
}
} else {
data = new Object[nodes.length + extraArg];
}
if (extraArg > 0) {
data[0] = session.getInternalConnection();
}
}
Type[] dataTypes = routine.getParameterTypes();
for (int i = 0; i < nodes.length; i++) {
Expression e = nodes[i];
Object value = e.getValue(session, dataTypes[i]);
if (value == null) {
if (routine.isNullInputOutput()) {
return null;
}
if (!routine.getParameter(i).isNullable()) {
return Result.newErrorResult(
Error.error(ErrorCode.X_39004));
}
}
if (routine.isPSM()) {
data[i] = value;
} else {
data[i + extraArg] = e.dataType.convertSQLToJava(session,
value);
}
}
if (push) {
session.sessionContext.push();
}
if (routine.isPSM()) {
try {
session.sessionContext.routineArguments = data;
session.sessionContext.routineVariables =
ValuePool.emptyObjectArray;
if (variableCount > 0) {
session.sessionContext.routineVariables =
new Object[variableCount];
}
result = routine.statement.execute(session);
if (aggregateData != null) {
for (int i = 0; i < aggregateData.length; i++) {
aggregateData[i] = data[i + 1];
}
}
} catch (Throwable e) {
result = Result.newErrorResult(e);
}
} else {
if (opType == OpTypes.USER_AGGREGATE) {
data = routine.convertArgsToJava(session, data);
}
result = routine.invokeJavaMethod(session, data);
if (opType == OpTypes.USER_AGGREGATE) {
Object[] callResult = new Object[data.length];
routine.convertArgsToSQL(session, callResult, data);
for (int i = 0; i < aggregateData.length; i++) {
aggregateData[i] = callResult[i + 1];
}
}
}
if (push) {
session.sessionContext.pop();
}
if (result.isError()) {
throw result.getException();
}
if (isValue) {
return result.valueData;
} else {
return result;
}
}
public Object getValue(Session session) {
if (opType == OpTypes.SIMPLE_COLUMN) {
Object[] data =
session.sessionContext.rangeIterators[rangePosition]
.getCurrent();
return data[columnIndex];
}
Object returnValue = getValueInternal(session, null);
if (returnValue instanceof Result) {
Result result = (Result) returnValue;
if (result.isError()) {
throw result.getException();
} else if (result.isSimpleValue()) {
returnValue = result.getValueObject();
} else if (result.isData()) {
returnValue = result;
} else {
throw Error.error(ErrorCode.X_2F005, routine.getName().name);
}
}
return returnValue;
}
public Result getResult(Session session) {
Object value = getValueInternal(session, null);
if (value instanceof Result) {
return (Result) value;
}
return Result.newPSMResult(value);
}
void collectObjectNames(Set set) {
set.add(routine.getSpecificName());
}
public String getSQL() {
StringBuffer sb = new StringBuffer();
sb.append(routineSchema.getName().getSchemaQualifiedStatementName());
sb.append('(');
int nodeCount = nodes.length;
if (opType == OpTypes.USER_AGGREGATE) {
nodeCount = 1;
}
for (int i = 0; i < nodeCount; i++) {
if (i != 0) {
sb.append(',');
}
sb.append(nodes[i].getSQL());
}
sb.append(')');
return sb.toString();
}
public String describe(Session session, int blanks) {
return super.describe(session, blanks);
}
boolean isSelfAggregate() {
return routineSchema.isAggregate();
}
public boolean isDeterministic() {
return routine.isDeterministic();
}
public Object updateAggregatingValue(Session session, Object currValue) {
Object[] array = (Object[]) currValue;
if (array == null) {
array = new Object[3];
}
array[0] = Boolean.FALSE;
getValueInternal(session, array);
return array;
}
public Object getAggregatedValue(Session session, Object currValue) {
Object[] array = (Object[]) currValue;
if (array == null) {
array = new Object[3];
}
array[0] = Boolean.TRUE;
Result result = (Result) getValueInternal(session, array);
Object returnValue;
if (result.isError()) {
throw result.getException();
} else {
return result.getValueObject();
}
}
}