sqlUtil

package com.cmbchina.monitor.utils;

import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.parser.ParserException;
import com.alibaba.druid.sql.parser.SQLParserUtils;
import com.alibaba.druid.sql.parser.SQLStatementParser;
import com.cmbchina.monitor.dao.mapper.ApiConfigMapper;
import com.cmbchina.monitor.dao.mapper.CopybookFieldMapper;
import com.cmbchina.monitor.entity.configs.ApiConfigs;
import com.cmbchina.monitor.entity.configs.CopybookFieldConfig;
import org.springframework.util.StringUtils;

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

import static com.alibaba.druid.sql.SQLUtils.toSQLString;

public class SqlCheckUtil {

    public static void main(String[] args) {
        String str = "UPDATE COXDBO.COPYBOOK_FIELD_CONFIG_A SET FILLER = 0 WHERE COPYBOOK = 'CPS102' AND COLUMN_NAME='CM_EMPTY_UACCT_FLAG';".toUpperCase();
//        sqlFormat(str);
        System.out.println(mergeContinuetyBlank(str));
        System.out.println(getUpdateTableName(str));
//        System.out.println(str.indexOf("WHERE"));
//        System.out.println(str.substring(str.indexOf("WHERE")));
        System.out.println(sqlToBack(str, null, null));
        System.out.println(getUpdateWhere(str));

    }

    /**
     * 格式化显示sql语句
     *
     * @param sql
     * @return
     */
    public static String sqlFormat(String sql) {
        if (sql == null || "".equals(sql.trim()))
            return "";
        List<SQLStatement> statementList = null;
        SQLStatementParser parser = null;
        try {
            parser = SQLParserUtils.createSQLStatementParser(sql, "mysql");
            statementList = parser.parseStatementList();
        } catch (ParserException e) {
            System.out.println("SQL转换中发生了错误：" + e.getMessage());
            throw e;
        }
        return toSQLString(statementList, "mysql");
    }

    /**
     * 检查sql语法是否正确
     *
     * @param sql
     * @return
     */
    public static Boolean checkSql(String sql) {
        if (sql == null || "".equals(sql.trim()))
            return false;
        List<SQLStatement> statementList = null;
        SQLStatementParser parser = null;
        try {
            parser = SQLParserUtils.createSQLStatementParser(sql, "oracle");
            statementList = parser.parseStatementList();
        } catch (ParserException e) {
            return false;
        }
        return true;
    }

    public static String sqlToBack(String sql, CopybookFieldMapper copybookFieldMapper, ApiConfigMapper apiConfigMapper) {
        if (StringUtils.isEmpty(sql))
            return "";
        //合并连续空格
        sql = mergeContinuetyBlank(sql).toUpperCase();
        //判断时更新还是插入操作
        if (sql.contains("INSERT INTO")) {
            //获取表名
            String tableName = getInsertTableName(sql);
            //判断时api表还是field表
            if (tableName.contains("API")) {
                //获取API_ID
                String apiId = getInsertColumnValue(sql, "API_ID");
                //生成回退语句
                String backSql = "DELETE FROM COXDBO.BILL_QUERY_API_CONFIG WHERE API_ID=" + apiId;
                //TODO 插入历史记录

                return backSql;
            } else {
                //获取插入primaryKey
                String copybook = getInsertColumnValue(sql, "COPYBOOK");
                String fieldNo = getInsertColumnValue(sql, "FIELD_NO");
                //生成回退语句
                String backSql = "DELETE FROM " + tableName + " WHERE COPYBOOK=" + copybook + " AND FIELD_NO=" + fieldNo + "";
                //TODO 插入历史记录
                return backSql;
            }

        } else {
            //获取表名
            String tableName = getUpdateTableName(sql);
            //确定回退表信息
            if (tableName.contains("API")) {
                //获取where过滤条件
                Map<String, Object> whereMap = getUpdateWhere(sql);
                //获取更新前的值
                List<ApiConfigs> api_id = apiConfigMapper.query(1, 2, whereMap.get("API_ID").toString());
                ApiConfigs apiConfigs = null;
                if (api_id != null)
                    apiConfigs = api_id.get(0);
                //生成回退语句
                String backSql = getApiInsertBack(apiConfigs);
                //TODO 存入配置历史

            } else {
                //获取where过滤条件
                Map<String, Object> whereMap = getUpdateWhere(sql);
                CopybookFieldConfig copybookFieldConfig = new CopybookFieldConfig();
                copybookFieldConfig.setCopybook(whereMap.get("COPYBOOK") == null ? null : whereMap.get("COPYBOOK").toString());
                copybookFieldConfig.setCobolName(whereMap.get("COLUMN_NAME") == null ? null : whereMap.get("COLUMN_NAME").toString());
                copybookFieldConfig.setFieldNo(whereMap.get("FIELD_NO") == null ? null : Long.parseLong(whereMap.get("FIELD_NO").toString()));
                //获取更新前的值
                List<CopybookFieldConfig> copybookFieldConfigs = copybookFieldMapper.copybookFieldConfigQuery(copybookFieldConfig);
                CopybookFieldConfig copybookFieldConfig1 = copybookFieldConfigs.get(0);
                //拼接回退语句--只会更新filler字段
                StringBuilder backSql = new StringBuilder("UPDATE "+ tableName +" ");
                if (copybookFieldConfig1.getFiller() == null) {
                    backSql.append(" SET FILLER= NULL");
                }else {
                    backSql.append(" SET FILLER= '" + copybookFieldConfig1.getFiller() + "'");
                }
                backSql.append(" WHERE COPYBOOK='"+ copybookFieldConfig1.getCopybook() +"' ");
                backSql.append(" AND COLUMN_NAME='"+ copybookFieldConfig1.getColumnName() +"' ");
                backSql.append(" AND FIELD_NO="+ copybookFieldConfig1.getFieldNo() +" ");
                //TODO 存入配置历史

            }

        }

        return null;
    }

    private static String getApiInsertBack(ApiConfigs apiConfigs) {
        StringBuilder backSql = new StringBuilder("UPDATE COXDBO.QUERY_API_CONFIG " +
                "SET STATUS='1', COX_TPS=1000, CTG_TPS=50, \"TYPE\"='1', URL='/accountinfo/CORE4A06', PERCENTAGE=0, SWITCH_TAIL=NULL " +
                "WHERE API_ID='CORE4A06'");
        if (apiConfigs.getStatus() != null)
            backSql.append("STATUS='" + apiConfigs.getStatus() + "', ");
        if (apiConfigs.getType() != null)
            backSql.append("TYPE='" + apiConfigs.getType() + "', ");
        if (apiConfigs.getUrl() != null)
            backSql.append("URL='" + apiConfigs.getUrl() + "', ");
        if (apiConfigs.getSwitchTail() != null)
            backSql.append("SWITCH_TAIL='" + apiConfigs.getSwitchTail() + "', ");

        if (apiConfigs.getCoxTps() != null)
            backSql.append("COX_TPS=" + apiConfigs.getCoxTps() + ", ");
        if (apiConfigs.getCoxTps() != null)
            backSql.append("CTG_TPS=" + apiConfigs.getCtgTps() + ", ");
        if (apiConfigs.getCoxTps() != null)
            backSql.append("PERCENTAGE=" + apiConfigs.getPercentage() + ", ");
        backSql = new StringBuilder(backSql.substring(0, backSql.length() - 2));
        backSql.append(" WHERE API_ID='" + apiConfigs + "'");
        return backSql.toString();
    }

    private static Map<String, Object> getUpdateWhere(String sql) {
        if (StringUtils.isEmpty(sql))
            return null;
        sql = sql.replace(";", "").toUpperCase();

        String where = sql.substring(sql.indexOf("WHERE") + 5);
        String[] ands = where.split("AND");
        Map<String, Object> conditionMap = new HashMap<>();
        for (String and : ands) {
            and = and.replaceAll(" ", "");
            String[] split = and.split("=");
            conditionMap.put(split[0], split[1]);
        }
        return conditionMap;
    }

    /**
     * 将字符串中多个连续的空格合并为一个
     *
     * @param str
     * @return
     */
    public static String mergeContinuetyBlank(String str) {
        if (StringUtils.isEmpty(str))
            return "";
        String s = str.replaceAll("\n", " ").replaceAll(" {2,}", " ").replaceAll("\r", "").replaceAll("\t", "").replaceAll(";", "");
        //若第一位为空格，则将第一位移除
        if (s.charAt(0) == ' ') {
            s = s.substring(1, s.length());
        }
        return s;
    }

    /**
     * 获取插入操作表名
     *
     * @param sql
     * @return
     */
    public static String getInsertTableName(String sql) {
        if (StringUtils.isEmpty(sql))
            return "";
        String[] s = sql.split(" ");
        return s[2];
    }

    /**
     * 获取更新操作表名
     *
     * @param sql
     * @return
     */
    public static String getUpdateTableName(String sql) {
        if (StringUtils.isEmpty(sql))
            return "";
        String[] s = sql.split(" ");
        return s[1];
    }

    /**
     * 获取Api表主键值
     *
     * @param sql
     * @return
     */
    public static String getInsertColumnValue(String sql, String columnName) {
        sql = sql.toUpperCase();
        String firstBraket = sql.substring(sql.indexOf('(') + 1, sql.indexOf(')'));
        String firstBraketSql = firstBraket.replaceAll(" ", "");
        String subSql = sql.substring(sql.indexOf(')') + 1);
        String secondBraket = subSql.substring(subSql.indexOf('(') + 1, subSql.indexOf(')'));
        String secondBraketSql = secondBraket.replaceAll(" ", "");
        String[] nameSplit = firstBraketSql.split(",");
        String[] valueSplit = secondBraketSql.split(",");
        for (int i = 0; i < nameSplit.length; i++) {
            if (columnName.equals(nameSplit[i]))
                return valueSplit[i];
        }
        return "";
    }

}