com.zhuang.data.orm.sql.BaseSqlBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of my-data Show documentation
my-data
There is a newer version: 1.1.2
package com.zhuang.data.orm.sql;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.zhuang.data.exception.OrmException;
import com.zhuang.data.orm.enums.PlaceHolderType;
import com.zhuang.data.orm.mapping.ColumnMapping;
import com.zhuang.data.orm.mapping.TableMapping;
import com.zhuang.data.util.StringUtils;

public abstract class BaseSqlBuilder implements SqlBuilder {

    protected TableMapping tableMapping;
    protected PlaceHolderType placeHolderType;

    public BaseSqlBuilder(TableMapping tableMapping, PlaceHolderType placeHolderType) {
        this.tableMapping = tableMapping;
        this.placeHolderType = placeHolderType;
    }

    public BuildResult buildSelect() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        StringBuilder sbSql = new StringBuilder();
        List columns = tableMapping.getColumns();
        List keyColumns = tableMapping.getKeyColumns();
        List lsSelect = new ArrayList();
        List lsWhere = new ArrayList();
        if (keyColumns.size() < 1)
            throw new OrmException("实体没有设置主键！");

        for (ColumnMapping col : columns) {
            lsSelect.add(resolveColumnName(col.getColumnName()) + " as " + col.getPropertyName());
        }
        for (ColumnMapping keyCol : keyColumns) {
            lsWhere.add(resolveColumnName(keyCol.getColumnName()) + "=" + getPlaceHolder(placeHolderType, keyCol.getColumnName()));
            parameterOrder.put(keyCol.getPropertyName(), order++);
        }
        sbSql.append("SELECT " + String.join(" , ", lsSelect) + " FROM " + tableMapping.getTableName() + " WHERE "
                + String.join(" AND ", lsWhere));
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;

    }

    public BuildResult buildInsert() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        StringBuilder sbSql = new StringBuilder();
        List insertColumns = tableMapping.getInsertColumns();
        List columnNameList = new ArrayList();
        List lsParameterNames = new ArrayList();
        if (insertColumns.size() < 1)
            throw new OrmException("实体没有对应要插入到数据的属性！");
        for (ColumnMapping col : insertColumns) {
            columnNameList.add(resolveColumnName(col.getColumnName()));
            lsParameterNames.add(getPlaceHolder(placeHolderType, col.getColumnName()));
            parameterOrder.put(col.getPropertyName(), order++);
        }
        sbSql.append("INSERT INTO " + tableMapping.getTableName() + "(" + String.join(",", columnNameList) + ") VALUES("
                + String.join(",", lsParameterNames) + ")");
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;
    }

    public BuildResult buildUpdate() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        StringBuilder sbSql = new StringBuilder();
        List updateColumns = tableMapping.getUpdateColumns();
        List keyColumns = tableMapping.getKeyColumns();
        List lsUpdateSet = new ArrayList();
        List lsWhere = new ArrayList();
        if (keyColumns.size() < 1)
            throw new OrmException("实体没有设置主键！");
        if (updateColumns.size() < 1)
            throw new OrmException("实体没有对应要更新到数据的属性！");
        for (ColumnMapping col : updateColumns) {
            lsUpdateSet.add(resolveColumnName(col.getColumnName()) + "=" + getPlaceHolder(placeHolderType, col.getColumnName()));
            parameterOrder.put(col.getPropertyName(), order++);
        }
        for (ColumnMapping keyCol : keyColumns) {
            lsWhere.add(resolveColumnName(keyCol.getColumnName()) + "=" + getPlaceHolder(placeHolderType, keyCol.getColumnName()));
            parameterOrder.put(keyCol.getPropertyName(), order++);
        }
        sbSql.append("UPDATE " + tableMapping.getTableName() + " SET "
                + String.join(" , ", lsUpdateSet) + "  WHERE " + String.join(" AND ", lsWhere));
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;
    }

    public BuildResult buildDelete() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        StringBuilder sbSql = new StringBuilder();
        List keyColumns = tableMapping.getKeyColumns();
        List lsWhere = new ArrayList();
        if (keyColumns.size() < 1)
            throw new OrmException("实体没有设置主键！");
        for (ColumnMapping keyCol : keyColumns) {
            lsWhere.add(resolveColumnName(keyCol.getColumnName()) + "=" + getPlaceHolder(placeHolderType, keyCol.getColumnName()));
            parameterOrder.put(keyCol.getPropertyName(), order++);
        }
        sbSql.append("DELETE FROM ").append(tableMapping.getTableName()).append(" WHERE ")
                .append(String.join(" AND ", lsWhere));
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;
    }

    protected String getPlaceHolder(PlaceHolderType placeHolderType, String columnName) {
        String result = "";
        if (placeHolderType == PlaceHolderType.QuestionMark) {
            result = "?";
        } else if (placeHolderType == PlaceHolderType.NumberSign) {
            result = "#{" + columnName + "}";
        } else if (placeHolderType == PlaceHolderType.DollarSign) {
            result = "${" + columnName + "}";
        }
        return result;
    }

    private String resolveColumnName(String columnName) {
        String result;
        if (tableMapping.isUnderscoreNaming()) {
            result = StringUtils.camelCaseToUnderscore(columnName);
        } else {
            result = columnName;
        }
        return result;
    }
}