feat(snowflake)!: annotation support for CURRENT_ROLE.

sqlglot/expressions.py

@@ -6348,6 +6348,10 @@ class CurrentRegion(Func):
     arg_types = {}
 
 
+class CurrentRole(Func):
+    arg_types = {}
+
+
 class CurrentOrganizationUser(Func):
     arg_types = {}
 

sqlglot/typing/snowflake.py

@@ -362,6 +362,7 @@ def _annotate_math_with_float_decfloat(
             exp.Collation,
             exp.CurrentOrganizationUser,
             exp.CurrentRegion,
+            exp.CurrentRole,
             exp.CurrentOrganizationName,
             exp.DecompressString,
             exp.HexDecodeString,

tests/dialects/test_mysql.py

@@ -214,6 +214,7 @@ def test_identity(self):
         self.validate_identity("""SELECT 'ab' MEMBER OF('[23, "abc", 17, "ab", 10]')""")
         self.validate_identity("""SELECT * FROM foo WHERE 'ab' MEMBER OF(content)""")
         self.validate_identity("SELECT CURRENT_TIMESTAMP(6)")
+        self.validate_identity("SELECT CURRENT_ROLE()")
         self.validate_identity("x ->> '$.name'")
         self.validate_identity("SELECT CAST(`a`.`b` AS CHAR) FROM foo")
         self.validate_identity("SELECT TRIM(LEADING 'bla' FROM ' XXX ')")

tests/dialects/test_postgres.py

@@ -79,6 +79,7 @@ def test_postgres(self):
         self.validate_identity("EXEC AS myfunc @id = 123", check_command_warning=True)
         self.validate_identity("SELECT CURRENT_SCHEMA")
         self.validate_identity("SELECT CURRENT_USER")
+        self.validate_identity("SELECT CURRENT_ROLE()")
         self.validate_identity("SELECT * FROM ONLY t1")
         self.validate_identity("SELECT INTERVAL '-1 MONTH'")
         self.validate_identity("SELECT INTERVAL '4.1 DAY'")

tests/dialects/test_snowflake.py

@@ -231,6 +231,7 @@ def test_snowflake(self):
         self.validate_identity("SELECT OBJECT_CONSTRUCT()")
         self.validate_identity("SELECT CURRENT_ORGANIZATION_USER()")
         self.validate_identity("SELECT CURRENT_REGION()")
+        self.validate_identity("SELECT CURRENT_ROLE()")
         self.validate_identity("SELECT YEAR(CURRENT_TIMESTAMP())")
         self.validate_identity("SELECT YEAROFWEEK(CURRENT_TIMESTAMP())")
         self.validate_identity("SELECT YEAROFWEEKISO(CURRENT_TIMESTAMP())")

tests/fixtures/optimizer/annotate_functions.sql

@@ -2180,6 +2180,10 @@ VARCHAR;
 CURRENT_REGION();
 VARCHAR;
 
+# dialect: snowflake
+CURRENT_ROLE();
+VARCHAR;
+
 # dialect: snowflake
 CURRENT_ORGANIZATION_NAME();
 VARCHAR;