com.lq.cli.mybatis.CreateMapperTask Maven / Gradle / Ivy

Go to download
package com.lq.cli.mybatis;


import com.lq.cli.comment.CreateTask;
import com.lq.cli.comment.StringUtil;
import com.lq.cli.comment.TaskArgs;
import com.lq.cli.jdbc.TableInfo;

import java.util.List;

public class CreateMapperTask extends CreateTask {

    private List tableInfos;

    public CreateMapperTask(TaskArgs taskArgs, List tableInfos) {
        super(taskArgs);
        this.tableInfos = tableInfos;
        setPackageName("mapper");
    }

    @Override
    public Boolean call() throws Exception {
        if (checkDir()) {
            for (TableInfo tableInfo : tableInfos) {
                String mapper = tableInfo2Mapper(tableInfo);
                createFile(tableInfo.getTransformTableInfo().getTableName()+"Mapper.java",mapper);
            }
        }
        return true;
    }

    private String tableInfo2Mapper(TableInfo tableInfo) {
        TableInfo transformTableInfo = tableInfo.getTransformTableInfo();
        StringBuilder sb = new StringBuilder();
        sb.append("package ").append(taskArgs.packageName).append(".").append(getPackageName()).append(";\n\n")
                .append("import org.apache.ibatis.annotations.Param;\n")
                .append("import ").append(taskArgs.packageName).append(".entity.")
                .append(transformTableInfo.getTableName()).append(";\n\n")
                .append("public interface ").append(transformTableInfo.getTableName()).append("Mapper {\n\n");
        sb.append("\tboolean insert")
                .append(transformTableInfo.getTableName())
                .append("(")
                .append(transformTableInfo.getTableName())
                .append(" ")
                .append(StringUtil.firstToLowerCase(transformTableInfo.getTableName()))
                .append(");\n\n");
        transformTableInfo.getFiledEntities().stream()
                .filter(tableFiledEntity -> tableFiledEntity.getKey().equals("PRI"))
                .findFirst()
                .ifPresent(priKey -> sb.append("\t")
                        .append(transformTableInfo.getTableName())
                        .append(" query")
                        .append(transformTableInfo.getTableName())
                        .append("By")
                        .append(StringUtil.firstToUpperCase(priKey.getName()))
                        .append("(@Param(\"")
                        .append(priKey.getName())
                        .append("\") ")
                        .append(priKey.getType())
                        .append(" ")
                        .append(priKey.getName())
                        .append(");\n\n")
                        .append("\tint delete")
                        .append(transformTableInfo.getTableName())
                        .append("By")
                        .append(StringUtil.firstToUpperCase(priKey.getName()))
                        .append("(@Param(\"")
                        .append(priKey.getName())
                        .append("\") ")
                        .append(priKey.getType())
                        .append(" ")
                        .append(priKey.getName())
                        .append(");\n\n")
                        .append("\tboolean update")
                        .append(transformTableInfo.getTableName())
                        .append("(")
                        .append(transformTableInfo.getTableName())
                        .append(" ")
                        .append(StringUtil.firstToLowerCase(transformTableInfo.getTableName()))
                        .append(");\n\n")
                );
        sb.append("\n}");
        return sb.toString();
    }

}