/* * Copyright 2011, Mysema Ltd * * Licensed 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 com.mysema.query.sql; import java.util.ArrayList; import java.util.List; import com.mysema.query.types.ConstantImpl; import com.mysema.query.types.Expression; import com.mysema.query.types.Template; import com.mysema.query.types.TemplateExpression; import com.mysema.query.types.TemplateExpressionImpl; import com.mysema.query.types.TemplateFactory; import com.mysema.query.types.Visitor; import com.mysema.query.types.expr.SimpleExpression; /** * Represents a table valued function call * * @author tiwe * * @param */ public class RelationalFunctionCall extends SimpleExpression implements TemplateExpression { private static final long serialVersionUID = 256739044928186923L; private static final Template createTemplate(String function, int argCount) { StringBuilder builder = new StringBuilder(); builder.append(function); builder.append("("); for (int i = 0; i < argCount; i++) { if (i > 0) builder.append(", "); builder.append("{"+ i + "}"); } builder.append(")"); return TemplateFactory.DEFAULT.create(builder.toString()); } private static final List> normalizeArgs(Object... args) { List> expressions = new ArrayList>(); for (Object arg : args) { if (arg instanceof Expression) { expressions.add((Expression)arg); } else { expressions.add(new ConstantImpl(arg)); } } return expressions; } private final TemplateExpression templateMixin; /** * Create a new TableValuedFunctionCall for the given function and arguments * * @param type * @param function * @param args * @return */ public static RelationalFunctionCall create(Class type, String function, Object... args) { return new RelationalFunctionCall(type, function, args); } public RelationalFunctionCall(Class type, String function, Object... args) { super(new TemplateExpressionImpl(type, createTemplate(function, args.length), normalizeArgs(args))); templateMixin = (TemplateExpression)mixin; } @Override public final R accept(Visitor v, C context) { return v.visit(this, context); } @Override public Expression getArg(int index) { return templateMixin.getArg(index); } @Override public List> getArgs() { return templateMixin.getArgs(); } @Override public Template getTemplate() { return templateMixin.getTemplate(); } }