package org.jooq.impl;
import static java.lang.Boolean.TRUE;
import static org.jooq.Clause.FIELD;
import static org.jooq.Clause.FIELD_FUNCTION;
import static org.jooq.SQLDialect.FIREBIRD;
import static org.jooq.SQLDialect.POSTGRES;
import static org.jooq.XMLFormat.RecordFormat.COLUMN_NAME_ELEMENTS;
import static org.jooq.conf.ThrowExceptions.THROW_NONE;
import static org.jooq.impl.DSL.field;
import static org.jooq.impl.DSL.function;
import static org.jooq.impl.DSL.inline;
import static org.jooq.impl.DSL.name;
import static org.jooq.impl.DSL.param;
import static org.jooq.impl.DSL.sql;
import static org.jooq.impl.DSL.table;
import static org.jooq.impl.DSL.using;
import static org.jooq.impl.DSL.val;
import static org.jooq.impl.Keywords.K_BEGIN;
import static org.jooq.impl.Keywords.K_BOOLEAN;
import static org.jooq.impl.Keywords.K_CASE;
import static org.jooq.impl.Keywords.K_COLUMNS;
import static org.jooq.impl.Keywords.K_DECLARE;
import static org.jooq.impl.Keywords.K_ELSE;
import static org.jooq.impl.Keywords.K_END;
import static org.jooq.impl.Keywords.K_FALSE;
import static org.jooq.impl.Keywords.K_FOR;
import static org.jooq.impl.Keywords.K_FROM;
import static org.jooq.impl.Keywords.K_FUNCTION;
import static org.jooq.impl.Keywords.K_IS;
import static org.jooq.impl.Keywords.K_NOT;
import static org.jooq.impl.Keywords.K_NULL;
import static org.jooq.impl.Keywords.K_OPEN;
import static org.jooq.impl.Keywords.K_PASSING;
import static org.jooq.impl.Keywords.K_RECORD;
import static org.jooq.impl.Keywords.K_RETURN;
import static org.jooq.impl.Keywords.K_SELECT;
import static org.jooq.impl.Keywords.K_THEN;
import static org.jooq.impl.Keywords.K_TRUE;
import static org.jooq.impl.Keywords.K_TYPE;
import static org.jooq.impl.Keywords.K_WHEN;
import static org.jooq.impl.Keywords.K_XMLTABLE;
import static org.jooq.impl.SQLDataType.INTEGER;
import static org.jooq.impl.SQLDataType.NUMERIC;
import static org.jooq.impl.Tools.EMPTY_FIELD;
import static org.jooq.impl.Tools.executeStatementAndGetFirstResultSet;
import static org.jooq.impl.Tools.settings;
import static org.jooq.impl.Tools.DataKey.DATA_TOP_LEVEL_CTE;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import org.jooq.AggregateFunction;
import org.jooq.BindContext;
import org.jooq.Binding;
import org.jooq.Catalog;
import org.jooq.Clause;
import org.jooq.Configuration;
import org.jooq.Context;
import org.jooq.Converter;
import org.jooq.DSLContext;
import org.jooq.DataType;
import org.jooq.ExecuteContext;
import org.jooq.ExecuteListener;
import org.jooq.Field;
import org.jooq.Name;
import org.jooq.Package;
import org.jooq.Param;
import org.jooq.Parameter;
import org.jooq.QueryPart;
import org.jooq.Record;
import org.jooq.RenderContext;
import org.jooq.Result;
import org.jooq.Results;
import org.jooq.Routine;
import org.jooq.SQLDialect;
import org.jooq.Schema;
import org.jooq.Statement;
import org.jooq.UDT;
import org.jooq.UDTField;
import org.jooq.UDTRecord;
import org.jooq.XMLFormat;
import org.jooq.conf.SettingsTools;
import org.jooq.exception.ControlFlowSignal;
import org.jooq.exception.MappingException;
import org.jooq.impl.ResultsImpl.ResultOrRowsImpl;
import org.jooq.tools.Convert;
import org.jooq.tools.reflect.Reflect;
@org.jooq.Internal
public abstract class AbstractRoutine<T> extends AbstractNamed implements Routine<T> {
private static final long serialVersionUID = 6330037113167106443L;
private static final Clause[] CLAUSES = { FIELD, FIELD_FUNCTION };
private final Schema schema;
private final List<Parameter<?>> allParameters;
private final List<Parameter<?>> inParameters;
private final List<Parameter<?>> outParameters;
private final DataType<T> type;
private Parameter<T> returnParameter;
private ResultsImpl results;
private boolean overloaded;
private boolean hasUnnamedParameters;
private final Map<Parameter<?>, Field<?>> inValues;
private final Set<Parameter<?>> inValuesDefaulted;
private final Set<Parameter<?>> inValuesNonDefaulted;
private transient Field<T> function;
private Configuration configuration;
private final Map<Parameter<?>, Object> outValues;
private final Map<Parameter<?>, Integer> resultIndexes;
protected AbstractRoutine(String name, Schema schema) {
this(name, schema, null, null, null, null);
}
protected AbstractRoutine(String name, Schema schema, Package pkg) {
this(name, schema, pkg, null, null, null);
}
protected AbstractRoutine(String name, Schema schema, DataType<T> type) {
this(name, schema, null, type, null, null);
}
protected <X> AbstractRoutine(String name, Schema schema, DataType<X> type, Converter<X, T> converter) {
this(name, schema, null, type, converter, null);
}
protected <X> AbstractRoutine(String name, Schema schema, DataType<X> type, Binding<X, T> binding) {
this(name, schema, null, type, null, binding);
}
protected <X, Y> AbstractRoutine(String name, Schema schema, DataType<X> type, Converter<Y, T> converter, Binding<X, Y> binding) {
this(name, schema, null, type, converter, binding);
}
protected AbstractRoutine(String name, Schema schema, Package pkg, DataType<T> type) {
this(name, schema, pkg, type, null, null);
}
protected <X> AbstractRoutine(String name, Schema schema, Package pkg, DataType<X> type, Converter<X, T> converter) {
this(name, schema, pkg, type, converter, null);
}
protected <X> AbstractRoutine(String name, Schema schema, Package pkg, DataType<X> type, Binding<X, T> binding) {
this(name, schema, pkg, type, null, binding);
}
@SuppressWarnings({ "unchecked", "rawtypes" })
protected <X, Y> AbstractRoutine(String name, Schema schema, Package pkg, DataType<X> type, Converter<Y, T> converter, Binding<X, Y> binding) {
super(qualify(pkg != null ? pkg : schema, DSL.name(name)), CommentImpl.NO_COMMENT);
this.resultIndexes = new HashMap<>();
this.schema = schema;
this.allParameters = new ArrayList<>();
this.inParameters = new ArrayList<>();
this.outParameters = new ArrayList<>();
this.results = new ResultsImpl(null);
this.inValues = new HashMap<>();
this.inValuesDefaulted = new HashSet<>();
this.inValuesNonDefaulted = new HashSet<>();
this.outValues = new HashMap<>();
this.type = converter == null && binding == null
? (DataType<T>) type
: type.asConvertedDataType(DefaultBinding.newBinding((Converter) converter, type, binding));
}
protected final <N extends Number> void setNumber(Parameter<N> parameter, Number value) {
setValue(parameter, Convert.convert(value, parameter.getType()));
}
protected final void setNumber(Parameter<? extends Number> parameter, Field<? extends Number> value) {
setField(parameter, value);
}
@Override
public final <Z> void setValue(Parameter<Z> parameter, Z value) {
set(parameter, value);
}
@Override
public final <Z> void set(Parameter<Z> parameter, Z value) {
setField(parameter, val(value, parameter.getDataType()));
}
protected final void setField(Parameter<?> parameter, Field<?> value) {
if (value == null) {
setField(parameter, val(null, parameter.getDataType()));
}
else {
inValues.put(parameter, value);
inValuesDefaulted.remove(parameter);
inValuesNonDefaulted.add(parameter);
}
}
@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) {
Configuration previous = configuration();
try {
attach(c);
return execute();
}
finally {
attach(previous);
}
}
@Override
public final int execute() {
SQLDialect family = configuration.family();
results.clear();
outValues.clear();
if ( family == POSTGRES) {
return executeSelectFromPOSTGRES();
}
else if (type == null) {
return executeCallableStatement();
}
else {
switch (family) {
case HSQLDB:
if (SQLDataType.RESULT.equals(type.getSQLDataType())) {
return executeSelectFromHSQLDB();
}
else {
}
case H2:
return executeSelect();
default:
return executeCallableStatement();
}
}
}
private final int executeSelectFromHSQLDB() {
DSLContext create = create(configuration);
Result<?> result = create.selectFrom(table(asField())).fetch();
outValues.put(returnParameter, result);
return 0;
}
private final int executeSelectFromPOSTGRES() {
DSLContext create = create(configuration);
List<Field<?>> fields = new ArrayList<>(1 + outParameters.size());
if (returnParameter != null)
fields.add(DSL.field(DSL.name(getName()), returnParameter.getDataType()));
for (Parameter<?> p : outParameters)
fields.add(DSL.field(DSL.name(p.getName()), p.getDataType()));
Result<?> result = create.select(fields).from("{0}", asField()).fetch();
int i = 0;
if (returnParameter != null)
outValues.put(returnParameter, returnParameter.getDataType().convert(result.getValue(0, i++)));
for (Parameter<?> p : outParameters)
outValues.put(p, p.getDataType().convert(result.getValue(0, i++)));
return 0;
}
private final int executeSelect() {
final Field<T> field = asField();
outValues.put(returnParameter, create(configuration).select(field).fetchOne(field));
return 0;
}
private final int executeCallableStatement() {
ExecuteContext ctx = new DefaultExecuteContext(configuration, this);
ExecuteListener listener = ExecuteListeners.get(ctx);
try {
listener.start(ctx);
Connection connection = ctx.connection();
listener.renderStart(ctx);
ctx.sql(create(configuration).render(this));
listener.renderEnd(ctx);
listener.prepareStart(ctx);
if (ctx.statement() == null)
ctx.statement(connection.prepareCall(ctx.sql()));
Tools.setFetchSize(ctx, 0);
listener.prepareEnd(ctx);
int t = SettingsTools.getQueryTimeout(0, ctx.settings());
if (t != 0)
ctx.statement().setQueryTimeout(t);
listener.bindStart(ctx);
using(configuration).bindContext(ctx.statement()).visit(this);
registerOutParameters(ctx);
listener.bindEnd(ctx);
SQLException e = execute0(ctx, listener);
if (ctx.family() != FIREBIRD)
Tools.consumeResultSets(ctx, listener, results, null, e);
listener.outStart(ctx);
fetchOutParameters(ctx);
listener.outEnd(ctx);
return 0;
}
catch (ControlFlowSignal e) {
throw e;
}
catch (RuntimeException e) {
ctx.exception(e);
listener.exception(ctx);
throw ctx.exception();
}
catch (SQLException e) {
ctx.sqlException(e);
listener.exception(ctx);
throw ctx.exception();
}
finally {
Tools.safeClose(listener, ctx);
}
}
private final SQLException execute0(ExecuteContext ctx, ExecuteListener listener) throws SQLException {
listener.executeStart(ctx);
SQLException e = executeStatementAndGetFirstResultSet(ctx, 0);
listener.executeEnd(ctx);
if (e != null)
results.resultsOrRows().add(new ResultOrRowsImpl(Tools.translate(ctx.sql(), e)));
return e;
}
@Override
public final Clause[] clauses(Context<?> ctx) {
return CLAUSES;
}
@Override
public void accept(Context<?> ctx) {
if (ctx instanceof RenderContext)
toSQL0((RenderContext) ctx);
else
bind0((BindContext) ctx);
}
final void bind0(BindContext context) {
List<Parameter<?>> all = getParameters0(context.configuration());
List<Parameter<?>> in = getInParameters0(context.configuration());
for (Parameter<?> parameter : all) {
if (in.contains(parameter) && inValuesDefaulted.contains(parameter))
continue;
bind1(context, parameter, getInValues().get(parameter) != null, resultParameter(context.configuration(), parameter));
}
}
private final void bind1(BindContext context, Parameter<?> parameter, boolean bindAsIn, boolean bindAsOut) {
int index = context.peekIndex();
if (bindAsOut) {
resultIndexes.put(parameter, index);
}
if (bindAsIn) {
context.visit(getInValues().get(parameter));
if (index == context.peekIndex() && bindAsOut)
context.nextIndex();
}
else {
context.nextIndex();
}
}
final void toSQL0(RenderContext context) {
toSQLDeclare(context);
toSQLBegin(context);
if (getReturnParameter0(context.configuration()) != null)
toSQLAssign(context);
toSQLCall(context);
context.sql(" (");
String separator = "";
List<Parameter<?>> all = getParameters0(context.configuration());
Map<Integer, Parameter<?>> indexes = new LinkedHashMap<>();
for (int i = 0; i < all.size(); i++) {
Parameter<?> parameter = all.get(i);
if (parameter.equals(getReturnParameter0(context.configuration())))
continue;
else if (inValuesDefaulted.contains(parameter))
continue;
else
indexes.put(i, parameter);
}
boolean indent = false;
if (indent)
context.formatIndentStart()
.formatNewLine();
int i = 0;
for (Entry<Integer, Parameter<?>> entry : indexes.entrySet()) {
Parameter<?> parameter = entry.getValue();
int index = entry.getKey();
context.sql(separator);
if (indent && i++ > 0)
context.formatNewLine();
if (getOutParameters0(context.configuration()).contains(parameter))
toSQLOutParam(context, parameter, index);
else
toSQLInParam(context, parameter, index, getInValues().get(parameter));
separator = ", ";
}
if (indent)
context.formatIndentEnd().formatNewLine();
context.sql(')');
toSQLEnd(context);
}
private final void toSQLEnd(RenderContext context) {
{
context.sql(" }");
}
}
private final void toSQLDeclare(RenderContext context) {
}
private final void toSQLBegin(RenderContext context) {
{
context.sql("{ ");
}
}
private final void toSQLAssign(RenderContext context) {
{
context.sql("? = ");
}
}
private final void toSQLCall(RenderContext context) {
{
context.sql("call ");
}
toSQLQualifiedName(context);
}
private final void toSQLOutParam(RenderContext ctx, Parameter<?> parameter, int index) {
ctx.sql('?');
}
private final void toSQLInParam(RenderContext ctx, Parameter<?> parameter, int index, Field<?> value) {
ctx.visit(value);
}
private final void toSQLQualifiedName(Context<?> ctx) {
if (ctx.qualify()) {
Schema mapped = Tools.getMappedSchema(ctx.configuration(), getSchema());
if (mapped != null && !"".equals(mapped.getName()))
ctx.visit(mapped)
.sql('.');
}
ctx.visit(getUnqualifiedName());
}
private final void fetchOutParameters(ExecuteContext ctx) throws SQLException {
for (Parameter<?> parameter : getParameters0(ctx.configuration()))
if (resultParameter(ctx.configuration(), parameter))
try {
fetchOutParameter(ctx, parameter);
}
catch (SQLException e) {
if (ctx.settings().getThrowExceptions() != THROW_NONE)
throw e;
}
}
private final <U> void fetchOutParameter(ExecuteContext ctx, Parameter<U> parameter) throws SQLException {
{
DefaultBindingGetStatementContext<U> out = new DefaultBindingGetStatementContext<>(
ctx.configuration(),
ctx.data(),
(CallableStatement) ctx.statement(),
resultIndexes.get(parameter)
);
parameter.getBinding().get(out);
outValues.put(parameter, out.value());
}
}
private final void registerOutParameters(ExecuteContext ctx) throws SQLException {
Configuration c = ctx.configuration();
Map<Object, Object> data = ctx.data();
CallableStatement statement = (CallableStatement) ctx.statement();
for (Parameter<?> parameter : getParameters0(ctx.configuration()))
if (resultParameter(c, parameter))
registerOutParameter(c, data, statement, parameter);
}
private final <U> void registerOutParameter(Configuration c, Map<Object, Object> data, CallableStatement statement, Parameter<U> parameter) throws SQLException {
parameter.getBinding().register(new DefaultBindingRegisterContext<>(c, data, statement, resultIndexes.get(parameter)));
}
@Override
public final T getReturnValue() {
if (returnParameter != null)
return getValue(returnParameter);
return null;
}
@Override
public final Results getResults() {
return results;
}
@Override
public final <Z> Z getValue(Parameter<Z> parameter) {
return get(parameter);
}
@Override
@SuppressWarnings("unchecked")
public final <Z> Z get(Parameter<Z> parameter) {
return (Z) outValues.get(parameter);
}
@Override
@SuppressWarnings("unchecked")
public final <Z> Z getInValue(Parameter<Z> parameter) {
return (Z) inValues.get(parameter);
}
protected final Map<Parameter<?>, Field<?>> getInValues() {
return inValues;
}
@Override
public final List<Parameter<?>> getOutParameters() {
return Collections.unmodifiableList(outParameters);
}
@Override
public final List<Parameter<?>> getInParameters() {
return Collections.unmodifiableList(inParameters);
}
@Override
public final List<Parameter<?>> getParameters() {
return Collections.unmodifiableList(allParameters);
}
private final List<Parameter<?>> getOutParameters0(Configuration c) {
return getOutParameters();
}
@SuppressWarnings("unused")
private final List<Parameter<?>> getInParameters0(Configuration c) {
return getInParameters();
}
private final List<Parameter<?>> getParameters0(Configuration c) {
return getParameters();
}
@Override
public final Catalog getCatalog() {
return getSchema() == null ? null : getSchema().getCatalog();
}
@Override
public final Schema getSchema() {
return schema;
}
@Override
public final Parameter<T> getReturnParameter() {
return returnParameter;
}
private final Parameter<T> getReturnParameter0(Configuration c) {
return getReturnParameter();
}
protected final void setOverloaded(boolean overloaded) {
this.overloaded = overloaded;
}
protected final boolean isOverloaded() {
return overloaded;
}
private final boolean pgArgNeedsCasting(Parameter<?> parameter) {
return isOverloaded() || parameter.getType() == Byte.class || parameter.getType() == Short.class;
}
private final boolean hasUnnamedParameters() {
return hasUnnamedParameters;
}
private final void addParameter(Parameter<?> parameter) {
allParameters.add(parameter);
hasUnnamedParameters |= parameter.isUnnamed();
}
private final boolean resultParameter(Configuration c, Parameter<?> parameter) {
return parameter.equals(getReturnParameter0(c)) || getOutParameters0(c).contains(parameter);
}
protected final void addInParameter(Parameter<?> parameter) {
addParameter(parameter);
inParameters.add(parameter);
inValues.put(parameter, val(null, parameter.getDataType()));
if (parameter.isDefaulted())
inValuesDefaulted.add(parameter);
else
inValuesNonDefaulted.add(parameter);
}
protected final void addInOutParameter(Parameter<?> parameter) {
addInParameter(parameter);
outParameters.add(parameter);
}
protected final void addOutParameter(Parameter<?> parameter) {
addParameter(parameter);
outParameters.add(parameter);
}
protected final void setReturnParameter(Parameter<T> parameter) {
addParameter(parameter);
returnParameter = parameter;
}
public final Field<T> asField() {
if (function == null)
function = new RoutineField();
return function;
}
public final Field<T> asField(String alias) {
return asField().as(alias);
}
public final AggregateFunction<T> asAggregateFunction() {
Field<?>[] array = new Field<?>[getInParameters().size()];
int i = 0;
for (Parameter<?> p : getInParameters()) {
array[i] = getInValues().get(p);
i++;
}
return new DefaultAggregateFunction<>(false, getQualifiedName(), type, array);
}
@Deprecated
protected static final <T> Parameter<T> createParameter(String name, DataType<T> type) {
return createParameter(name, type, false, null, null);
}
@Deprecated
protected static final <T> Parameter<T> createParameter(String name, DataType<T> type, boolean isDefaulted) {
return createParameter(name, type, isDefaulted, null, null);
}
@Deprecated
protected static final <T, U> Parameter<U> createParameter(String name, DataType<T> type, boolean isDefaulted, Converter<T, U> converter) {
return createParameter(name, type, isDefaulted, converter, null);
}
@Deprecated
protected static final <T, U> Parameter<U> createParameter(String name, DataType<T> type, boolean isDefaulted, Binding<T, U> binding) {
return createParameter(name, type, isDefaulted, null, binding);
}
@Deprecated
protected static final <T, X, U> Parameter<U> createParameter(String name, DataType<T> type, boolean isDefaulted, Converter<X, U> converter, Binding<T, X> binding) {
return createParameter(name, type, isDefaulted, false, converter, binding);
}
@Deprecated
protected static final <T> Parameter<T> createParameter(String name, DataType<T> type, boolean isDefaulted, boolean isUnnamed) {
return createParameter(name, type, isDefaulted, isUnnamed, null, null);
}
@Deprecated
protected static final <T, U> Parameter<U> createParameter(String name, DataType<T> type, boolean isDefaulted, boolean isUnnamed, Converter<T, U> converter) {
return createParameter(name, type, isDefaulted, isUnnamed, converter, null);
}
@Deprecated
protected static final <T, U> Parameter<U> createParameter(String name, DataType<T> type, boolean isDefaulted, boolean isUnnamed, Binding<T, U> binding) {
return createParameter(name, type, isDefaulted, isUnnamed, null, binding);
}
@Deprecated
@SuppressWarnings("unchecked")
protected static final <T, X, U> Parameter<U> createParameter(String name, DataType<T> type, boolean isDefaulted, boolean isUnnamed, Converter<X, U> converter, Binding<T, X> binding) {
final Binding<T, U> actualBinding = DefaultBinding.newBinding(converter, type, binding);
final DataType<U> actualType = converter == null && binding == null
? (DataType<U>) type
: type.asConvertedDataType(actualBinding);
return new ParameterImpl<>(name, actualType, actualBinding, isDefaulted, isUnnamed);
}
private class RoutineField extends AbstractField<T> {
private static final long serialVersionUID = -5730297947647252624L;
@SuppressWarnings("unchecked")
RoutineField() {
super(DSL.name(AbstractRoutine.this.getName()),
AbstractRoutine.this.type == null
? (DataType<T>) SQLDataType.RESULT
: AbstractRoutine.this.type);
}
@SuppressWarnings({ "rawtypes", "unchecked" })
@Override
public void accept(Context<?> ctx) {
SQLDialect family = ctx.family();
String name;
DataType<?> returnType;
List<Field<?>> fields = new ArrayList<>(getInParameters0(ctx.configuration()).size());
returnType = getDataType();
{
RenderContext local = create(ctx).renderContext();
toSQLQualifiedName(local);
name = local.render();
}
for (Parameter<?> parameter : getInParameters0(ctx.configuration())) {
if (inValuesDefaulted.contains(parameter))
continue;
if ( family == POSTGRES)
if (hasUnnamedParameters())
if (pgArgNeedsCasting(parameter))
fields.add(new Cast(getInValues().get(parameter), parameter.getDataType()));
else
fields.add(getInValues().get(parameter));
else
if (pgArgNeedsCasting(parameter))
fields.add(DSL.field("{0} := {1}", name(parameter.getName()), new Cast(getInValues().get(parameter), parameter.getDataType())));
else
fields.add(DSL.field("{0} := {1}", name(parameter.getName()), getInValues().get(parameter)));
else
fields.add(getInValues().get(parameter));
}
Field<?> result = function(name, returnType, fields.toArray(EMPTY_FIELD));
if (TRUE.equals(settings(ctx.configuration()).isRenderScalarSubqueriesForStoredFunctions()))
result = DSL.select(result).asField();
ctx.visit(result);
}
}
}