• 基于Durid解析SQL语法树获取单查询该查询的最终输出字段


    该记录不涉及任何业务信息,仅作学习记录,代码在最下面。

    需求:要解析用户的SQL,从语法上获取SQL的输出字段,这样好和后面的入库字段做映射。

    本来方案是,直接用Durid里面自带得HiveSchemaStatVisitor,进行SQL遍历,就可以获取所需的Select字段。测试结果发现不是这样,原生的解析拿不到子查询的明细字段,也拿不到"*"对应的字段,如下图。

    这个是一个测试用例:

    1. CREATE TABLE `tb_a` (
    2. `id` varchar(64) COLLATE utf8_bin NOT NULL,
    3. `name` varchar(64) DEFAULT NULL,
    4. `phone` varchar(64) DEFAULT NULL
    5. ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;
    6. CREATE TABLE `tb_c` (
    7. `id` varchar(64) COLLATE utf8_bin NOT NULL,
    8. `address` varchar(64) DEFAULT NULL,
    9. `cost` varchar(64) DEFAULT NULL
    10. ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;
    11. CREATE TABLE `tb_a` (
    12. `id` varchar(64) COLLATE utf8_bin NOT NULL,
    13. `level` varchar(64) DEFAULT NULL,
    14. `email` varchar(64) DEFAULT NULL
    15. ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin;
    16. SELECT
    17. a.*,
    18. b.*,
    19. tb_c.email
    20. FROM
    21. tb_a a
    22. JOIN (
    23. SELECT id as id2, sum( cost ) AS sum FROM tb_b GROUP BY id ) b
    24. ON tb_a.id = tb_b
    25. JOIN tb_c
    26. ON tb_c.id = tb_a.id
    27. #应该要输出
    28. #tb_a.id或者a.id,以及别名id
    29. #tb_a.name或者a.name, 以及别名name
    30. #tb_a.phone或者a.phone, 以及别名phone
    31. #b.id, 以及别名id
    32. #b.sum(cost), 以及别名sum
    33. #tb_c.email,以及别名email

    所以,又双叒叕是头秃的一天,老老实实写递归(写链表也得,怎么都得的),实现这个逻辑,并且配合元数据服务,获取相关实体表的字段,就能做到:

    1、将“*”对应的字段展现

    2、将子查询内的字段展现(最终是输出的字段),如果最终不输出,也不考虑这些字段。

    3、将各类计算的字段和其别名展现。

    下图是和上图用的同一句SQL,测试输出的结果。

    代码还是全部放出来,直接运行就可以了,有些冗余。如果有修改或更好的方式,欢迎沟通。

    :D

    1. package com.dapp;
    2. import com.alibaba.druid.DbType;
    3. import com.alibaba.druid.sql.SQLUtils;
    4. import com.alibaba.druid.sql.ast.SQLExpr;
    5. import com.alibaba.druid.sql.ast.SQLObject;
    6. import com.alibaba.druid.sql.ast.SQLStatement;
    7. import com.alibaba.druid.sql.ast.expr.SQLPropertyExpr;
    8. import com.alibaba.druid.sql.ast.statement.*;
    9. import java.util.*;
    10. /**
    11. * Created by LH on 2022/9/2 17:43
    12. */
    13. public class ParserDemo{
    14. //原始表字段
    15. public static HashMap> originalTables = new HashMap
  • 相关阅读:
    按关键字搜索lazada商品接口(支持高并发请求多个国家站数据),代码对接教程
    LeetCode算法练习top100:(4)链表
    学完Spring框架回头再来看反射你会发现真的不一样
    jdk的安装及环境配置
    2022选择了交大,回顾这一年的成长
    UE5实战篇二(对话系统1):导语
    图信号处理论文
    Fabric.js 自由绘制圆形
    用一个 Python 脚本实现依次运行其他多个带 argparse 命令行参数的 .py 文件
    C++入门笔记分享
  • 原文地址:https://blog.csdn.net/lilyjoke/article/details/126679248