(improvement)(chat) support nested query sql (#210)

This commit is contained in:
lexluo09
2023-10-13 17:24:10 +08:00
committed by GitHub
parent 5583426115
commit 151963ea79
2 changed files with 16 additions and 1 deletions

View File

@@ -139,8 +139,16 @@ public class SqlParserSelectHelper {
}
public static List<String> getAllFields(String sql) {
List<PlainSelect> plainSelects = getPlainSelects(getPlainSelect(sql));
Set<String> results = new HashSet<>();
for (PlainSelect plainSelect : plainSelects) {
List<String> fields = getFieldsByPlainSelect(plainSelect);
results.addAll(fields);
}
return new ArrayList<>(results);
}
PlainSelect plainSelect = getPlainSelect(sql);
private static ArrayList<String> getFieldsByPlainSelect(PlainSelect plainSelect) {
if (Objects.isNull(plainSelect)) {
return new ArrayList<>();
}