MyBatis自定义插件:DO模型添加变量

2016-09-02

自定义插件DO添加变量

在文章MyBatisGenerate自定义插件我们讲到了简单的修改DOMapper的Java和XML函数，并没有给出如何为DO模型添加变量

重写接口

重写接口 modelBaseRecordClassGenerated 可以对DO模型进行操作

@Override
public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
    addTargetField(topLevelClass,"columnCount",Integer.class);
    addTargetField(topLevelClass,"key",String.class);
    return super.modelBaseRecordClassGenerated(topLevelClass, introspectedTable);
}

其中 addTargetField 如下

private void addTargetField(TopLevelClass topLevelClass, String name, Class clazz) {
    FullyQualifiedJavaType fieldType = new FullyQualifiedJavaType(clazz.getName());
    Field field = new Field(name,fieldType);
    field.setVisibility(JavaVisibility.PRIVATE);
    topLevelClass.addField(field);
    topLevelClass.addImportedType(field.getType());
    //添加Getter Setter方法
    topLevelClass.addMethod(getJavaBeansGetter(name,fieldType));
    topLevelClass.addMethod(getJavaBeansSetter(name,fieldType));
}

拼接Getter和Setter方法

由于 topLevelClass.addField() 仅仅添加了Filed变量本身，并不会添加Getter和Setter方法，所以说Getter和Setter还要自己写，(╯°Д°）╯︵ /(.□ . \) 这尼玛太坑了，通过读MyBatisGenerate源码我大致Copy了一份，用到了MyBatisGenerate中的JavaBeansUtil

/**
 * 添加Getter方法
 * @param name 参数名
 * @param type 参数类型
 */
public Method getJavaBeansGetter(String name, FullyQualifiedJavaType type) {

    Method method = new Method();
    method.setVisibility(JavaVisibility.PUBLIC);
    method.setReturnType(type);
    method.setName(JavaBeansUtil.getGetterMethodName(name, type));

    StringBuilder sb = new StringBuilder();
    sb.append("return " + name + ";");
    method.addBodyLine(sb.toString());

    return method;
}

/**
 * 添加Setter方法
 * @param name 参数名
 * @param type 参数类型
 */
public Method getJavaBeansSetter(String name, FullyQualifiedJavaType type) {
    Method method = new Method();
    method.setVisibility(JavaVisibility.PUBLIC);
    method.setName(JavaBeansUtil.getSetterMethodName(name));
    method.addParameter(new Parameter(type, name));

    StringBuilder sb = new StringBuilder();
    sb.append("this." + name + " = " + name + ";");
    method.addBodyLine(sb.toString());
    return method;
}

在完成了DO变量的添加之后，一定记得也要在ResultMap中设定好从SQL的数据库表单到模型的映射，不然获取不到值的，参考文章BaseResult添加映射