added renderer for SqlGenerateUuidExpression

metricflow/sql/render/big_query.py

@@ -4,9 +4,11 @@
     SqlExpressionRenderResult,
 )
 from metricflow.sql.render.sql_plan_renderer import DefaultSqlQueryPlanRenderer
+from metricflow.sql.sql_bind_parameters import SqlBindParameters
 from metricflow.sql.sql_exprs import (
     SqlCastToTimestampExpression,
     SqlDateTruncExpression,
+    SqlGenerateUuidExpression,
     SqlTimeDeltaExpression,
 )
 from metricflow.time.time_granularity import TimeGranularity
@@ -57,6 +59,12 @@ def visit_time_delta_expr(self, node: SqlTimeDeltaExpression) -> SqlExpressionRe
             execution_parameters=column.execution_parameters,
         )
 
+    def visit_generate_uuid_expr(self, node: SqlGenerateUuidExpression) -> SqlExpressionRenderResult:  # noqa: D
+        return SqlExpressionRenderResult(
+            sql="GENERATE_UUID()",
+            execution_parameters=SqlBindParameters(),
+        )
+
 
 class BigQuerySqlQueryPlanRenderer(DefaultSqlQueryPlanRenderer):
     """Plan renderer for the BigQuery engine."""

metricflow/sql/render/duckdb_renderer.py

@@ -4,7 +4,9 @@
     SqlExpressionRenderResult,
 )
 from metricflow.sql.render.sql_plan_renderer import DefaultSqlQueryPlanRenderer
+from metricflow.sql.sql_bind_parameters import SqlBindParameters
 from metricflow.sql.sql_exprs import (
+    SqlGenerateUuidExpression,
     SqlTimeDeltaExpression,
 )
 from metricflow.time.time_granularity import TimeGranularity
@@ -32,6 +34,12 @@ def visit_time_delta_expr(self, node: SqlTimeDeltaExpression) -> SqlExpressionRe
             execution_parameters=arg_rendered.execution_parameters,
         )
 
+    def visit_generate_uuid_expr(self, node: SqlGenerateUuidExpression) -> SqlExpressionRenderResult:  # noqa: D
+        return SqlExpressionRenderResult(
+            sql="GEN_RANDOM_UUID()",
+            execution_parameters=SqlBindParameters(),
+        )
+
 
 class DuckDbSqlQueryPlanRenderer(DefaultSqlQueryPlanRenderer):
     """Plan renderer for the DuckDB engine."""

metricflow/sql/render/expr_renderer.py

@@ -17,6 +17,7 @@
     SqlComparisonExpression,
     SqlExpressionNode,
     SqlFunction,
+    SqlGenerateUuidExpression,
     SqlAggregateFunctionExpression,
     SqlNullExpression,
     SqlLogicalExpression,
@@ -314,3 +315,9 @@ def visit_window_function_expr(self, node: SqlWindowFunctionExpression) -> SqlEx
             sql=f"{node.sql_function.value}({sql_function_args_string}) OVER ({window_string})",
             execution_parameters=combined_params,
         )
+
+    def visit_generate_uuid_expr(self, node: SqlGenerateUuidExpression) -> SqlExpressionRenderResult:  # noqa: D
+        return SqlExpressionRenderResult(
+            sql="UUID()",
+            execution_parameters=SqlBindParameters(),
+        )

metricflow/sql/render/postgres.py

@@ -4,7 +4,11 @@
     SqlExpressionRenderResult,
 )
 from metricflow.sql.render.sql_plan_renderer import DefaultSqlQueryPlanRenderer
-from metricflow.sql.sql_exprs import SqlTimeDeltaExpression
+from metricflow.sql.sql_bind_parameters import SqlBindParameters
+from metricflow.sql.sql_exprs import (
+    SqlGenerateUuidExpression,
+    SqlTimeDeltaExpression,
+)
 from metricflow.time.time_granularity import TimeGranularity
 
 
@@ -34,6 +38,12 @@ def visit_time_delta_expr(self, node: SqlTimeDeltaExpression) -> SqlExpressionRe
             execution_parameters=arg_rendered.execution_parameters,
         )
 
+    def visit_generate_uuid_expr(self, node: SqlGenerateUuidExpression) -> SqlExpressionRenderResult:  # noqa: D
+        return SqlExpressionRenderResult(
+            sql="GEN_RANDOM_UUID()",
+            execution_parameters=SqlBindParameters(),
+        )
+
 
 class PostgresSQLSqlQueryPlanRenderer(DefaultSqlQueryPlanRenderer):
     """Plan renderer for the PostgreSQL engine."""

metricflow/sql/render/redshift.py

@@ -1,5 +1,11 @@
-from metricflow.sql.render.expr_renderer import DefaultSqlExpressionRenderer, SqlExpressionRenderer
+from metricflow.sql.render.expr_renderer import (
+    DefaultSqlExpressionRenderer,
+    SqlExpressionRenderer,
+    SqlExpressionRenderResult,
+)
 from metricflow.sql.render.sql_plan_renderer import DefaultSqlQueryPlanRenderer
+from metricflow.sql.sql_bind_parameters import SqlBindParameters
+from metricflow.sql.sql_exprs import SqlGenerateUuidExpression
 
 
 class RedshiftSqlExpressionRenderer(DefaultSqlExpressionRenderer):
@@ -10,6 +16,19 @@ def double_data_type(self) -> str:
         """Custom double data type for the Redshift engine"""
         return "DOUBLE PRECISION"
 
+    def visit_generate_uuid_expr(self, node: SqlGenerateUuidExpression) -> SqlExpressionRenderResult:  # noqa: D
+        """Generates a "good enough" random key to simulate a UUID.
+
+        NOTE: This is a temporary hacky solution as redshift does not have any UUID generation function.
+
+        Proposed solutions that requires more thinking:
+            - create a python UDF (Could we insert this without needing additional permissions?)
+        """
+        return SqlExpressionRenderResult(
+            sql="CONCAT(CAST(RANDOM()*100000000 AS INT)::VARCHAR,CAST(RANDOM()*100000000 AS INT)::VARCHAR)",
+            execution_parameters=SqlBindParameters(),
+        )
+
 
 class RedshiftSqlQueryPlanRenderer(DefaultSqlQueryPlanRenderer):
     """Plan renderer for the Redshift engine."""

metricflow/sql/render/snowflake.py

@@ -0,0 +1,28 @@
+from metricflow.sql.render.expr_renderer import (
+    DefaultSqlExpressionRenderer,
+    SqlExpressionRenderer,
+    SqlExpressionRenderResult,
+)
+from metricflow.sql.render.sql_plan_renderer import DefaultSqlQueryPlanRenderer
+from metricflow.sql.sql_bind_parameters import SqlBindParameters
+from metricflow.sql.sql_exprs import SqlGenerateUuidExpression
+
+
+class SnowflakeSqlExpressionRenderer(DefaultSqlExpressionRenderer):
+    """Expression renderer for the Snowflake engine."""
+
+    def visit_generate_uuid_expr(self, node: SqlGenerateUuidExpression) -> SqlExpressionRenderResult:  # noqa: D
+        return SqlExpressionRenderResult(
+            sql="UUID_STRING()",
+            execution_parameters=SqlBindParameters(),
+        )
+
+
+class SnowflakeSqlQueryPlanRenderer(DefaultSqlQueryPlanRenderer):
+    """Plan renderer for the Snowflake engine."""
+
+    EXPR_RENDERER = SnowflakeSqlExpressionRenderer()
+
+    @property
+    def expr_renderer(self) -> SqlExpressionRenderer:  # noqa :D
+        return self.EXPR_RENDERER

metricflow/sql_clients/snowflake.py

@@ -12,6 +12,7 @@
 import sqlalchemy
 from sqlalchemy.exc import ProgrammingError
 
+from metricflow.sql.render.snowflake import SnowflakeSqlQueryPlanRenderer
 from metricflow.protocols.sql_client import SqlEngine, SqlIsolationLevel
 from metricflow.protocols.sql_client import SqlEngineAttributes
 from metricflow.protocols.sql_request import (
@@ -21,7 +22,6 @@
     MF_EXTRA_TAGS_KEY,
     SqlJsonTag,
 )
-from metricflow.sql.render.sql_plan_renderer import DefaultSqlQueryPlanRenderer
 from metricflow.sql.render.sql_plan_renderer import SqlQueryPlanRenderer
 from metricflow.sql.sql_bind_parameters import SqlBindParameters
 from metricflow.sql_clients.async_request import SqlStatementCommentMetadata, CombinedSqlTags
@@ -56,7 +56,7 @@ class SnowflakeEngineAttributes:
     random_function_name: ClassVar[str] = "RANDOM"
 
     # MetricFlow attributes
-    sql_query_plan_renderer: ClassVar[SqlQueryPlanRenderer] = DefaultSqlQueryPlanRenderer()
+    sql_query_plan_renderer: ClassVar[SqlQueryPlanRenderer] = SnowflakeSqlQueryPlanRenderer()
 
 
 class SnowflakeSqlClient(SqlAlchemySqlClient):