Merge pull request #491 from liuxy0551/fix_487

fix(hive): support hyphenated config property names in SET statement (#487)

Author: Cythia828

src/grammar/hive/HiveSqlParser.g4

@@ -86,7 +86,7 @@ execStatement
     | KW_MERGE QUERY_HINT? KW_INTO tableName (KW_AS? id_)? KW_USING joinSourcePart KW_ON expression whenClauses
     | KW_PREPARE id_ KW_FROM queryStatementExpression
     | KW_EXECUTE id_ KW_USING constantList
-    | KW_SET configPropertiesItem ((DOT | COLON) configPropertiesItem)* EQUAL .*?
+    | KW_SET configProperty EQUAL .*?
     ;
 
 loadStatement
@@ -2265,6 +2265,16 @@ sql11ReservedKeywordsUsedAsFunctionName
     | KW_TIMESTAMP
     ;
 
+configProperty
+    : configPropertyPart ((DOT | COLON) configPropertyPart)*
+    ;
+
+configPropertyPart
+    : configPropertiesItem (
+        {this.isNextTokenAdjacent()}? MINUS {this.isNextTokenAdjacent()}? configPropertiesItem
+    )*
+    ;
+
 configPropertiesItem
     : id_
     | KW_SELECT

src/lib/SQLParserBase.ts

@@ -56,4 +56,22 @@ export abstract class SQLParserBase<T = antlr.ParserRuleContext> extends antlr.P
         // This allows ANTLR to report errors naturally
         return false;
     }
+
+    /**
+     * Semantic predicate to check whether the previous token and the next token
+     * are adjacent (no whitespace in between).
+     *
+     * Used to support hyphens inside Hive SET config property names, e.g.
+     * `set tez.grouping.max-size = 1`, while still rejecting invalid spacing
+     * like `set hive.a - b = 1`.
+     */
+    public isNextTokenAdjacent(): boolean {
+        const previousToken = this.tokenStream.LT(-1);
+        const nextToken = this.tokenStream.LT(1);
+        return (
+            previousToken !== null &&
+            nextToken !== null &&
+            previousToken.stop + 1 === nextToken.start
+        );
+    }
 }