package org.jooq.meta.mysql;
import static java.util.Arrays.asList;
import static org.jooq.meta.mysql.information_schema.Tables.PARAMETERS;
import java.util.regex.Matcher;
import org.jooq.Record;
import org.jooq.meta.AbstractRoutineDefinition;
import org.jooq.meta.DataTypeDefinition;
import org.jooq.meta.DefaultDataTypeDefinition;
import org.jooq.meta.DefaultParameterDefinition;
import org.jooq.meta.InOutDefinition;
import org.jooq.meta.ParameterDefinition;
import org.jooq.meta.SchemaDefinition;
import org.jooq.meta.mysql.mysql.enums.ProcType;
import org.jooq.tools.StringUtils;
public class MySQLRoutineDefinition extends AbstractRoutineDefinition {
private static Boolean is55;
private final String params;
private final String returns;
private final ProcType procType;
@Deprecated
public MySQLRoutineDefinition(SchemaDefinition schema, String name, String comment, String params, String returns) {
this(schema, name, comment, params, returns, null, null);
}
public MySQLRoutineDefinition(SchemaDefinition schema, String name, String comment, String params, String returns, ProcType procType, String overload) {
super(schema, null, name, comment, overload);
this.params = params;
this.returns = returns;
this.procType = procType;
}
@Override
protected void init0() {
if (is55())
init55();
else
init54();
}
private void init55() {
for (Record record : create()
.select(
PARAMETERS.ORDINAL_POSITION,
PARAMETERS.PARAMETER_NAME,
PARAMETERS.PARAMETER_MODE,
PARAMETERS.DATA_TYPE,
PARAMETERS.DTD_IDENTIFIER,
PARAMETERS.CHARACTER_MAXIMUM_LENGTH,
PARAMETERS.NUMERIC_PRECISION,
PARAMETERS.NUMERIC_SCALE
)
.from(PARAMETERS)
.where(PARAMETERS.SPECIFIC_SCHEMA.in(getSchema().getInputName(), getSchema().getInputName()))
.and(PARAMETERS.SPECIFIC_NAME.eq(getInputName()))
.and(PARAMETERS.ROUTINE_TYPE.eq(procType.name()))
.orderBy(PARAMETERS.ORDINAL_POSITION.asc())
.fetch()) {
String inOut = record.get(PARAMETERS.PARAMETER_MODE);
String dataType = record.get(PARAMETERS.DATA_TYPE);
if (getDatabase().supportsUnsignedTypes()) {
if (asList("tinyint", "smallint", "mediumint", "int", "bigint").contains(dataType.toLowerCase())) {
if (record.get(PARAMETERS.DTD_IDENTIFIER).toLowerCase().contains("unsigned")) {
dataType += "unsigned";
}
}
}
DataTypeDefinition type = new DefaultDataTypeDefinition(
getDatabase(),
getSchema(),
dataType,
record.get(PARAMETERS.CHARACTER_MAXIMUM_LENGTH),
record.get(PARAMETERS.NUMERIC_PRECISION),
record.get(PARAMETERS.NUMERIC_SCALE),
null,
(String) null
);
if (inOut == null) {
addParameter(InOutDefinition.RETURN, new DefaultParameterDefinition(this, "RETURN_VALUE", -1, type));
}
else {
ParameterDefinition parameter = new DefaultParameterDefinition(
this,
record.get(PARAMETERS.PARAMETER_NAME).replaceAll("@", ""),
record.get(PARAMETERS.ORDINAL_POSITION, int.class),
type);
addParameter(InOutDefinition.getFromString(inOut), parameter);
}
}
}
private void init54() {
String[] split = params.split(",(?!\\s*\\d+\\s*\\))");
Matcher matcher = TYPE_PATTERN.matcher(returns);
if (matcher.find()) {
addParameter(InOutDefinition.RETURN, createParameter(matcher, 0, -1, "RETURN_VALUE"));
}
for (int i = 0; i < split.length; i++) {
String param = split[i];
param = param.trim();
matcher = PARAMETER_PATTERN.matcher(param);
while (matcher.find()) {
InOutDefinition inOut = InOutDefinition.getFromString(matcher.group(2));
addParameter(inOut, createParameter(matcher, 3, i + 1));
}
}
}
private ParameterDefinition createParameter(Matcher matcher, int group, int columnIndex) {
return createParameter(matcher, group, columnIndex, matcher.group(group));
}
private ParameterDefinition createParameter(Matcher matcher, int group, int columnIndex, String paramName) {
String paramType = matcher.group(group + 1);
Number precision = 0;
Number scale = 0;
if (!StringUtils.isBlank(matcher.group(group + 2))) {
precision = Integer.valueOf(matcher.group(group + 2));
}
if (!StringUtils.isBlank(matcher.group(group + 3))) {
scale = Integer.valueOf(matcher.group(group + 3));
}
DataTypeDefinition type = new DefaultDataTypeDefinition(
getDatabase(),
getSchema(),
paramType,
precision,
precision,
scale,
null,
(String) null
);
return new DefaultParameterDefinition(this, paramName, columnIndex, type);
}
private boolean is55() {
if (is55 == null) {
try {
create().selectOne().from(PARAMETERS).limit(1).fetchOne();
is55 = true;
}
catch (Exception e) {
is55 = false;
}
}
return is55;
}
}