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;
        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++);
        }
        String strSelect = String.join(" , ", lsSelect);
        String strWhere = String.join(" AND ", lsWhere);
        StringBuilder sbSql = new StringBuilder();
        sbSql.append("SELECT ").append(strSelect).append(" FROM ").append(tableMapping.getTableName()).append(" WHERE ").append(strWhere);
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;
    }

    public BuildResult buildSelectCount() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        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++);
        }
        String strWhere = String.join(" AND ", lsWhere);
        StringBuilder sbSql = new StringBuilder();
        sbSql.append("SELECT ").append("count(1)").append(" FROM ").append(tableMapping.getTableName()).append(" WHERE ").append(strWhere);
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;
    }

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

    public BuildResult buildUpdate() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        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++);
        }
        String strUpdateSet = String.join(" , ", lsUpdateSet);
        String strWhere = String.join(" AND ", lsWhere);
        StringBuilder sbSql = new StringBuilder();
        sbSql.append("UPDATE ").append(tableMapping.getTableName()).append(" SET ").append(strUpdateSet).append(" WHERE ").append(strWhere);
        result.setParametersIndex(parameterOrder);
        result.setSql(sbSql.toString());
        return result;
    }

    public BuildResult buildDelete() {
        BuildResult result = new BuildResult();
        Map parameterOrder = new HashMap();
        int order = 1;
        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++);
        }
        String strWhere = String.join(" AND ", lsWhere);
        StringBuilder sbSql = new StringBuilder();
        sbSql.append("DELETE FROM ").append(tableMapping.getTableName()).append(" WHERE ").append(strWhere);
        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;
    }
}