org.jooq.impl.AbstractRoutine Maven / Gradle / Ivy
/**
* Copyright (c) 2009-2014, Data Geekery GmbH (http://www.datageekery.com)
* All rights reserved.
*
* This work is dual-licensed
* - under the Apache Software License 2.0 (the "ASL")
* - under the jOOQ License and Maintenance Agreement (the "jOOQ License")
* =============================================================================
* You may choose which license applies to you:
*
* - If you're using this work with Open Source databases, you may choose
* either ASL or jOOQ License.
* - If you're using this work with at least one commercial database, you must
* choose jOOQ License
*
* For more information, please visit http://www.jooq.org/licenses
*
* Apache Software License 2.0:
* -----------------------------------------------------------------------------
* 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.
*
* jOOQ License and Maintenance Agreement:
* -----------------------------------------------------------------------------
* Data Geekery grants the Customer the non-exclusive, timely limited and
* non-transferable license to install and use the Software under the terms of
* the jOOQ License and Maintenance Agreement.
*
* This library is distributed with a LIMITED WARRANTY. See the jOOQ License
* and Maintenance Agreement for more details: http://www.jooq.org/licensing
*/
package org.jooq.impl;
import static org.jooq.Clause.FIELD;
import static org.jooq.Clause.FIELD_FUNCTION;
import static org.jooq.SQLDialect.POSTGRES;
// ...
import static org.jooq.impl.DSL.function;
import static org.jooq.impl.DSL.table;
import static org.jooq.impl.DSL.using;
import static org.jooq.impl.DSL.val;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.jooq.AggregateFunction;
// ...
import org.jooq.AttachableInternal;
import org.jooq.BindContext;
import org.jooq.Clause;
import org.jooq.Configuration;
import org.jooq.Context;
import org.jooq.DSLContext;
import org.jooq.DataType;
import org.jooq.ExecuteContext;
import org.jooq.ExecuteListener;
import org.jooq.Field;
import org.jooq.Package;
import org.jooq.Parameter;
import org.jooq.RenderContext;
import org.jooq.Result;
import org.jooq.Routine;
import org.jooq.Schema;
import org.jooq.UDTField;
import org.jooq.UDTRecord;
import org.jooq.tools.Convert;
/**
* A common base class for stored procedures
*
* This type is for JOOQ INTERNAL USE only. Do not reference directly
*
* @author Lukas Eder
*/
public abstract class AbstractRoutine extends AbstractQueryPart implements Routine, AttachableInternal {
/**
* Generated UID
*/
private static final long serialVersionUID = 6330037113167106443L;
private static final Clause[] CLAUSES = { FIELD, FIELD_FUNCTION };
// ------------------------------------------------------------------------
// Meta-data attributes (the same for every call)
// ------------------------------------------------------------------------
private final Schema schema;
private final Package pkg;
private final String name;
private final List> allParameters;
private final List> inParameters;
private final List> outParameters;
private final DataType type;
private Parameter returnParameter;
private boolean overloaded;
// ------------------------------------------------------------------------
// Call-data attributes (call-specific)
// ------------------------------------------------------------------------
private final Map, Field> inValues;
private final Set> inValuesNonDefaulted;
private transient Field function;
private Configuration configuration;
private final Map, Object> results;
private final Map, Integer> parameterIndexes;
// ------------------------------------------------------------------------
// Constructors
// ------------------------------------------------------------------------
protected AbstractRoutine(String name, Schema schema) {
this(name, schema, null, null);
}
protected AbstractRoutine(String name, Schema schema, Package pkg) {
this(name, schema, pkg, null);
}
protected AbstractRoutine(String name, Schema schema, DataType type) {
this(name, schema, null, type);
}
protected AbstractRoutine(String name, Schema schema, Package pkg, DataType type) {
this.parameterIndexes = new HashMap, Integer>();
this.schema = schema;
this.pkg = pkg;
this.name = name;
this.allParameters = new ArrayList>();
this.inParameters = new ArrayList>();
this.outParameters = new ArrayList>();
this.inValues = new HashMap, Field>();
this.inValuesNonDefaulted = new HashSet>();
this.results = new HashMap, Object>();
this.type = type;
}
// ------------------------------------------------------------------------
// Initialise a routine call
// ------------------------------------------------------------------------
protected final void setNumber(Parameter parameter, Number value) {
setValue(parameter, Convert.convert(value, parameter.getType()));
}
protected final void setNumber(Parameter parameter, Field value) {
setField(parameter, value);
}
protected final void setValue(Parameter parameter, Object value) {
setField(parameter, val(value, parameter.getDataType()));
}
/*
* #326 - Avoid overloading setValue()
*/
protected final void setField(Parameter parameter, Field value) {
// Be sure null is correctly represented as a null field
if (value == null) {
setField(parameter, val(null, parameter.getDataType()));
}
// [#1183] Add the field to the in-values and mark them as non-defaulted
else {
inValues.put(parameter, value);
inValuesNonDefaulted.add(parameter);
}
}
// ------------------------------------------------------------------------
// Call the routine
// ------------------------------------------------------------------------
@Override
public final void attach(Configuration c) {
configuration = c;
}
@Override
public final void detach() {
attach(null);
}
@Override
public final Configuration configuration() {
return configuration;
}
@Override
public final int execute(Configuration c) {
// Ensure that all depending Attachables are attached
Configuration previous = configuration();
try {
attach(c);
return execute();
}
finally {
attach(previous);
}
}
@Override
public final int execute() {
// Procedures (no return value) are always executed as CallableStatement
if (type == null) {
return executeCallableStatement();
}
else {
switch (configuration.dialect().family()) {
// [#852] Some RDBMS don't allow for using JDBC procedure escape
// syntax for functions. Select functions from DUAL instead
case HSQLDB:
// [#692] HSQLDB cannot SELECT f() FROM [...] when f()
// returns a cursor. Instead, SELECT * FROM table(f()) works
if (SQLDataType.RESULT.equals(type.getSQLDataType())) {
return executeSelectFrom();
}
// Fall through
else {
}
case H2:
/* [pro] xx
xxxx xxxx
xx xxxxxx xxxxxxxxxxxxxxxxxxxxxxxxxxx xxxxxxx xxxx xxx
xxxx xxxxxxx
xx [/pro] */
return executeSelect();
// [#773] If JDBC escape syntax is available for functions, use
// it to prevent transactional issues when functions issue
// DML statements
default:
return executeCallableStatement();
}
}
}
private final int executeSelectFrom() {
DSLContext create = create(configuration);
Result result = create.selectFrom(table(asField())).fetch();
results.put(returnParameter, result);
return 0;
}
private final int executeSelect() {
final Field field = asField();
results.put(returnParameter, create(configuration).select(field).fetchOne(field));
return 0;
}
private final int executeCallableStatement() {
ExecuteContext ctx = new DefaultExecuteContext(configuration, this);
ExecuteListener listener = new ExecuteListeners(ctx);
try {
Connection connection = ctx.connection();
listener.renderStart(ctx);
// [#1520] TODO: Should the number of bind values be checked, here?
ctx.sql(create(configuration).render(this));
listener.renderEnd(ctx);
listener.prepareStart(ctx);
ctx.statement(connection.prepareCall(ctx.sql()));
// [#1856] TODO: Add Statement flags like timeout here
listener.prepareEnd(ctx);
listener.bindStart(ctx);
using(configuration).bindContext(ctx.statement()).visit(this);
registerOutParameters(configuration, (CallableStatement) ctx.statement());
listener.bindEnd(ctx);
listener.executeStart(ctx);
ctx.statement().execute();
listener.executeEnd(ctx);
fetchOutParameters(ctx);
return 0;
}
catch (SQLException e) {
ctx.sqlException(e);
listener.exception(ctx);
throw ctx.exception();
}
finally {
Utils.safeClose(listener, ctx);
}
}
@Override
public final Clause[] clauses(Context ctx) {
return CLAUSES;
}
@Override
public final void bind(BindContext context) {
for (Parameter parameter : getParameters()) {
// [#1183] Skip defaulted parameters
if (getInParameters().contains(parameter) && !inValuesNonDefaulted.contains(parameter)) {
continue;
}
int index = context.peekIndex();
parameterIndexes.put(parameter, index);
if (getInValues().get(parameter) != null) {
context.visit(getInValues().get(parameter));
// [#391] This happens when null literals are used as IN/OUT
// parameters. They're not bound as in value, but they need to
// be registered as OUT parameter
if (index == context.peekIndex() && getOutParameters().contains(parameter)) {
context.nextIndex();
}
}
// Skip one index for OUT parameters
else {
context.nextIndex();
}
}
}
@Override
public final void toSQL(RenderContext context) {
toSQLBegin(context);
if (getReturnParameter() != null) {
toSQLAssign(context);
}
toSQLCall(context);
context.sql("(");
String separator = "";
for (Parameter parameter : getParameters()) {
// The return value has already been written
if (parameter.equals(getReturnParameter())) {
continue;
}
// OUT and IN OUT parameters are always written as a '?' bind variable
else if (getOutParameters().contains(parameter)) {
context.sql(separator);
toSQLOutParam(context, parameter);
}
// [#1183] Omit defaulted parameters
else if (!inValuesNonDefaulted.contains(parameter)) {
continue;
}
// IN parameters are rendered normally
else {
Field value = getInValues().get(parameter);
// Disambiguate overloaded procedure signatures
if (POSTGRES == context.configuration().dialect() && isOverloaded()) {
value = value.cast(parameter.getType());
}
context.sql(separator);
toSQLInParam(context, parameter, value);
}
separator = ", ";
}
context.sql(")");
toSQLEnd(context);
}
private final void toSQLEnd(RenderContext context) {
switch (context.configuration().dialect().family()) {
/* [pro] xx
xxxx xxxxxxx
xxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxx
xxxxxx
xx [/pro] */
default:
context.sql(" }");
break;
}
}
private final void toSQLBegin(RenderContext context) {
switch (context.configuration().dialect().family()) {
/* [pro] xx
xxxx xxxxxxx
xxxxxxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxxx
xxxxxx
xx [/pro] */
default:
context.sql("{ ");
break;
}
}
private final void toSQLAssign(RenderContext context) {
switch (context.configuration().dialect().family()) {
/* [pro] xx
xxxx xxxxxxx
xxxxxxxxxxxxxx xx xxx
xxxxxx
xx [/pro] */
default:
context.sql("? = ");
break;
}
}
private final void toSQLCall(RenderContext context) {
switch (context.configuration().dialect().family()) {
/* [pro] xx
xxxx xxxxxxx
xxxxxx
xx [/pro] */
default:
context.sql("call ");
break;
}
toSQLQualifiedName(context);
}
private final void toSQLOutParam(RenderContext context, Parameter parameter) {
/* [pro] xx
xxxxxx xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx x
xxxx xxxxxxx
xxxxxxxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxx xx xxx
xxxxxx
x
xx [/pro] */
context.sql("?");
}
private final void toSQLInParam(RenderContext context, Parameter parameter, Field value) {
/* [pro] xx
xxxxxx xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx x
xxxx xxxxxxx
xxxxxxxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxx xx xxx
xxxxxx
x
xx [/pro] */
context.visit(value);
}
private final void toSQLQualifiedName(RenderContext context) {
Schema mappedSchema = Utils.getMappedSchema(context.configuration(), getSchema());
if (context.qualify()) {
if (mappedSchema != null) {
context.visit(mappedSchema);
context.sql(".");
}
/* [pro] xx
xx xxxxxxx xx xxx xxxxxxx xxxxxxxx xxxxxx xxxx xx xx xxxxx xxxxxxxxx
xxxx xx xxxxxxxxxxxx xx xxxx xx xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx xx xxxxxxxxxx x
xxxxxxxxxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxx
x
xx [/pro] */
if (getPackage() != null) {
context.visit(getPackage());
context.sql(".");
}
}
context.literal(getName());
}
private final void fetchOutParameters(ExecuteContext ctx) throws SQLException {
for (Parameter parameter : getParameters()) {
if (parameter.equals(getReturnParameter()) ||
getOutParameters().contains(parameter)) {
int index = parameterIndexes.get(parameter);
results.put(parameter, Utils.getFromStatement(ctx, parameter.getType(), index));
}
}
}
@SuppressWarnings("unchecked")
private final void registerOutParameters(Configuration c, CallableStatement statement) throws SQLException {
// Register all out / inout parameters according to their position
// Note that some RDBMS do not support binding by name very well
for (Parameter parameter : getParameters()) {
if (parameter.equals(getReturnParameter()) ||
getOutParameters().contains(parameter)) {
int index = parameterIndexes.get(parameter);
int sqlType = parameter.getDataType().getDataType(c).getSQLType();
switch (c.dialect().family()) {
/* [pro] xx
xx xxx xxxx xxxx xxxxxxx xxxxx xxxxxx xxxxx xx xxxx
xx xxxx xxx xxxx xxxx
xxxx xxxxxxx x
xx xxxxxxxx xx xxxxxxxxxxxxx x
xxxxxxxxxxxx xxxxxx x xxxxx
xxxxxxxxxxxxxxxxxxx xxxxxxx xxxxxxxxxxxxxx xxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx xxxxxxxxxxxxx xxxxxxxxxxxxxxxxxxxxxxxxx
x
xxxx xx xxxxxxxx xx xxxxxxxxxxxx x
xxxxxxxxxxxxxx xxxxxx x xxxxxxxxxxxxxxxxxxxxx
xxxxxxxx xxxxxxx xxxxxxxxxxxxxxxx xxxxxxxxxxxxxxxxxxxx xxx
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx xxxxxxxxxxxx xxxxxxxxxxxxxxxxxx
x
xx xxx xxxxxxx xxxxxxxxx xx xxx xx xxxxxxxx x xxxx
xx xxxxxxx
xxxx x
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx xxxxxxxxx
x
xxxxxx
x
xx [/pro] */
default: {
statement.registerOutParameter(index, sqlType);
break;
}
}
}
}
}
// ------------------------------------------------------------------------
// Fetch routine results
// ------------------------------------------------------------------------
@Override
public final T getReturnValue() {
if (returnParameter != null) {
return getValue(returnParameter);
}
return null;
}
@SuppressWarnings("unchecked")
protected final Z getValue(Parameter parameter) {
return (Z) results.get(parameter);
}
protected final Map, Field> getInValues() {
return inValues;
}
// ------------------------------------------------------------------------
// Access to routine configuration objects
// ------------------------------------------------------------------------
@Override
public final List> getOutParameters() {
return Collections.unmodifiableList(outParameters);
}
@Override
public final List> getInParameters() {
return Collections.unmodifiableList(inParameters);
}
@Override
public final List> getParameters() {
return Collections.unmodifiableList(allParameters);
}
@Override
public final Schema getSchema() {
return schema;
}
@Override
public final Package getPackage() {
return pkg;
}
@Override
public final String getName() {
return name;
}
protected final Parameter getReturnParameter() {
return returnParameter;
}
protected final void setOverloaded(boolean overloaded) {
this.overloaded = overloaded;
}
protected final boolean isOverloaded() {
return overloaded;
}
protected final void addInParameter(Parameter parameter) {
inParameters.add(parameter);
allParameters.add(parameter);
// IN parameters are initialised with null by default
inValues.put(parameter, val(null, parameter.getDataType()));
// [#1183] non-defaulted parameters are marked as such
if (!parameter.isDefaulted()) {
inValuesNonDefaulted.add(parameter);
}
}
protected final void addInOutParameter(Parameter parameter) {
addInParameter(parameter);
outParameters.add(parameter);
}
protected final void addOutParameter(Parameter parameter) {
outParameters.add(parameter);
allParameters.add(parameter);
}
protected final void setReturnParameter(Parameter parameter) {
returnParameter = parameter;
allParameters.add(parameter);
}
public final Field asField() {
if (function == null) {
function = new RoutineField();
}
return function;
}
public final Field asField(String alias) {
return asField().as(alias);
}
public final AggregateFunction asAggregateFunction() {
Field[] array = new Field[getInParameters().size()];
int i = 0;
for (Parameter p : getInParameters()) {
array[i] = getInValues().get(p);
i++;
}
// [#2393] Fully qualify custom aggregate functions.
// TODO: Merge this code into RoutineField!
List names = new ArrayList();
if (schema != null) {
names.add(schema.getName());
}
if (pkg != null) {
names.add(pkg.getName());
}
names.add(name);
return (AggregateFunction) function(DSL.name(names.toArray(new String[names.size()])), type, array);
}
/**
* Subclasses may call this method to create {@link UDTField} objects that
* are linked to this table.
*
* @param name The name of the field (case-sensitive!)
* @param type The data type of the field
*/
protected static final Parameter createParameter(String name, DataType type) {
return createParameter(name, type, false);
}
/**
* Subclasses may call this method to create {@link UDTField} objects that
* are linked to this table.
*
* @param name The name of the field (case-sensitive!)
* @param type The data type of the field
* @param isDefaulted Whether the parameter is defaulted (see
* {@link Parameter#isDefaulted()}
*/
protected static final Parameter createParameter(String name, DataType type, boolean isDefaulted) {
return new ParameterImpl(name, type, isDefaulted);
}
/**
* The {@link Field} representation of this {@link Routine}
*/
private class RoutineField extends AbstractFunction {
/**
* Generated UID
*/
private static final long serialVersionUID = -5730297947647252624L;
RoutineField() {
super(AbstractRoutine.this.getName(),
AbstractRoutine.this.type);
}
@Override
final Field getFunction0(Configuration c) {
RenderContext local = create(c).renderContext();
toSQLQualifiedName(local);
Field[] array = new Field[getInParameters().size()];
int i = 0;
for (Parameter p : getInParameters()) {
// Disambiguate overloaded function signatures
if (POSTGRES == c.dialect() && isOverloaded()) {
array[i] = getInValues().get(p).cast(p.getType());
}
else {
array[i] = getInValues().get(p);
}
i++;
}
return function(local.render(), getDataType(), array);
}
}
// ------------------------------------------------------------------------
// XXX: Object API
// ------------------------------------------------------------------------
@Override
public int hashCode() {
// [#1938] This is a much more efficient hashCode() implementation
// compared to that of standard QueryParts
return name.hashCode();
}
}