feat(datafusion, flink, mssql): add uuid operation (#8545)

Continued from [PR](#8438): add uuid operation for mssql, flink, datafusion,
ibis-project · Mar 7, 2024 · 2f85a42 · 2f85a42
1 parent df2b69e
commit 2f85a42
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 6 deletions.
diff --git a/ibis/backends/datafusion/compiler.py b/ibis/backends/datafusion/compiler.py
@@ -82,6 +82,7 @@ class DataFusionCompiler(SQLGlotCompiler):
         ops.Last: "last_value",
         ops.Median: "median",
         ops.StringLength: "character_length",
+        ops.RandomUUID: "uuid",
         ops.RegexSplit: "regex_split",
     }
 

diff --git a/ibis/backends/flink/compiler.py b/ibis/backends/flink/compiler.py
@@ -80,6 +80,7 @@ class FlinkCompiler(SQLGlotCompiler):
         ops.MapValues: "map_values",
         ops.Power: "power",
         ops.RandomScalar: "rand",
+        ops.RandomUUID: "uuid",
         ops.RegexSearch: "regexp",
         ops.StrRight: "right",
         ops.StringLength: "char_length",

diff --git a/ibis/backends/mssql/compiler.py b/ibis/backends/mssql/compiler.py
@@ -133,6 +133,7 @@ class MSSQLCompiler(SQLGlotCompiler):
         ops.Log10: "log10",
         ops.Power: "power",
         ops.RandomScalar: "rand",
+        ops.RandomUUID: "newid",
         ops.Repeat: "replicate",
         ops.Reverse: "reverse",
         ops.StringAscii: "ascii",

diff --git a/ibis/backends/tests/test_uuid.py b/ibis/backends/tests/test_uuid.py
@@ -19,6 +19,7 @@
     "exasol": "UUID",
     "flink": "CHAR(36) NOT NULL",
     "impala": "STRING",
+    "mssql": "uniqueidentifier",
     "postgres": "uuid",
     "risingwave": "character varying",
     "snowflake": "VARCHAR",
@@ -42,11 +43,8 @@ def test_uuid_literal(con, backend):
 
 @pytest.mark.notimpl(
     [
-        "datafusion",
         "druid",
         "exasol",
-        "flink",
-        "mssql",
         "mysql",
         "oracle",
         "polars",
@@ -64,11 +62,8 @@ def test_uuid_function(con):
 
 @pytest.mark.notimpl(
     [
-        "datafusion",
         "druid",
         "exasol",
-        "flink",
-        "mssql",
         "mysql",
         "oracle",
         "polars",