com.fastchar.database.FastSql Maven / Gradle / Ivy

package com.fastchar.database;

import com.fastchar.core.FastChar;
import com.fastchar.core.FastEntity;
import com.fastchar.database.info.FastColumnInfo;
import com.fastchar.database.info.FastSqlInfo;
import com.fastchar.exception.FastSqlException;

import com.fastchar.utils.FastArrayUtils;
import com.fastchar.utils.FastStringUtils;

import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class FastSql {
    private String type;//数据库类型

    public static FastSql newInstance(String type) {
        FastSql fastSql = FastChar.getOverrides().newInstance(FastSql.class);
        return (FastSql) fastSql.setType(type.toLowerCase());
    }

    protected FastSqlInfo newSqlInfo() {
        return new FastSqlInfo().setType(this.type);
    }


    public String getType() {
        return type;
    }

    public T setType(String type) {
        this.type = type;
        return (T) this;
    }

    /**
     * 获得统计的sql语句
     */
    public String getCountSql(String sqlStr, String alias) {
        int[] orderIndex = getTokenIndex("order by", sqlStr);
        if (orderIndex[0] > 0) {
            String orderSql = sqlStr.substring(orderIndex[0], orderIndex[1] + orderIndex[0]);
            sqlStr = sqlStr.replace(orderSql, "");
        }

        int startIndex = sqlStr.toLowerCase().indexOf("select") + 6;
        int endIndex = sqlStr.toLowerCase().indexOf("from");
        if (endIndex > 0) {
            String select = sqlStr.substring(startIndex, endIndex);
            return sqlStr.replace(select, " count(1) as " + alias + " ");
        }
        throw new FastSqlException(FastChar.getLocal().getInfo("Db_Sql_Error2"));
    }

    /**
     * 获得分页的sql
     */
    public String getPageSql(String sqlStr, int page, int pageSize) {
        if (page > 0) {
            if (type.toLowerCase().equals("mysql")) {
                sqlStr += " limit " + (page - 1) * pageSize + "," + pageSize;
            } else if (type.toLowerCase().equals("sql_server")) {//后续跟进

            } else if (type.toLowerCase().equals("oracle")) {//后续跟进

            }
        }
        return sqlStr;
    }

    protected Object getColumnValue(FastEntity entity, FastColumnInfo columnInfo) {
        Object value = entity.get(columnInfo.getName());
        if (value == null) {
            return null;
        }
        if (value.toString().equalsIgnoreCase("")) {
            return null;
        }
        if (FastStringUtils.isNotEmpty(columnInfo.getEncrypt())) {
            if (columnInfo.getEncrypt().equalsIgnoreCase("md5")) {
                value = FastChar.getSecurity().MD5_Encrypt(String.valueOf(value));
            } else if (columnInfo.getEncrypt().equalsIgnoreCase("true")) {
                value = FastChar.getSecurity().AES_Encrypt(String.valueOf(value));
            }
        }
        return value;
    }

    public FastSqlInfo toSelectSql(String sqlStr, FastEntity entity) {
        return convertSqlName(sqlStr, entity);
    }

    public FastSqlInfo toSelectSql(FastEntity entity) {
        return toSelectSql("select * from " + entity.getTableName(), entity);
    }

    public FastSqlInfo toSelectSql(FastEntity entity, Object... ids) {
        if (ids.length == 0) {
            return null;
        }
        StringBuilder sqlStr = new StringBuilder("select * from " + entity.getTableName() + " where 1=1 ");
        int count = 0;
        for (FastColumnInfo primary : entity.getPrimaries()) {
            sqlStr.append(" and ").append(primary.getName()).append(" = ? ");
            count++;
            if (count >= ids.length) {
                break;
            }
        }

        FastSqlInfo sqlInfo = newSqlInfo();
        sqlInfo.setSql(sqlStr.toString());
        sqlInfo.setLog(entity.getBoolean("log", true));
        sqlInfo.setParams(Arrays.asList(ids));
        return sqlInfo;
    }

    public FastSqlInfo toInsertSql(FastEntity entity) {
        if (entity == null) {
            return null;
        }
        List columns = new ArrayList<>();
        List placeholders = new ArrayList<>();
        List