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企服之家，中国第一个企服评测及商务社交产业平台。