/**
* Copyright 2006-2016 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.mybatis.generator.plugins;
import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.JavaVisibility;
import org.mybatis.generator.api.dom.java.Method;
import org.mybatis.generator.api.dom.java.TopLevelClass;
import java.util.List;
/**
* This plugin adds fluent builder methods to the generated model
* classes.
*
* Example:
* <p>
* Given the domain class <code>MyDomainClass</code> with setter-method <code>setValue(Object v)</code>
* <p>
* The plugin will create the additional Method <code>public MyDomainClass withValue(Object v)</code>
*
*
* @author Stefan Lack
*/
public class FluentBuilderMethodsPlugin extends PluginAdapter {
public boolean validate(List<String> warnings) {
return true;
}
@Override
public boolean modelSetterMethodGenerated(Method method,
TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn,
IntrospectedTable introspectedTable,
ModelClassType modelClassType) {
Method fluentMethod = new Method();
fluentMethod.setVisibility(JavaVisibility.PUBLIC);
fluentMethod.setReturnType(topLevelClass.getType());
fluentMethod.setName("with" + method.getName().substring(3)); //$NON-NLS-1$
fluentMethod.getParameters().addAll(method.getParameters());
context.getCommentGenerator().addGeneralMethodComment(fluentMethod,
introspectedTable);
StringBuilder sb = new StringBuilder()
.append("this.") //$NON-NLS-1$
.append(method.getName())
.append('(')
.append(introspectedColumn.getJavaProperty())
.append(");"); //$NON-NLS-1$
fluentMethod.addBodyLine(sb.toString()); //$NON-NLS-1$
fluentMethod.addBodyLine("return this;"); //$NON-NLS-1$
topLevelClass.addMethod(fluentMethod);
return super.modelSetterMethodGenerated(method, topLevelClass, introspectedColumn, introspectedTable, modelClassType);
}
}