feat: Adding NAME data type in Postgres/Redshift (#3128)

* feat: Adding NAME data type in Postgres/Redshift * Changing test case
tobymao · Mar 12, 2024 · 0ea849b · 0ea849b
1 parent 9b25a8e
commit 0ea849b
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 1 deletion.
diff --git a/sqlglot/dialects/postgres.py b/sqlglot/dialects/postgres.py
@@ -278,6 +278,7 @@ class Tokenizer(tokens.Tokenizer):
             "REVOKE": TokenType.COMMAND,
             "SERIAL": TokenType.SERIAL,
             "SMALLSERIAL": TokenType.SMALLSERIAL,
+            "NAME": TokenType.NAME,
             "TEMP": TokenType.TEMPORARY,
             "CSTRING": TokenType.PSEUDO_TYPE,
             "OID": TokenType.OBJECT_IDENTIFIER,

diff --git a/sqlglot/expressions.py b/sqlglot/expressions.py
@@ -3787,6 +3787,7 @@ class Type(AutoName):
         MEDIUMINT = auto()
         MEDIUMTEXT = auto()
         MONEY = auto()
+        NAME = auto()
         NCHAR = auto()
         NESTED = auto()
         NULL = auto()
@@ -3843,6 +3844,7 @@ class Type(AutoName):
         Type.NVARCHAR,
         Type.TEXT,
         Type.VARCHAR,
+        Type.NAME,
     }
 
     INTEGER_TYPES = {

diff --git a/sqlglot/parser.py b/sqlglot/parser.py
@@ -258,6 +258,7 @@ class Parser(metaclass=_Parser):
         TokenType.IPV6,
         TokenType.UNKNOWN,
         TokenType.NULL,
+        TokenType.NAME,
         *ENUM_TYPE_TOKENS,
         *NESTED_TYPE_TOKENS,
         *AGGREGATE_TYPE_TOKENS,

diff --git a/sqlglot/tokens.py b/sqlglot/tokens.py
@@ -135,6 +135,7 @@ class TokenType(AutoName):
     LONGBLOB = auto()
     TINYBLOB = auto()
     TINYTEXT = auto()
+    NAME = auto()
     BINARY = auto()
     VARBINARY = auto()
     JSON = auto()

diff --git a/tests/dialects/test_postgres.py b/tests/dialects/test_postgres.py
@@ -608,7 +608,16 @@ def test_postgres(self):
                 "postgres": "x / y ^ z",
             },
         )
-
+        self.validate_all(
+            "CAST(x AS NAME)",
+            read={
+                "redshift": "CAST(x AS NAME)",
+            },
+            write={
+                "postgres": "CAST(x AS NAME)",
+                "redshift": "CAST(x AS NAME)",
+            },
+        )
         self.assertIsInstance(self.parse_one("id::UUID"), exp.Cast)
 
     def test_ddl(self):

diff --git a/tests/fixtures/identity.sql b/tests/fixtures/identity.sql
@@ -859,3 +859,4 @@ SELECT only
 TRUNCATE(a, b)
 SELECT enum
 SELECT unlogged
+SELECT name