cn.schoolwow.quickdao.module.operation.flow.generateEntity.single.GenerateClassTablePropertyFlow Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of QuickDAO Show documentation
There is a newer version: 5.3.1
package cn.schoolwow.quickdao.module.operation.flow.generateEntity.single;

import cn.schoolwow.quickdao.annotation.IdStrategy;
import cn.schoolwow.quickdao.domain.QuickDAOConfig;
import cn.schoolwow.quickdao.domain.entity.Entity;
import cn.schoolwow.quickdao.domain.entity.Property;
import cn.schoolwow.quickdao.domain.operation.GenerateJavaSourceFileRequest;
import cn.schoolwow.quickdao.util.StringUtil;
import cn.schoolwow.quickflow.domain.FlowContext;
import cn.schoolwow.quickflow.flow.BusinessFlow;

public class GenerateClassTablePropertyFlow implements BusinessFlow {
    @Override
    public void executeBusinessFlow(FlowContext flowContext) throws Exception {
        Entity databaseEntity = (Entity) flowContext.checkData("databaseEntity");

        for(Property property:databaseEntity.properties){
            flowContext.putCurrentFlowData("property", property);
            generateComment(flowContext);
            generateIdStrategy(flowContext);
            generateColumnName(flowContext);
            generatePrivateField(flowContext);
        }
    }

    @Override
    public String name() {
        return "生成实体类的成员表变量部分";
    }

    private void generateComment(FlowContext flowContext){
        Property property = (Property) flowContext.checkData("property");
        if(null==property.comment||property.comment.isEmpty()){
            return;
        }
        StringBuilder builder = (StringBuilder) flowContext.checkData("builder");

        property.comment = property.comment
                .replaceAll("\r","\\n")
                .replaceAll("\n","\\n")
                .replaceAll("\r\n","\\n");
        builder.append("\t@Comment(\""+property.comment +"\")\n");
    }

    private void generateIdStrategy(FlowContext flowContext){
        Property property = (Property) flowContext.checkData("property");
        if(null==property.comment||property.comment.isEmpty()){
            return;
        }
        StringBuilder builder = (StringBuilder) flowContext.checkData("builder");
        if(property.id){
            builder.append("\t@Id");
            if(!IdStrategy.AutoIncrement.equals(property.strategy)){
                builder.append("(strategy = IdStrategy.None)");
            }
            builder.append("\n");
        }
    }

    private void generateColumnName(FlowContext flowContext){
        Property property = (Property) flowContext.checkData("property");
        StringBuilder builder = (StringBuilder) flowContext.checkData("builder");

        builder.append("\t@ColumnName(\""+property.column+"\")\n");
        builder.append("\t@ColumnType(\""+property.columnType+(null==property.range?"":"("+property.range+")")+"\")\n");
    }

    private void generatePrivateField(FlowContext flowContext){
        GenerateJavaSourceFileRequest generateJavaSourceFileRequest = (GenerateJavaSourceFileRequest) flowContext.checkData("generateJavaSourceFileRequest");
        Property property = (Property) flowContext.checkData("property");
        StringBuilder builder = (StringBuilder) flowContext.checkData("builder");
        QuickDAOConfig quickDAOConfig = (QuickDAOConfig) flowContext.checkData("quickDAOConfig");

        if(null!=generateJavaSourceFileRequest.columnFieldTypeMapping){
            property.className = generateJavaSourceFileRequest.columnFieldTypeMapping.apply(property.columnType);
        }
        if(null==property.className){
            property.className = quickDAOConfig.databaseContext.databaseProvider.convertColumnTypeToJavaType(property.columnType);
        }
        if(null==property.className){
            property.className = "{{"+property.columnType+"}}";
            System.out.println("无法转换的列类型!"+property.columnType);
        }
        if(null==property.name||property.name.isEmpty()){
            property.name = StringUtil.underline2Camel(property.column);
        }
        builder.append("\tprivate "+property.className+" "+property.name+";\n\n");
    }

}