package org.jooq.meta;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Pattern;
import org.jooq.tools.JooqLogger;
public abstract class AbstractRoutineDefinition extends AbstractDefinition implements RoutineDefinition {
private static final JooqLogger log = JooqLogger.getLogger(AbstractRoutineDefinition.class);
private static final String INOUT = "(?i:(IN|OUT|INOUT)\\s+?)?";
private static final String PARAM_NAME = "(?:(\\S+?)\\s+?)";
private static final String PARAM_TYPE = "([^\\s\\(]+)(?:\\s*\\((\\d+)(?:\\s*,\\s*(\\d+))?\\))?";
private static final String PARAMETER = "(" + INOUT + PARAM_NAME + PARAM_TYPE + ")";
protected static final Pattern PARAMETER_PATTERN = Pattern.compile(PARAMETER);
protected static final Pattern TYPE_PATTERN = Pattern.compile(PARAM_TYPE);
protected List<ParameterDefinition> inParameters;
protected List<ParameterDefinition> outParameters;
protected ParameterDefinition returnValue;
protected List<ParameterDefinition> allParameters;
private final boolean aggregate;
public AbstractRoutineDefinition(SchemaDefinition schema, PackageDefinition pkg, String name, String comment, String overload) {
this(schema, pkg, name, comment, overload, false);
}
public AbstractRoutineDefinition(SchemaDefinition schema, PackageDefinition pkg, String name, String comment, String overload, boolean aggregate) {
super(schema.getDatabase(), schema, pkg, name, comment, overload);
this.aggregate = aggregate;
}
protected void init() {
inParameters = new ArrayList<>();
outParameters = new ArrayList<>();
allParameters = new ArrayList<>();
try {
if (returnValue != null)
addParameter(InOutDefinition.RETURN, returnValue);
init0();
}
catch (Exception e) {
log.error("Error while initialising routine", e);
}
}
protected abstract void init0() throws SQLException;
@Override
public final List<ParameterDefinition> getInParameters() {
if (inParameters == null)
init();
return inParameters;
}
@Override
public final List<ParameterDefinition> getOutParameters() {
if (outParameters == null)
init();
return outParameters;
}
@Override
public final List<ParameterDefinition> getAllParameters() {
if (allParameters == null)
init();
return allParameters;
}
@Override
public final ParameterDefinition getReturnValue() {
if (allParameters == null) {
init();
}
return returnValue;
}
@Override
public final DataTypeDefinition getReturnType() {
if (getReturnValue() != null)
return getReturnValue().getType();
else
return new DefaultDataTypeDefinition(getDatabase(), getSchema(), "unknown");
}
@Override
public final DataTypeDefinition getReturnType(JavaTypeResolver resolver) {
if (getReturnValue() != null)
return getReturnValue().getType(resolver);
else
return new DefaultDataTypeDefinition(getDatabase(), getSchema(), "unknown");
}
@Override
public boolean isSQLUsable() {
return getReturnValue() != null && getOutParameters().isEmpty();
}
@Override
public final boolean isAggregate() {
return aggregate;
}
protected final void addParameter(InOutDefinition inOut, ParameterDefinition parameter) {
allParameters.add(parameter);
switch (inOut) {
case IN:
inParameters.add(parameter);
break;
case OUT:
outParameters.add(parameter);
break;
case INOUT:
inParameters.add(parameter);
outParameters.add(parameter);
break;
case RETURN:
returnValue = parameter;
break;
}
}
}