SQLMesh 内置宏详解:@PIVOT等常用宏的核心用法与示例

发布于:2025-05-21 ⋅ 阅读:(13) ⋅ 点赞:(0)

本文系统解析 SQLMesh 的四个核心内置宏,涵盖行列转换的 @PIVOT、精准去重的 @DEDUPLICATE、灵活生成日期范围的 @DATE_SPINE,以及动态表路径解析的 @RESOLVE_TEMPLATE。通过真实案例演示参数配置与 SQL 渲染逻辑,并对比宏调用与传统 SQL 的差异,助您高效构建标准化数据处理流程。

SQLMesh 提供了一系列强大的内置宏,用于简化 SQL 查询和数据处理任务。本文将深入介绍四个常用宏:

  1. @PIVOT:将长格式数据转换为宽格式(行列转换)。
  2. @DEDUPLICATE:基于分区列和排序条件去重。
  3. @DATE_SPINE:生成日期范围表,便于时间序列分析。
  4. @RESOLVE_TEMPLATE:动态解析表路径,适用于云存储场景。

通过实际示例,我们将展示这些宏如何提升 SQL 查询的灵活性和效率。在这里插入图片描述

在这里插入图片描述

1. @PIVOT:行列转换利器

功能概述

@PIVOT 用于将长格式数据(单列存储多个值)转换为宽格式(每行对应多个列)。常用于统计分析,如订单状态分布、用户行为分类等。

参数说明

参数 说明 默认值
column 要透视的列 必填
values 用于透视的值列表 必填
alias 是否为结果列添加别名 true
agg 聚合函数(如 SUM, COUNT SUM
cmp 比较运算符(=, >, < =
prefix / suffix 别名前缀/后缀
then_value / else_value 条件成立/不成立时的值 1 / 0
quote 是否对别名加引号 true
distinct 聚合时是否去重 false

示例

假设有一个 rides 表,记录骑行订单状态(cancelled, completed),我们希望统计每日各状态的订单数:

SQLMesh 宏写法
SELECT
  date_day,
  @PIVOT(status, ['cancelled', 'completed'])
FROM rides
GROUP BY 1
渲染后的 SQL
SELECT
  date_day,
  SUM(CASE WHEN status = 'cancelled' THEN 1 ELSE 0 END) AS "'cancelled'",
  SUM(CASE WHEN status = 'completed' THEN 1 ELSE 0 END) AS "'completed'"
FROM rides
GROUP BY 1
进阶用法

如果只想统计 completed 状态的订单数,并计算取消率:

SELECT
  date_day,
  @PIVOT(status, ['completed'], agg := 'COUNT') AS completed_orders,
  @PIVOT(status, ['cancelled'], agg := 'COUNT') AS cancelled_orders,
  @PIVOT(status, ['cancelled'], agg := 'COUNT') * 1.0 / 
    NULLIF(@PIVOT(status, ['completed'], agg := 'COUNT'), 0) AS cancellation_rate
FROM rides
GROUP BY 1

2. @DEDUPLICATE:高效去重

功能概述

@DEDUPLICATE 使用窗口函数 ROW_NUMBER() 对数据进行去重,适用于需要保留每组最新/最旧记录的场景。

参数说明

参数 说明 示例
relation 表名或 CTE my_table
partition_by 分区列 [id, event_date]
order_by 排序条件 ['event_date DESC', 'status ASC']

示例

假设 my_table 存储用户事件日志,我们希望保留每个用户每天的最新记录:

SQLMesh 宏写法
WITH raw_data AS (
  @DEDUPLICATE(my_table, [id, CAST(event_date AS DATE)], ['event_date DESC', 'status ASC'])
)
SELECT * FROM raw_data
渲染后的 SQL
WITH "raw_data" AS (
  SELECT *
  FROM "my_table" AS "my_table"
  QUALIFY ROW_NUMBER() OVER (
    PARTITION BY "id", CAST("event_date" AS DATE) 
    ORDER BY "event_date" DESC, "status" ASC
  ) = 1
)
SELECT * FROM "raw_data" AS "raw_data"

3. @DATE_SPINE:生成日期范围表

功能概述

@DATE_SPINE 用于生成连续的日期序列,常用于时间序列分析、数据补全等场景。

参数说明

参数 说明 示例
datepart 时间粒度(day, week, month 'day'
start_date 起始日期 '2024-01-01'
end_date 结束日期 '2024-01-16'

示例

生成 2024 年 1 月 1 日至 1 月 16 日的日期表:

SQLMesh 宏写法
WITH discount_promotion_dates AS (
  @DATE_SPINE('day', '2024-01-01', '2024-01-16')
)
SELECT * FROM discount_promotion_dates
渲染后的 SQL(DuckDB)
WITH "discount_promotion_dates" AS (
  SELECT "_exploded"."date_day" AS "date_day"
  FROM UNNEST(CAST(GENERATE_SERIES(
    CAST('2024-01-01' AS DATE), 
    CAST('2024-01-16' AS DATE), 
    INTERVAL '1' DAY
  ) AS DATE[])) AS "_exploded"("date_day")
)
SELECT "discount_promotion_dates"."date_day" AS "date_day"
FROM "discount_promotion_dates" AS "discount_promotion_dates"

4. @RESOLVE_TEMPLATE:动态表路径解析

功能概述

@RESOLVE_TEMPLATE 用于动态生成表路径,适用于云存储(如 S3)或引擎元数据管理。

参数说明

参数 说明 示例
template 字符串模板 's3://bucket/@{catalog_name}/@{schema_name}/@{table_name}'
mode 返回类型(literaltable literal

示例

在模型定义中动态指定存储路径:

SQLMesh 宏写法
MODEL (
  name datalake.landing.customers,
  physical_properties (
    location = @resolve_template('s3://warehouse-data/@{catalog_name}/prod/@{schema_name}/@{table_name}')
  )
)
渲染后的 SQL
-- 生成的表路径示例:
-- s3://warehouse-data/datalake/prod/sqlmesh__landing/landing__customers__2517971505

总结

SQLMesh 的内置宏极大提升了 SQL 查询的灵活性和可维护性:

  • @PIVOT 简化行列转换,适用于统计分析。
  • @DEDUPLICATE 高效去重,优化数据清洗流程。
  • @DATE_SPINE 自动生成日期序列,简化时间序列分析。
  • @RESOLVE_TEMPLATE 动态解析表路径,适配云存储场景。

掌握这些宏的使用方法,可以显著提升 SQL 开发效率,减少重复代码。建议在实际项目中结合业务需求灵活运用! 🚀


网站公告

今日签到

点亮在社区的每一天
去签到