IT评测·应用市场-qidao123.com技术社区

标题: SQL字段来源表的解析 [打印本页]
作者: 我爱普洱茶    时间: 2024-11-23 02:43
标题: SQL字段来源表的解析
测试例子：

  1. SELECT e.NAME, d.DEPT_NAME,d.DEPT_ID,EMP_ID,100+EMP_ID+100 FROM EMP e JOIN DEPT d ON e.DEPT_ID = d.DEPT_ID WHERE e.EMP_ID IN (SELECT EMP_ID FROM EMP WHERE DEPT_ID = 10)
复制代码
代码示例：

  1. package com.test;
  2. import org.apache.calcite.jdbc.CalciteConnection;
  3. import org.apache.calcite.rel.RelNode;
  4. import org.apache.calcite.rel.metadata.RelColumnOrigin;
  5. import org.apache.calcite.rel.type.RelDataType;
  6. import org.apache.calcite.rel.type.RelDataTypeField;
  7. import org.apache.calcite.schema.impl.AbstractTable;
  8. import org.apache.calcite.sql.SqlNode;
  9. import org.apache.calcite.sql.parser.SqlParseException;
  10. import org.apache.calcite.sql.type.SqlTypeName;
  11. import org.apache.calcite.tools.FrameworkConfig;
  12. import org.apache.calcite.tools.Frameworks;
  13. import org.apache.calcite.tools.Planner;
  14. import org.apache.calcite.tools.RelConversionException;
  15. import org.apache.calcite.tools.ValidationException;
  16. import org.apache.calcite.rel.RelRoot;
  17. import org.apache.calcite.schema.SchemaPlus;
  18. import org.apache.calcite.rel.type.RelDataTypeFactory;
  19. import org.apache.calcite.rel.metadata.RelMetadataQuery;
  21. import java.sql.DriverManager;
  22. import java.sql.SQLException;
  23. import java.util.Properties;
  24. import java.util.Set;
  26. public class SqlLineageExample {
  28.     public static void main(String[] args) throws SQLException, ValidationException, RelConversionException, SqlParseException {
  29.         // 创建 Calcite 连接
  30.         Properties info = new Properties();
  31.         info.setProperty("lex", "JAVA");
  32.         CalciteConnection connection = (CalciteConnection) DriverManager.getConnection("jdbc:calcite:", info);
  34.         SchemaPlus rootSchema = connection.getRootSchema();
  36.         // 定义并添加自定义表到 schema
  37.         rootSchema.add("EMP", new AbstractTable() {
  38.             @Override
  39.             public RelDataType getRowType(RelDataTypeFactory typeFactory) {
  40.                 return typeFactory.builder()
  41.                         .add("EMP_ID", SqlTypeName.INTEGER)
  42.                         .add("NAME", SqlTypeName.VARCHAR, 20)
  43.                         .add("DEPT_ID", SqlTypeName.INTEGER)
  44.                         .build();
  45.             }
  46.         });
  48.         rootSchema.add("DEPT", new AbstractTable() {
  49.             @Override
  50.             public RelDataType getRowType(RelDataTypeFactory typeFactory) {
  51.                 return typeFactory.builder()
  52.                         .add("DEPT_ID", SqlTypeName.INTEGER)
  53.                         .add("DEPT_NAME", SqlTypeName.VARCHAR, 20)
  54.                         .build();
  55.             }
  56.         });
  58.         // 创建 FrameworkConfig
  59.         FrameworkConfig config = Frameworks.newConfigBuilder()
  60.                 .defaultSchema(rootSchema)
  61.                 .build();
  63.         // 创建 Planner
  64.         Planner planner = Frameworks.getPlanner(config);
  66.         // 复杂 SQL 查询
  67.         String sql = "SELECT e.NAME, d.DEPT_NAME,d.DEPT_ID,EMP_ID,100+EMP_ID+100 FROM EMP e JOIN DEPT d ON e.DEPT_ID = d.DEPT_ID " +
  68.                 "WHERE e.EMP_ID IN (SELECT EMP_ID FROM EMP WHERE DEPT_ID = 10)";
  70.         // 解析 SQL
  71.         SqlNode parsedNode = planner.parse(sql);
  73.         // 校验 SQL
  74.         SqlNode validatedNode = planner.validate(parsedNode);
  76.         // 转换为关系代数树
  77.         RelRoot relRoot = planner.rel(validatedNode);
  78.         RelNode relNode = relRoot.project();
  80.         // 打印字段来源
  81.         printFieldLineage(relNode);
  82.     }
  84.     private static void printFieldLineage(RelNode relNode) {
  85.         RelMetadataQuery mq = relNode.getCluster().getMetadataQuery();
  86.         for (RelDataTypeField field : relNode.getRowType().getFieldList()) {
  87.             Set<RelColumnOrigin> origins = mq.getColumnOrigins(relNode, field.getIndex());
  88.             if (origins != null) {
  89.                 for (RelColumnOrigin origin : origins) {
  90.                     String fieldName = field.getName();
  91.                     String tableName = origin.getOriginTable().getQualifiedName().get(0);
  92.                     System.out.println("Field: " + fieldName + " comes from Table: " + tableName);
  93.                 }
  94.             } else {
  95.                 System.out.println("Field: " + field.getName() + " origin unknown");
  96.             }
  97.         }
  98.     }
  99. }
复制代码
结果输出：

打印结果如下：
Field: NAME comes from Table: EMP
Field: DEPT_NAME comes from Table: DEPT
Field: DEPT_ID comes from Table: DEPT
Field: EMP_ID comes from Table: EMP
Field: EXPR$4 comes from Table: EMP
 
依赖 JAR包

  1. <dependencies>
  2.     <dependency>
  3.         <groupId>org.apache.calcite</groupId>
  4.         <artifactId>calcite-core</artifactId>
  5.         <version>1.34.0</version>
  6.     </dependency>
  7.     <dependency>
  8.         <groupId>org.slf4j</groupId>
  9.         <artifactId>slf4j-api</artifactId>
  10.         <version>1.7.36</version>
  11.     </dependency>
  12.     <dependency>
  13.         <groupId>org.slf4j</groupId>
  14.         <artifactId>slf4j-log4j12</artifactId>
  15.         <version>1.7.36</version>
  16.     </dependency>
  17. </dependencies>
复制代码
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。



欢迎光临 IT评测·应用市场-qidao123.com技术社区 (https://dis.qidao123.com/) Powered by Discuz! X3.4