| New file |
| | |
|---|
| | | package com.hx.mybatis.aes.springbean; |
|---|
| | | |
|---|
| | | import com.alibaba.druid.sql.SQLUtils; |
|---|
| | | import com.alibaba.druid.sql.ast.SQLExpr; |
|---|
| | | import com.alibaba.druid.sql.ast.SQLObject; |
|---|
| | | import com.alibaba.druid.sql.ast.SQLStatement; |
|---|
| | | import com.alibaba.druid.sql.ast.statement.*; |
|---|
| | | import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlDeleteStatement; |
|---|
| | | import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlInsertStatement; |
|---|
| | | import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlUpdateStatement; |
|---|
| | | import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser; |
|---|
| | | import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor; |
|---|
| | | import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor; |
|---|
| | | import com.alibaba.druid.stat.TableStat; |
|---|
| | | import com.alibaba.druid.util.JdbcConstants; |
|---|
| | | import com.alibaba.druid.util.JdbcUtils; |
|---|
| | | import com.hx.util.StringUtils; |
|---|
| | | |
|---|
| | | import java.util.ArrayList; |
|---|
| | | import java.util.Collection; |
|---|
| | | import java.util.List; |
|---|
| | | import java.util.Map; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * sql语句处理工具 |
|---|
| | | * @author CJH 2022-01-12 |
|---|
| | | */ |
|---|
| | | public class SqlUtils { |
|---|
| | | |
|---|
| | | /**查询加密数据处理，只对查询做处理，select返回不做处理 |
|---|
| | | * @param sql sql语句 |
|---|
| | | * @param aesKeysTable aes秘钥 |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public static String selectSql(String sql,Map<String,Map<String,String>> aesKeysTable){ |
|---|
| | | |
|---|
| | | MySqlStatementParser parser = new MySqlStatementParser(sql); |
|---|
| | | SQLSelectStatement sqlStatement = (SQLSelectStatement) parser.parseSelect(); |
|---|
| | | //获取格式化的slq语句 |
|---|
| | | sql = sqlStatement.toString(); |
|---|
| | | |
|---|
| | | //解析select查询 |
|---|
| | | //SQLSelect sqlSelect = sqlStatement.getSelect() |
|---|
| | | //获取sql查询块 |
|---|
| | | SQLSelectQueryBlock sqlSelectQuery = (SQLSelectQueryBlock)sqlStatement.getSelect().getQuery() ; |
|---|
| | | StringBuffer out = new StringBuffer() ; |
|---|
| | | //创建sql解析的标准化输出 |
|---|
| | | SQLASTOutputVisitor sqlastOutputVisitor = SQLUtils.createFormatOutputVisitor(out , null , JdbcUtils.MYSQL) ; |
|---|
| | | |
|---|
| | | //获取表和别名 |
|---|
| | | ExportTableAliasVisitor visitorTable = new ExportTableAliasVisitor(); |
|---|
| | | sqlStatement.accept(visitorTable); |
|---|
| | | Map<String,String> tableMaps = visitorTable.getTableMap(); |
|---|
| | | |
|---|
| | | //获取所有的字段 |
|---|
| | | MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor(); |
|---|
| | | sqlStatement.accept(visitor); |
|---|
| | | //遍历所有字段 |
|---|
| | | Collection<TableStat.Column> columns= visitor.getColumns(); |
|---|
| | | |
|---|
| | | StringBuilder sqlWhere = new StringBuilder(); |
|---|
| | | |
|---|
| | | StringBuilder sqlSelect = new StringBuilder(); |
|---|
| | | String expr = null; |
|---|
| | | sqlSelect.append("SELECT "); |
|---|
| | | //解析select返回的数据字段项 |
|---|
| | | for (SQLSelectItem sqlSelectItem : sqlSelectQuery.getSelectList()) { |
|---|
| | | if(sqlSelect.length() > 7){ |
|---|
| | | sqlSelect.append(","); |
|---|
| | | } |
|---|
| | | |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectItem.accept(sqlastOutputVisitor) ; |
|---|
| | | expr = out.toString(); |
|---|
| | | sqlSelect.append(expr); |
|---|
| | | |
|---|
| | | /* if(expr.indexOf("SELECT") == -1){ |
|---|
| | | sqlSelect.append(expr); |
|---|
| | | }else{ |
|---|
| | | //sqlSelect.append("("); |
|---|
| | | sqlSelect.append(expr); |
|---|
| | | //sqlSelect.append(")"); |
|---|
| | | *//* if(!StringUtils.isEmpty(sqlSelectItem.getAlias())){ |
|---|
| | | sqlSelect.append(" AS "+sqlSelectItem.getAlias()); |
|---|
| | | }*//* |
|---|
| | | }*/ |
|---|
| | | } |
|---|
| | | |
|---|
| | | //解析from |
|---|
| | | if(sqlSelectQuery.getFrom() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getFrom().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" FROM "+out); |
|---|
| | | } |
|---|
| | | |
|---|
| | | //解析where |
|---|
| | | if(sqlSelectQuery.getWhere() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getWhere().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" WHERE "+out+" "); |
|---|
| | | } |
|---|
| | | |
|---|
| | | if(sqlSelectQuery.getGroupBy() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getGroupBy().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" "+out); |
|---|
| | | } |
|---|
| | | if(sqlSelectQuery.getOrderBy() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getOrderBy().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" "+out); |
|---|
| | | } |
|---|
| | | if(sqlSelectQuery.getLimit() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getLimit().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" "+out); |
|---|
| | | } |
|---|
| | | |
|---|
| | | //处理where需要加密得字段 |
|---|
| | | sql = sqlWhere.toString(); |
|---|
| | | if(!StringUtils.isEmpty(sql)){ |
|---|
| | | Map<String,String> aesKeys = null; |
|---|
| | | String aeskey = null; |
|---|
| | | //把剩下的拼接上来 |
|---|
| | | String tableAl = null; |
|---|
| | | for(TableStat.Column column:columns){ |
|---|
| | | aesKeys= aesKeysTable.get(column.getTable()); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | aeskey = aesKeys.getOrDefault(column.getName(),null); |
|---|
| | | if(StringUtils.isEmpty(aeskey)){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | tableAl = tableMaps.get(column.getTable()); |
|---|
| | | if(!StringUtils.isEmpty(tableAl)){ |
|---|
| | | tableAl = tableAl+"."+column.getName(); |
|---|
| | | }else{ |
|---|
| | | tableAl = column.getName(); |
|---|
| | | } |
|---|
| | | sql = sql.replaceAll("\\b"+tableAl+"\\b","AES_DECRYPT(UNHEX("+tableAl+"),'"+aeskey+"')"); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return sqlSelect.toString()+sql; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 处理select返回字段的参数 |
|---|
| | | * @param sql |
|---|
| | | * @param aesKeysTable |
|---|
| | | * @param tableMaps |
|---|
| | | * @param columns |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public static String selectSqlHandle(String sql,Map<String,Map<String,String>> aesKeysTable |
|---|
| | | ,Map<String,String> tableMaps,Collection<TableStat.Column> columns){ |
|---|
| | | |
|---|
| | | |
|---|
| | | MySqlStatementParser parser = new MySqlStatementParser(sql); |
|---|
| | | SQLSelectStatement sqlStatement = (SQLSelectStatement) parser.parseSelect(); |
|---|
| | | //获取格式化的slq语句 |
|---|
| | | sql = sqlStatement.toString(); |
|---|
| | | |
|---|
| | | //解析select查询 |
|---|
| | | //SQLSelect sqlSelect = sqlStatement.getSelect() ; |
|---|
| | | //获取sql查询块 |
|---|
| | | SQLSelectQueryBlock sqlSelectQuery = (SQLSelectQueryBlock)sqlStatement.getSelect().getQuery() ; |
|---|
| | | StringBuffer out = new StringBuffer() ; |
|---|
| | | //创建sql解析的标准化输出 |
|---|
| | | SQLASTOutputVisitor sqlastOutputVisitor = SQLUtils.createFormatOutputVisitor(out , null , JdbcUtils.MYSQL) ; |
|---|
| | | |
|---|
| | | StringBuilder sqlWhere = new StringBuilder(); |
|---|
| | | |
|---|
| | | StringBuilder sqlSelect = new StringBuilder(); |
|---|
| | | String expr = null; |
|---|
| | | sqlSelect.append("SELECT "); |
|---|
| | | //解析select返回的数据字段项 |
|---|
| | | for (SQLSelectItem sqlSelectItem : sqlSelectQuery.getSelectList()) { |
|---|
| | | if(sqlSelect.length() > 7){ |
|---|
| | | sqlSelect.append(","); |
|---|
| | | } |
|---|
| | | expr = sqlSelectItem.getExpr().toString(); |
|---|
| | | if(expr.indexOf("SELECT") == -1){ |
|---|
| | | sqlSelect.append(expr); |
|---|
| | | if(!StringUtils.isEmpty(sqlSelectItem.getAlias())){ |
|---|
| | | sqlSelect.append(" AS "+sqlSelectItem.getAlias()); |
|---|
| | | } |
|---|
| | | }else{ |
|---|
| | | sqlSelect.append("("); |
|---|
| | | selectSqlHandle(expr,aesKeysTable,tableMaps,columns); |
|---|
| | | sqlSelect.append(")"); |
|---|
| | | if(!StringUtils.isEmpty(sqlSelectItem.getAlias())){ |
|---|
| | | sqlSelect.append(" AS "+sqlSelectItem.getAlias()); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | //解析from |
|---|
| | | if(sqlSelectQuery.getFrom() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getFrom().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" FROM "+out); |
|---|
| | | } |
|---|
| | | |
|---|
| | | //解析where |
|---|
| | | if(sqlSelectQuery.getWhere() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getWhere().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" WHERE "+out+" "); |
|---|
| | | } |
|---|
| | | |
|---|
| | | if(sqlSelectQuery.getGroupBy() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getGroupBy().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" "+out); |
|---|
| | | } |
|---|
| | | if(sqlSelectQuery.getOrderBy() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getOrderBy().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" "+out); |
|---|
| | | } |
|---|
| | | if(sqlSelectQuery.getLimit() != null){ |
|---|
| | | out.delete(0, out.length()) ; |
|---|
| | | sqlSelectQuery.getLimit().accept(sqlastOutputVisitor) ; |
|---|
| | | sqlWhere.append(" "+out); |
|---|
| | | } |
|---|
| | | |
|---|
| | | sql = sqlWhere.toString(); |
|---|
| | | if(!StringUtils.isEmpty(sql)){ |
|---|
| | | Map<String,String> aesKeys = null; |
|---|
| | | String aeskey = null; |
|---|
| | | //把剩下的拼接上来 |
|---|
| | | String tableAl = null; |
|---|
| | | |
|---|
| | | for(TableStat.Column column:columns){ |
|---|
| | | aesKeys= aesKeysTable.get(column.getTable()); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | aeskey = aesKeys.getOrDefault(column.getName(),null); |
|---|
| | | if(StringUtils.isEmpty(aeskey)){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | tableAl = tableMaps.get(column.getTable()); |
|---|
| | | if(!StringUtils.isEmpty(tableAl)){ |
|---|
| | | tableAl = tableAl+"."+column.getName(); |
|---|
| | | }else{ |
|---|
| | | tableAl = column.getName(); |
|---|
| | | } |
|---|
| | | sql = sql.replaceAll("\\b"+tableAl+"\\b","AES_DECRYPT(UNHEX("+tableAl+"),'"+aeskey+"')"); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return sqlSelect.toString()+sql; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /**新增加密数据处理 |
|---|
| | | * @param sql sql语句 |
|---|
| | | * @param aesKeysTable aes秘钥 |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public static String insertSql(String sql,Map<String,Map<String,String>> aesKeysTable){ |
|---|
| | | //装载重写的sql语句 |
|---|
| | | StringBuilder splicingSql = new StringBuilder(); |
|---|
| | | |
|---|
| | | sql = SQLUtils.format(sql, JdbcConstants.MYSQL); |
|---|
| | | String[] datas = sql.split("VALUES",2); |
|---|
| | | |
|---|
| | | splicingSql.append(datas[0]+"VALUES "); |
|---|
| | | |
|---|
| | | //重新拼接SQL语句 |
|---|
| | | |
|---|
| | | //解析sql语句 |
|---|
| | | MySqlStatementParser parser = new MySqlStatementParser(sql); |
|---|
| | | SQLStatement statement = parser.parseStatement(); |
|---|
| | | MySqlInsertStatement insert = (MySqlInsertStatement)statement; |
|---|
| | | |
|---|
| | | String insertName = insert.getTableName().getSimpleName(); |
|---|
| | | |
|---|
| | | //根据表名称获取到AES秘钥 |
|---|
| | | Map<String,String> aesKeys= aesKeysTable.get(insertName); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | return sql; |
|---|
| | | } |
|---|
| | | |
|---|
| | | //获取所有的字段 |
|---|
| | | List<SQLExpr> columns = insert.getColumns(); |
|---|
| | | |
|---|
| | | String fildValue = null; |
|---|
| | | String aeskey = null; |
|---|
| | | //遍历值 |
|---|
| | | List<SQLInsertStatement.ValuesClause> vcl = insert.getValuesList(); |
|---|
| | | for(int j = 0; j<vcl.size(); j++){ |
|---|
| | | if( j != 0){ |
|---|
| | | splicingSql.append(","); |
|---|
| | | } |
|---|
| | | for(int i = 0;i < columns.size();i++){ |
|---|
| | | //查询改字段是否需要加密 |
|---|
| | | aeskey = aesKeys.getOrDefault(columns.get(i).toString(),null); |
|---|
| | | fildValue = vcl.get(j).getValues().get(i).toString(); |
|---|
| | | if(i == 0){ |
|---|
| | | splicingSql.append("("); |
|---|
| | | if(aeskey != null && fildValue.indexOf("AES_ENCRYPT") == -1){ |
|---|
| | | splicingSql.append("HEX(AES_ENCRYPT("+fildValue+",'"+aeskey+"'))"); |
|---|
| | | }else{ |
|---|
| | | splicingSql.append(fildValue); |
|---|
| | | } |
|---|
| | | }else if(i == columns.size()-1){ |
|---|
| | | splicingSql.append(","); |
|---|
| | | if(aeskey != null && fildValue.indexOf("AES_ENCRYPT") == -1){ |
|---|
| | | splicingSql.append("HEX(AES_ENCRYPT("+fildValue+",'"+aeskey+"'))"); |
|---|
| | | }else{ |
|---|
| | | splicingSql.append(fildValue); |
|---|
| | | } |
|---|
| | | splicingSql.append(")"); |
|---|
| | | }else{ |
|---|
| | | splicingSql.append(","); |
|---|
| | | if(aeskey != null && fildValue.indexOf("AES_ENCRYPT") == -1){ |
|---|
| | | splicingSql.append("HEX(AES_ENCRYPT("+fildValue+",'"+aeskey+"'))"); |
|---|
| | | }else{ |
|---|
| | | splicingSql.append(fildValue); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return splicingSql.toString(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /**更新加密数据处理 |
|---|
| | | * @param sql sql语句 |
|---|
| | | * @param aesKeysTable aes秘钥 |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public static String updateSql(String sql,Map<String,Map<String,String>> aesKeysTable){ |
|---|
| | | |
|---|
| | | //装载重写的sql语句 |
|---|
| | | StringBuilder splicingSql = new StringBuilder(); |
|---|
| | | |
|---|
| | | //sql = SQLUtils.format(sql, JdbcConstants.MYSQL); |
|---|
| | | MySqlStatementParser parser = new MySqlStatementParser(sql); |
|---|
| | | SQLStatement sqlStatement = parser.parseStatement(); |
|---|
| | | //获取格式化的slq语句 |
|---|
| | | sql = sqlStatement.toString(); |
|---|
| | | |
|---|
| | | MySqlUpdateStatement updateStatement = (MySqlUpdateStatement)sqlStatement; |
|---|
| | | |
|---|
| | | String insertName = updateStatement.getTableName().getSimpleName(); |
|---|
| | | |
|---|
| | | String[] datas = sql.split("WHERE",2); |
|---|
| | | |
|---|
| | | Map<String,String> aesKeys = aesKeysTable.get(insertName); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | return sql; |
|---|
| | | } |
|---|
| | | |
|---|
| | | splicingSql.append("UPDATE "+insertName+" SET "); |
|---|
| | | String aeskey = null; |
|---|
| | | String fildValue = null; |
|---|
| | | List<SQLUpdateSetItem> items = updateStatement.getItems(); |
|---|
| | | for(int i = 0;i<items.size();i++){ |
|---|
| | | if(i != 0){ |
|---|
| | | splicingSql.append(","); |
|---|
| | | } |
|---|
| | | SQLUpdateSetItem item = items.get(i); |
|---|
| | | //查询改字段是否需要加密 |
|---|
| | | aeskey = aesKeys.getOrDefault(item.getColumn().toString(),null); |
|---|
| | | |
|---|
| | | fildValue = item.getValue().toString(); |
|---|
| | | if(aeskey != null && fildValue.indexOf("AES_ENCRYPT") == -1){ |
|---|
| | | splicingSql.append(item.getColumn()+" = HEX(AES_ENCRYPT("+fildValue+",'"+aeskey+"'))"); |
|---|
| | | }else{ |
|---|
| | | splicingSql.append(item.getColumn()+" = "+fildValue); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | String sqlWhere = " WHERE"; |
|---|
| | | //把剩下的拼接上来 |
|---|
| | | if(datas.length > 1){ |
|---|
| | | for(int i =1;i<datas.length;i++){ |
|---|
| | | sqlWhere = sqlWhere+datas[i]; |
|---|
| | | } |
|---|
| | | |
|---|
| | | parser = new MySqlStatementParser("SELECT * FROM "+insertName+" "+sqlWhere); |
|---|
| | | sqlStatement = parser.parseStatement(); |
|---|
| | | |
|---|
| | | ExportTableAliasVisitor visitorTable = new ExportTableAliasVisitor(); |
|---|
| | | sqlStatement.accept(visitorTable); |
|---|
| | | |
|---|
| | | //获取表和别名 |
|---|
| | | Map<String,String> tableMaps = visitorTable.getTableMap(); |
|---|
| | | tableMaps.put(insertName,null); |
|---|
| | | |
|---|
| | | //获取所有的字段 |
|---|
| | | MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor(); |
|---|
| | | sqlStatement.accept(visitor); |
|---|
| | | |
|---|
| | | String tableAl = null; |
|---|
| | | //遍历所有字段 |
|---|
| | | Collection<TableStat.Column> columns= visitor.getColumns(); |
|---|
| | | for(TableStat.Column column:columns){ |
|---|
| | | |
|---|
| | | aesKeys= aesKeysTable.get(column.getTable()); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | aeskey = aesKeys.getOrDefault(column.getName(),null); |
|---|
| | | if(StringUtils.isEmpty(aeskey)){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | tableAl = tableMaps.get(column.getTable()); |
|---|
| | | if(!StringUtils.isEmpty(tableAl)){ |
|---|
| | | tableAl = tableAl+"."+column.getName(); |
|---|
| | | }else{ |
|---|
| | | tableAl = column.getName(); |
|---|
| | | } |
|---|
| | | sqlWhere = sqlWhere.replaceAll("\\b"+tableAl+"\\b","AES_DECRYPT(UNHEX("+tableAl+"),'"+aeskey+"')"); |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
| | | splicingSql.append(sqlWhere.toString()); |
|---|
| | | return splicingSql.toString(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /**删除加密数据处理 |
|---|
| | | * @param sql sql语句 |
|---|
| | | * @param aesKeysTable aes秘钥 |
|---|
| | | * @return |
|---|
| | | */ |
|---|
| | | public static String deleteSql(String sql,Map<String,Map<String,String>> aesKeysTable){ |
|---|
| | | //装载重写的sql语句 |
|---|
| | | StringBuilder splicingSql = new StringBuilder(); |
|---|
| | | |
|---|
| | | //sql = SQLUtils.format(sql, JdbcConstants.MYSQL); |
|---|
| | | MySqlStatementParser parser = new MySqlStatementParser(sql); |
|---|
| | | SQLStatement sqlStatement = parser.parseStatement(); |
|---|
| | | //获取格式化的slq语句 |
|---|
| | | sql = sqlStatement.toString(); |
|---|
| | | |
|---|
| | | MySqlDeleteStatement deleteStatement = (MySqlDeleteStatement)sqlStatement; |
|---|
| | | |
|---|
| | | String insertName = deleteStatement.getTableName().getSimpleName(); |
|---|
| | | |
|---|
| | | String[] datas = sql.split("WHERE",2); |
|---|
| | | |
|---|
| | | Map<String,String> aesKeys = aesKeysTable.get(insertName); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | return sql; |
|---|
| | | } |
|---|
| | | |
|---|
| | | splicingSql.append("DELETE FROM "+insertName); |
|---|
| | | |
|---|
| | | String aeskey = null; |
|---|
| | | |
|---|
| | | String sqlWhere = " WHERE"; |
|---|
| | | //把剩下的拼接上来 |
|---|
| | | if(datas.length > 1){ |
|---|
| | | for(int i =1;i<datas.length;i++){ |
|---|
| | | sqlWhere = sqlWhere+datas[i]; |
|---|
| | | } |
|---|
| | | |
|---|
| | | parser = new MySqlStatementParser("SELECT * FROM "+insertName+" "+sqlWhere); |
|---|
| | | sqlStatement = parser.parseStatement(); |
|---|
| | | |
|---|
| | | ExportTableAliasVisitor visitorTable = new ExportTableAliasVisitor(); |
|---|
| | | sqlStatement.accept(visitorTable); |
|---|
| | | |
|---|
| | | //获取表和别名 |
|---|
| | | Map<String,String> tableMaps = visitorTable.getTableMap(); |
|---|
| | | tableMaps.put(insertName,null); |
|---|
| | | |
|---|
| | | //获取所有的字段 |
|---|
| | | MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor(); |
|---|
| | | sqlStatement.accept(visitor); |
|---|
| | | |
|---|
| | | String tableAl = null; |
|---|
| | | //遍历所有字段 |
|---|
| | | Collection<TableStat.Column> columns= visitor.getColumns(); |
|---|
| | | for(TableStat.Column column:columns){ |
|---|
| | | |
|---|
| | | aesKeys= aesKeysTable.get(column.getTable()); |
|---|
| | | if(aesKeys == null){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | aeskey = aesKeys.getOrDefault(column.getName(),null); |
|---|
| | | if(StringUtils.isEmpty(aeskey)){ |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | tableAl = tableMaps.get(column.getTable()); |
|---|
| | | if(!StringUtils.isEmpty(tableAl)){ |
|---|
| | | tableAl = tableAl+"."+column.getName(); |
|---|
| | | }else{ |
|---|
| | | tableAl = column.getName(); |
|---|
| | | } |
|---|
| | | sqlWhere = sqlWhere.replaceAll("\\b"+tableAl+"\\b","AES_DECRYPT(UNHEX("+tableAl+"),'"+aeskey+"')"); |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
| | | splicingSql.append(sqlWhere.toString()); |
|---|
| | | return splicingSql.toString(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
