TruncateTablePlugin.java

package com.revengemission.plugins.mybatis;


import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
import org.mybatis.generator.api.dom.java.Interface;
import org.mybatis.generator.api.dom.java.Method;
import org.mybatis.generator.api.dom.xml.Attribute;
import org.mybatis.generator.api.dom.xml.Document;
import org.mybatis.generator.api.dom.xml.TextElement;
import org.mybatis.generator.api.dom.xml.XmlElement;

import java.util.List;

/**
 * Mysql Truncate table plugin
 */
public class TruncateTablePlugin extends AbstractXmbgPlugin {

    private static final String CLIENT_METHOD_NAME = "truncateTable";

    @Override
    public void initialized(IntrospectedTable introspectedTable) {
    }

    @Override
    public boolean validate(List<String> warnings) {
        return true;
    }

    @Override
    public boolean clientGenerated(Interface interfaze, IntrospectedTable introspectedTable) {

        Method method = new Method(CLIENT_METHOD_NAME);
        method.setAbstract(true);
        method.setReturnType(FullyQualifiedJavaType.getIntInstance());
        interfaze.addMethod(method);
        return true;
    }


    @Override
    public boolean sqlMapDocumentGenerated(Document document, IntrospectedTable introspectedTable) {

        String tableName = getTableName(introspectedTable);

        XmlElement selectElement = new XmlElement("update");
        selectElement.addAttribute(new Attribute("id", CLIENT_METHOD_NAME));
        String tempString = "TRUNCATE TABLE " + tableName;
        selectElement.addElement(
            new TextElement(tempString
            ));
        document.getRootElement().addElement(selectElement);
        return true;
    }


}