feat(flinksql): collect table alias

Author: zhaoge

src/grammar/flink/FlinkSqlParser.g4

@@ -512,8 +512,8 @@ tableReference
     ;
 
 tablePrimary
-    : KW_TABLE? tablePath systemTimePeriod? (KW_AS? correlationName)?
-    | viewPath systemTimePeriod? (KW_AS? correlationName)?
+    : KW_TABLE? tablePath systemTimePeriod?
+    | viewPath systemTimePeriod?
     | KW_LATERAL KW_TABLE LR_BRACKET functionName LR_BRACKET functionParam (COMMA functionParam)* RR_BRACKET RR_BRACKET
     | KW_LATERAL? LR_BRACKET queryStatement RR_BRACKET
     | KW_UNNEST LR_BRACKET expression RR_BRACKET
@@ -833,7 +833,7 @@ intervalValue
     ;
 
 tableAlias
-    : KW_AS? identifier identifierList?
+    : KW_AS? tbAlias=identifier identifierList?
     ;
 
 errorCapturingIdentifier

src/parser/common/entityCollector.ts

@@ -65,11 +65,13 @@ const baseAlias: BaseAliasContext = {
 export enum attrName {
     comment = '_comment',
     colType = '_colType',
+    alias = '_tbAlias',
 }
 
 export const attrNameInRule = {
     [attrName.comment]: 'comment',
     [attrName.colType]: 'colType',
+    [attrName.alias]: 'alias',
 } as const;
 
 type ParserRuleContextWithAttr = ParserRuleContext & {
@@ -83,6 +85,7 @@ export interface EntityContext extends BaseAliasContext {
     relatedEntities: EntityContext[] | null;
     columns: ColumnEntityContext[] | null;
     comment?: string;
+    alias?: string;
 }
 
 export interface FuncEntityContext extends EntityContext {
@@ -126,6 +129,7 @@ export function toEntityContext(
         for (let k = 0; k < attrInfo?.attrList?.length; k++) {
             const attributeName: attrName = attrInfo?.attrList[k];
             const attrToken = findAttribute(ctx, attributeName, attrInfo?.endContext);
+            console.log(attrToken, attributeName);
             if (attrToken) {
                 const attrVal: string = tokenToWord(attrToken, input)?.text;
                 extraInfo[attrNameInRule[attributeName]] = attrVal;

src/parser/flink/flinkEntityCollector.ts

@@ -45,7 +45,11 @@ export class FlinkEntityCollector extends EntityCollector implements FlinkSqlPar
     }
 
     exitTablePath(ctx: TablePathContext) {
-        this.pushEntity(ctx, EntityContextType.TABLE);
+        this.pushEntity(ctx, EntityContextType.TABLE, {
+            needCollectAttr: true,
+            attrList: [attrName.alias],
+            endContext: 'TableReferenceContext',
+        });
     }
 
     exitTablePathCreate(ctx: TablePathCreateContext) {

test/parser/flink/contextCollect/entityCollector.test.ts

@@ -167,7 +167,7 @@ describe('Flink entity collector tests', () => {
         expect(allEntities.length).toBe(1);
 
         const tableEntity = allEntities[0];
-
+        console.log(tableEntity, '---tableEntity');
         expect(tableEntity.entityContextType).toBe(EntityContextType.TABLE);
         expect(tableEntity.text).toBe('Orders');
         expect(tableEntity.belongStmt.stmtContextType).toBe(StmtContextType.SELECT_STMT);

test/parser/flink/contextCollect/fixtures/common.sql

@@ -20,7 +20,7 @@ CREATE TABLE Orders_with_watermark (
     INCLUDING GENERATED
 );
 
-SELECT order_id, price + tax FROM Orders;
+SELECT order_id, price + tax FROM Orders AS o1;
 
 SELECT * FROM Orders LEFT JOIN Product ON Orders.product_id = Product.id;