博客

  public static SQLSelect getSqlSelect(String sql, List<String> groupByList, String dbType) {
        if (CollectionUtils.isEmpty(groupByList)) {
            return null;
        }
    
        List<SQLStatement> statements = SQLUtils.parseStatements(sql, dbType);
        SQLSelectStatement selectStatement = (SQLSelectStatement) statements.get(0);
        SQLSelect select = selectStatement.getSelect();
        if (select.getQuery() instanceof SQLUnionQuery) {
            SQLUnionQuery sqlUnionQuery = (SQLUnionQuery) select.getQuery();
            SQLSelectQueryBlock rightQuery = (SQLSelectQueryBlock) sqlUnionQuery.getRight();
            rebuildQueryGroupByAndSelect(groupByList, rightQuery);
        
            SQLSelectQueryBlock leftQuery = (SQLSelectQueryBlock) sqlUnionQuery.getLeft();
            rebuildQueryGroupByAndSelect(groupByList, leftQuery);
        } else {
            SQLSelectQueryBlock query = (SQLSelectQueryBlock) select.getQuery();
            rebuildQueryGroupByAndSelect(groupByList, query);
        }
        return select;
    }

private static void rebuildQueryGroupByAndSelect(List<String> groupByList, SQLSelectQueryBlock query) {
        // 将 旧的group by的字段从select item 换成新的group by 字段
        SQLSelectGroupByClause oldGroupBy = query.getGroupBy();
        List<SQLSelectItem> selectList = query.getSelectList();
        if (oldGroupBy != null) {
            List<SQLExpr> oldGroupByItems = oldGroupBy.getItems();
           
            for (SQLExpr oldGroupByItem : oldGroupByItems) {
                int j = selectList.size();
                // 如果group by 是 a，则 a 和 t.a都可以删
                if (oldGroupByItem instanceof SQLIdentifierExpr) {
                    for (int i = j - 1; i >= 0; i--) {
                        SQLSelectItem selectItem = selectList.get(i);
                        if (selectItem.getExpr() instanceof SQLIdentifierExpr) {
                            if (oldGroupByItem.toString().equalsIgnoreCase(selectItem.getExpr().toString())) {
                                selectList.remove(i);
                            }
                        } else  if (selectItem.getExpr() instanceof SQLPropertyExpr) {
                            SQLPropertyExpr propertyExpr = (SQLPropertyExpr) selectItem.getExpr();
                            if (oldGroupByItem.toString().equalsIgnoreCase(propertyExpr.getName())) {
                                selectList.remove(i);
                            }
                        }
                    }
                } else if(oldGroupByItem instanceof SQLPropertyExpr) {
                    //如果 group by 是 t.a, 则 a 和 t.a都可以删
                    String name = ((SQLPropertyExpr) oldGroupByItem).getName();
                    String expr = oldGroupByItem.toString();
                    for (int i = j - 1; i >= 0; i--) {
                        SQLSelectItem selectItem = selectList.get(i);
                        if (selectItem.getExpr() instanceof SQLIdentifierExpr) {
                            if (name.equalsIgnoreCase(selectItem.getExpr().toString())) {
                                selectList.remove(i);
                            }
                        } else  if (selectItem.getExpr() instanceof SQLPropertyExpr) {
                            if (selectItem.getExpr().toString().equalsIgnoreCase(expr)) {
                                selectList.remove(i);
                            }
                        }
                    }
                }
            }
        }
        // 加上新的group by的字段到select
        for (String selectItem : groupByList) {
            selectList.add(new SQLSelectItem(new SQLIdentifierExpr(selectItem)));
        }
        
        // 重组group by
        SQLSelectGroupByClause group = new SQLSelectGroupByClause();
        for (String s : groupByList) {
            group.addItem(new SQLIdentifierExpr(s));
        }
        query.setGroupBy(group);
    }