jsqlparser无法解析带if语句问题
admin
2024-05-14 07:42:40
0次

项目中引入了sharding-sphere encry 包做数据脱敏版本 4.0.1
mybatis-plus版本用的是3.1.0使用内置分页插件,默认引用的jsqlparser版本是1.4,maven中央库的版本范围是1.4-4.5,实际上仅支持到1.4

业务上报了一个奇怪的错误,一下sql语句数组越界

SELECT COUNT(1) FROM ( select u.name,u.telephone,if(u.age>18,'成年','未成年') as ageType,from user uwhere u.deleted = 0AND u.telephone = ?)

这个语句是由mybatis-plus生成的sql,理论上生成出来的应该是优化过的sql,如下

SELECT COUNT(1) FROM from user uwhere u.deleted = 0AND u.telephone = ?

但是为什么用了源sql,这是第一个问题,sql优化使用的是jsqlparser-1.4版本,如果能优化就会生成上面的sql,如果抛异常使用原sql。为什么如此简单的sql会抛异常呢?

于是写了个测试用例,新建一个空项目引入对于版本包

com.github.jsqlparserjsqlparser1.4

用例

public class JsqlparserTest {public static final Logger logger= LoggerFactory.getLogger(JsqlparserTest.class);public static void main(String[] args) throws JSQLParserException {String sql ="SELECT COUNT(1) FROM ( select u.name,u.telephone,if(u.age>18,'成年','未成年') as ageType,\n"+ "        from user u\n"+ "        where u.deleted = 0\n"+ "            AND u.telephone = ?)";Select selectStatement = (Select) CCJSqlParserUtil.parse(sql);PlainSelect plainSelect = (PlainSelect) selectStatement.getSelectBody();SqlInfo parser = parser(sql);System.out.println("xxx");}public static SqlInfo parser(String sql) {if (logger.isDebugEnabled()) {logger.debug(" JsqlParserCountOptimize sql=" + sql);}SqlInfo sqlInfo = SqlInfo.newInstance();try {Select selectStatement = (Select) CCJSqlParserUtil.parse(sql);PlainSelect plainSelect = (PlainSelect) selectStatement.getSelectBody();Distinct distinct = plainSelect.getDistinct();List orderBy = plainSelect.getOrderByElements();List groupBy = plainSelect.getGroupByColumnReferences();// 优化 SQLplainSelect.setSelectItems(countSelectItem());sqlInfo.setSql(selectStatement.toString());return sqlInfo;} catch (Throwable e) {// 无法优化使用原 SQLreturn null;}}private static List countSelectItem() {Function function = new Function();function.setName("COUNT");List expressions = new ArrayList<>();LongValue longValue = new LongValue(1);ExpressionList expressionList = new ExpressionList();expressions.add(longValue);expressionList.setExpressions(expressions);function.setParameters(expressionList);List selectItems = new ArrayList<>();SelectExpressionItem selectExpressionItem = new SelectExpressionItem(function);selectItems.add(selectExpressionItem);return selectItems;}}

执行报错如下

Exception in thread "main" net.sf.jsqlparser.JSQLParserExceptionat net.sf.jsqlparser.parser.CCJSqlParserUtil.parse(CCJSqlParserUtil.java:55)at com.example.democ.JsqlparserTest.main(JsqlparserTest.java:34)
Caused by: net.sf.jsqlparser.parser.ParseException: Encountered unexpected token: "if" "IF"at line 1, column 50.Was expecting one of:
...此处省略
Process finished with exit code 1

异常说的是不支持if语句,这个也好办,使用case when 代替就好了。
第二个问题,即使这里优化不成功,使用原sql为什么sharding-sphere会报数组越界呢?
接着看数组越界的位置

@RequiredArgsConstructor
@Getter
public abstract class AbstractSQLBuilder implements SQLBuilder {private final SQLRewriteContext context;@Overridepublic final String toSQL() {if (context.getSqlTokens().isEmpty()) {return context.getSql();}Collections.sort(context.getSqlTokens());StringBuilder result = new StringBuilder();result.append(context.getSql().substring(0, context.getSqlTokens().get(0).getStartIndex()));for (SQLToken each : context.getSqlTokens()) {result.append(getSQLTokenText(each));result.append(getConjunctionText(each));}return result.toString();}protected abstract String getSQLTokenText(SQLToken sqlToken);private String getConjunctionText(final SQLToken sqlToken) {return context.getSql().substring(getStartIndex(sqlToken), getStopIndex(sqlToken));}private int getStartIndex(final SQLToken sqlToken) {int startIndex = sqlToken instanceof Substitutable ? ((Substitutable) sqlToken).getStopIndex() + 1 : sqlToken.getStartIndex();return Math.min(startIndex, context.getSql().length());}private int getStopIndex(final SQLToken sqlToken) {int currentSQLTokenIndex = context.getSqlTokens().indexOf(sqlToken);return context.getSqlTokens().size() - 1 == currentSQLTokenIndex ? context.getSql().length() : context.getSqlTokens().get(currentSQLTokenIndex + 1).getStartIndex();}
}

我们配置了telephone 字段加密,此时发现sql token集合中出现了4个元素 两个telephone 两个 ?意味着此处投影中的 telephone片段也被作为条件恶意取了个问号填冲进去,原因是它在子查询中。详见EncryptProjectionTokenGenerator
然后升级了个版本4.1.1之后发现,这个类做了一个更新
片段如下

    private SubstitutableColumnNameToken generateSQLToken(ColumnProjectionSegment segment, String tableName) {String encryptColumnName = this.getEncryptColumnName(tableName, segment.getColumn().getIdentifier().getValue());if (!segment.getAlias().isPresent()) {encryptColumnName = encryptColumnName + " AS " + segment.getColumn().getIdentifier().getValue();}return segment.getColumn().getOwner().isPresent() ? new SubstitutableColumnNameToken(((OwnerSegment)segment.getColumn().getOwner().get()).getStopIndex() + 2, segment.getStopIndex(), encryptColumnName) : new SubstitutableColumnNameToken(segment.getStartIndex(), segment.getStopIndex(), encryptColumnName);}

此处将投影和条件做了一个区分,投影字段加了个AS进去,这样在后续解析的时候就能区分的出来哪个是条件需要在preparement阶段加密传进来的参数,哪个是结果集,需要在resultset阶段解密。

总结
将if替换为case when 解决jsqpparser无法解析if的问题
升级shardingphere 4.1.1解决加解密数组越界的问题

由此引发的思考

在各种有相关联组件整合的时候,可能发生你意想不到的问题,可能在这个问题中,你用单独用任何一个包,可能都不会出现问题,但是一旦他们结合就出现问题了。遇到此类问题的时候可以考虑下版本的问题,当然此处的shardingsphere 本身就存在问题

相关内容

热门资讯

AI做攻略、“双向奔赴”……国... 本文转自:人民网人民网记者 许维娜今年国庆假期,AI(人工智能)成了不少年轻人的“旅行搭子”。行前做...
合肥的这个“梗”, 翻篇了 社交平台的评论区,曾长期流传一句专属于合肥的黑色幽默。有人问合肥哪里好玩,本地网友半开玩笑地回复:合...
7万+旅游中高层都在的群,今天... 做旅游这行,资源卡位、渠道互通、项目落地——往往就差一个“对的机会”。我们的行业社群已聚集70,00...
一场帐篷音乐节,如何让井研从“... 帐篷音乐节现场封面新闻记者 周洪攀10月的研溪湿地,晚风里裹着柑橘的清甜与民谣的回响。夜幕降临,10...
白鹤翩翩落星愿,上海迪士尼度假... 10月9日,世界候鸟日前夕,上海迪士尼度假区星愿公园的湖风里多了一分白鹤的灵动。三座白鹤雕塑正式揭幕...
成都锦江边赏初秋枫叶飘飘 目前,成都锦江边国华街枫叶树红了。近年来,国华街附近的几条街都栽种了红枫树,是城区较集中看枫叶的点位...
GBTA:商务旅行正在复苏,但... 亚太正在成为全球商务旅行增长的重要引擎。在经历疫情后的快速反弹之后,全球商务旅行正在进入一个新的增长...
从景区焕新到节赛引流 邹城双节... 齐鲁网·闪电新闻10月9日讯 中秋、国庆“双节”假期,邹城市文化和旅游局提前谋划、周密部署,立足自身...
江山如此多娇 且看石楼今朝 金秋十月,黄河岸畔秋意盎然,丰收盛会喜迎国庆。本次石楼国庆文旅活动以“江山如此多娇,且看石楼今朝”为...
海南国庆假期接待游客475.4... 2026年国庆假期,海南接待游客475.40万人次,较2025年增长13.1%,较2024年增长15...