业务中需要对Trino查询语句中涉及的字段进行处理,所以对这一段源码进行追踪分析,记录如下。
-
源码追踪是从QueuedStatementResource类开始的。
QueuedStatementResource类与ExecutingStatementResource类,提供用户执行查询相关的Restful接口。
由于解决业务问题中未涉及ExecutingStatementResource类,所以本文只从QueuedStatementResource类开始追踪分析。
上图中postStatement接口提交一个新的查询,在这里可以捕获到客户端传递来的sql语句(statement参数)。
从return createQueryResultsResponse(query.getQueryResults(query.getLastToken(), uriInfo));
可以看出其返回了查询结果,所以可以从这个方法开始查看sql语句是如何解析的。
sql语句(statement参数)应用在registeQuery()中,所以追踪这一方法。
在new Query()中进一步使用了statement,所以继续追踪,追踪至submitIfNeeded()
。 -
DispatchManager
继续追踪,进入DispatchManager的createQuery()方法。
之后,在createQueryInternal()
方法中,preparedQuery = queryPreparer.prepareQuery(session, query);
对SQL语句进行词法语法分析。之后,通过dispatchQueryFactory.createDispatchQuery()
返回一个调度查询。追踪这一方法在LocalDispatchQueryFactory类中createDispatchQuery()方法创建一个QueryExecution。
…
因为之前参考网友整理的源码流程图,涉及Analyzer,所以中间省略继续追踪过程,在Analysis中发现解析出的字段名及表名(部分流程可参考下图)。
-
Analyzer
省略部分过程(可能是SqlBaseParser、AstBuilder…)后,发现这里的tableName和columns为目标信息。
-
Analysis
所以继续追踪查找TableColumnReference相关设置。在Analysis类中,构建表信息getReferencedTables()
方法中可捕获SQL涉及的表及字段信息。