/*
* Copyright (c) 2011. The Apache Software Foundation
*
* 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.expreval.expr.function;
import org.apache.expreval.client.NullColumnValueException;
import org.apache.expreval.client.ResultMissingColumnException;
import org.apache.expreval.expr.DelegateStmt;
import org.apache.expreval.expr.node.GenericValue;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.hbql.client.HBqlException;
import org.apache.hadoop.hbase.hbql.impl.AggregateValue;
import org.apache.hadoop.hbase.hbql.impl.HConnectionImpl;
import org.apache.hadoop.hbase.hbql.impl.InvalidFunctionException;
import java.util.List;
public class DelegateFunction extends DelegateStmt<GenericFunction> {
private final String functionName;
public DelegateFunction(final String functionName, final List<GenericValue> exprList) {
super(null, exprList);
this.functionName = functionName;
}
public Class<? extends GenericValue> validateTypes(final GenericValue parentExpr,
final boolean allowCollections) throws HBqlException {
final GenericFunction genericFunction = this.getFunction(this.getFunctionName(), this.getGenericValueList(), parentExpr);
this.setTypedExpr(genericFunction);
return this.getTypedExpr().validateTypes(parentExpr, false);
}
public boolean isAnAggregateValue() {
return this.getTypedExpr().isAnAggregateValue();
}
private GenericFunction getFunction(final String funcName,
final List<GenericValue> exprList,
final GenericValue parentExpr) throws InvalidFunctionException {
GenericFunction genericFunction;
genericFunction = GenericFunction.FunctionType.getFunction(funcName, exprList);
if (genericFunction != null)
return genericFunction;
genericFunction = DateFunction.IntervalType.getFunction(funcName, exprList);
if (genericFunction != null)
return genericFunction;
genericFunction = DateFunction.ConstantType.getFunction(funcName);
if (genericFunction != null)
return genericFunction;
throw new InvalidFunctionException(funcName + " in " + parentExpr.asString());
}
public GenericValue getOptimizedValue() throws HBqlException {
this.optimizeAllArgs();
return !this.isAConstant() ? this : this.getTypedExpr().getOptimizedValue();
}
public Object getValue(final HConnectionImpl conn, final Object object) throws HBqlException,
ResultMissingColumnException,
NullColumnValueException {
return this.getTypedExpr().getValue(conn, object);
}
public void initAggregateValue(final AggregateValue aggregateValue) throws HBqlException {
this.getTypedExpr().initAggregateValue(aggregateValue);
}
public void applyResultToAggregateValue(final AggregateValue aggregateValue,
final Result result) throws HBqlException,
ResultMissingColumnException,
NullColumnValueException {
this.getTypedExpr().applyResultToAggregateValue(aggregateValue, result);
}
public String getFunctionName() {
return this.functionName;
}
public String asString() {
return this.getFunctionName() + super.asString();
}
}