private void addUserDefinedMethods(TopLevelClass exampleClass, Interface mapperClass, IntrospectedTable introspectedTable, MyBatisClasses cls) {
for (Method action : mapperClass.getMethods()) {
if (!userDefinedMethods.matcher(action.getName()).matches()) continue;
StringBuilder args = new StringBuilder();
List<Parameter> params = new ArrayList<Parameter>();
boolean example = false;
if (action.getParameters() != null)
for (Parameter param : action.getParameters()) {
String name;
if (Objects.equals(param.getType(), exampleClass.getType())) {
example = true;
name = "this";
} else {
name = param.getName();
params.add(new Parameter(param.getType(), name));
}
if (args.length() > 0)
args.append(", ");
args.append(name);
}
if (!example) {
//System.err.println("Invalid user-defined mapper method: "+action.getName());
continue;
}
exampleClass.addMethod(method(
PUBLIC, INT, action.getName(), _(sqlSession, "sql"), params.toArray(new Parameter[params.size()]), __(
"return sql.getMapper(" + cls.names.mapper + ".class)."+action.getName()+"("+args+");"
)));
exampleClass.addMethod(method(
PUBLIC, INT, action.getName(), _(cls.types.mapper, "mapper"), params.toArray(new Parameter[params.size()]), __(
"return mapper."+action.getName()+"("+args+");"
)));
}
}
private void addCriteriaMethods(TopLevelClass topLevelClass, int newMethodsStart) {
if (!generateCriteriaMethods) return;
InnerClass criteria = null;
for (InnerClass c : topLevelClass.getInnerClasses()) {
if (c.getType().getShortName().equals("Criteria")) criteria = c;
}
if (criteria == null) return;
boolean owner = false;
for (Field f : criteria.getFields()) if (ExampleMethodsChainPlugin.OWNER.equals(f.getName())) owner = true;
if (!owner) return;
for (ListIterator<Method> methods = topLevelClass.getMethods().listIterator(newMethodsStart); methods.hasNext(); ) {
Method base = methods.next();
if (base.getVisibility() != PUBLIC || base.isStatic() || base.isConstructor()) continue;
Method m = method(PUBLIC, base.getReturnType(), base.getName());
StringBuilder sb = new StringBuilder();
sb.append("return ").append(ExampleMethodsChainPlugin.OWNER).append(".").append(base.getName()).append("(");
for (ListIterator<Parameter> params = base.getParameters().listIterator(); params.hasNext(); ) {
if (params.hasPrevious()) sb.append(", ");
Parameter p = params.next();
m.addParameter(new Parameter(p.getType(), p.getName()));
sb.append(p.getName());
}
sb.append(");");
m.addBodyLine(sb.toString());
criteria.addMethod(m);
}
}