Reuse sessionProperty in SET SESSION grammer

trinodb · Sep 20, 2024 · bf3228e · bf3228e
1 parent 6b2b763
commit bf3228e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/core/trino-grammar/src/main/antlr4/io/trino/grammar/sql/SqlBase.g4 b/core/trino-grammar/src/main/antlr4/io/trino/grammar/sql/SqlBase.g4
@@ -178,7 +178,7 @@ statement
         (LIKE pattern=string (ESCAPE escape=string)?)?                 #showSession
     | SET SESSION AUTHORIZATION authorizationUser                      #setSessionAuthorization
     | RESET SESSION AUTHORIZATION                                      #resetSessionAuthorization
-    | SET SESSION qualifiedName EQ expression                          #setSession
+    | SET SESSION sessionProperty                                      #setSession
     | RESET SESSION qualifiedName                                      #resetSession
     | START TRANSACTION (transactionMode (',' transactionMode)*)?      #startTransaction
     | COMMIT WORK?                                                     #commit

diff --git a/core/trino-parser/src/main/java/io/trino/sql/parser/AstBuilder.java b/core/trino-parser/src/main/java/io/trino/sql/parser/AstBuilder.java
@@ -1529,7 +1529,8 @@ public Node visitShowSession(SqlBaseParser.ShowSessionContext context)
     @Override
     public Node visitSetSession(SqlBaseParser.SetSessionContext context)
     {
-        return new SetSession(getLocation(context), getQualifiedName(context.qualifiedName()), (Expression) visit(context.expression()));
+        SessionProperty sessionProperty = (SessionProperty) visit(context.sessionProperty());
+        return new SetSession(getLocation(context), sessionProperty.getName(), sessionProperty.getValue());
     }
 
     @Override