皮皮网

1.Cobar源码分析之AST

Cobar源码分析之AST

       Cobar是一款阿里开源的数据库中间件，专注于分库分表的路由功能。SQL，作为一种领域语言，广泛用于关系型数据库的管理和操作，其执行流程一般包括词法分析、海外没有溯源码语法分析、语义分析生成AST（抽象语法树），然后由优化器生成执行计划，最后由执行引擎执行。

       在Cobar中，SQL Parser负责将SQL语句解析为AST。开发SQL Parser通常有两种方式。Cobar的SQL Parser经历了三个版本迭代，主要目的一刻源码是提升性能。本文将深入解析Cobar中的SQL Parser和AST，包括其结构、操作和应用。

       AST（抽象语法树）直观表示SQL语句的结构。以SQL语句“select id,type from goods as g where type in (select type from type_config where status = 0)”为例，Cobar SQL Parser将其解析为AST对象，其根节点为select语句，公众号下单源码其他属性为叶子节点。

       AST的Node定义包括accept方法，用于遍历树结构。实现AST主要涉及几个类，如ComparisionEqualsExpression，其中1为比较的左右表达式，2为判断符，蛤乎项目源码3为计算表达式。evaluationInternal方法实现比较逻辑，将表达式结构化并穷举后，实现简单的数值比较。

       AST操作中，遍历是最基本的，利用ASTNode的流动登录注册源码accept方法和SQLASTVisitor接口实现。SQLParserDelegate.parse(sql)解析结果为DMLSelectStatement对象，其visit方法负责构建SQL输出格式。MySQLOutputASTVisitor实现遍历并输出SQL语句，如“SELECT id, type FROM goods AS G WHERE type IN (SELECT type FROM type_config WHERE status = 0)”。Cobar利用AST进行分库分表，获取表名、列名、比较的值。

       AST还可以生成SQL特征，如将原始SQL归一化为“select id, name, age from user as u where age >= ?”，在进行SQL慢查询统计、限流等操作时非常有用。Cobar利用AST对线上SQL进行安全检查，拦截无条件的update或delete语句。

       本文从SQL AST的来源、结构、遍历原理、应用等方面进行介绍，通过Cobar项目中的单元测试，读者可以实际感受AST的功能和应用。了解SQL AST有助于后端技术、架构设计、性能优化、源码阅读、问题排查等领域。

       欢迎搜索关注微信公众号"捉虫大师"，获取更多后端技术分享，包括架构设计、性能优化、源码阅读、问题排查和踩坑实践等内容。