feat(singlestore): support dcolonqmark (#6485)

* support dcolonqmark

* add testcases

Author: AbhishekASLK

sqlglot/dialects/singlestore.py

@@ -81,6 +81,7 @@ class Tokenizer(MySQL.Tokenizer):
             "!:>": TokenType.NCOLON_GT,
             "::$": TokenType.DCOLONDOLLAR,
             "::%": TokenType.DCOLONPERCENT,
+            "::?": TokenType.DCOLONQMARK,
         }
 
     class Parser(MySQL.Parser):
@@ -253,6 +254,12 @@ class Parser(MySQL.Parser):
             TokenType.DCOLONPERCENT: lambda self, this, path: build_json_extract_path(
                 exp.JSONExtractScalar, json_type="DOUBLE"
             )([this, exp.Literal.string(path.name)]),
+            TokenType.DCOLONQMARK: lambda self, this, path: self.expression(
+                exp.JSONExists,
+                this=this,
+                path=path.name,
+                from_dcolonqmark=True,
+            ),
         }
         COLUMN_OPERATORS.pop(TokenType.ARROW)
         COLUMN_OPERATORS.pop(TokenType.DARROW)
@@ -452,8 +459,10 @@ def _unicode_substitute(m: re.Match[str]) -> str:
             exp.JSONBExists: lambda self, e: self.func(
                 "BSON_MATCH_ANY_EXISTS", e.this, e.args.get("path")
             ),
-            exp.JSONExists: unsupported_args("passing", "on_condition")(
-                lambda self, e: self.func("JSON_MATCH_ANY_EXISTS", e.this, e.args.get("path"))
+            exp.JSONExists: lambda self, e: (
+                f"{self.sql(e.this)}::?{self.sql(e.args.get('path'))}"
+                if e.args.get("from_dcolonqmark")
+                else self.func("JSON_MATCH_ANY_EXISTS", e.this, e.args.get("path"))
             ),
             exp.JSONObject: unsupported_args(
                 "null_handling", "unique_keys", "return_type", "encoding"

sqlglot/expressions.py

@@ -7023,7 +7023,13 @@ class JSONArrayAgg(Func):
 
 
 class JSONExists(Func):
-    arg_types = {"this": True, "path": True, "passing": False, "on_condition": False}
+    arg_types = {
+        "this": True,
+        "path": True,
+        "passing": False,
+        "on_condition": False,
+        "from_dcolonqmark": False,
+    }
 
 
 # https://docs.oracle.com/en/database/oracle/oracle-database/19/sqlrf/JSON_TABLE.html

sqlglot/tokens.py

@@ -41,6 +41,7 @@ class TokenType(AutoName):
     DCOLON = auto()
     DCOLONDOLLAR = auto()
     DCOLONPERCENT = auto()
+    DCOLONQMARK = auto()
     DQMARK = auto()
     SEMICOLON = auto()
     STAR = auto()

tests/dialects/test_singlestore.py

@@ -965,3 +965,6 @@ def test_constraints(self):
         self.validate_identity(
             "CREATE TABLE ComputedColumnConstraint (points INT, score AS (points * 2) PERSISTED BIGINT NOT NULL)"
         )
+
+    def test_dcolonqmark(self):
+        self.validate_identity("SELECT * FROM employee WHERE JSON_MATCH_ANY(payroll::?names)")