如何克服 Snowflake SQL UDTF 相关子查询错误？

Question

我定义了以下 UDTF，我已确认查询与函数分开工作。

CREATE OR REPLACE FUNCTION UDF_RETAINED_CUST(period_end date)
    RETURNS TABLE(RETAINED_CUSTOMERS number)
    LANGUAGE SQL 
    AS 
    $$
    SELECT count(DISTINCT po.id_customer)
    FROM STR_PS_ORDERS po
    INNER JOIN (
        SELECT DISTINCT ID_CUSTOMER 
        FROM STR_PS_ORDERS po
        WHERE CURRENT_STATE NOT IN (6, 7, 8)
        AND ORDER_TYPE_ID NOT IN (2, 7)
        AND DATE_ADD >= dateadd(DAY, -181, period_end)
        AND DATE_ADD <= dateadd(DAY, -91, period_end)
    ) prev ON prev.id_customer = po.ID_CUSTOMER 
    WHERE CURRENT_STATE NOT IN (6, 7, 8)
    AND ORDER_TYPE_ID NOT IN (2, 7)
    AND DATE_ADD >= dateadd(DAY, -90, period_end)
    AND DATE_ADD <= period_end
    $$

我也可以用单个值成功运行函数

SELECT * FROM TABLE(UDF_RETAINED_CUST(CURRENT_DATE()))

问题是在尝试将此函数与来自另一个表的列输入一起使用时。

SELECT dd.DATE_SPK, x.retained_customers
FROM DIM_DATE dd 
JOIN TABLE(UDF_RETAINED_CUST(dd.DATE_SPK))) x

这会提示如下错误：

SQL 错误 [2031] [42601]：SQL 编译错误： 无法评估不受支持的子查询类型

Answer 1

使用 UDF 时请记住，它们本身并不是函数。在运行时它们被解包并且代码用于替换原始查询 - 因此，为了确保您的函数能够正常工作，您可以通过简单地将那段代码放入主查询并查看它的编译方式来测试它，即：

FROM DIM_DATE dd 
JOIN 
    SELECT count(DISTINCT po.id_customer)
    FROM STR_PS_ORDERS po
    INNER JOIN (
        SELECT DISTINCT ID_CUSTOMER 
        FROM STR_PS_ORDERS po
        WHERE CURRENT_STATE NOT IN (6, 7, 8)
        AND ORDER_TYPE_ID NOT IN (2, 7)
        AND DATE_ADD >= dateadd(DAY, -181, dd.DATE_SPK)
        AND DATE_ADD <= dateadd(DAY, -91, dd.DATE_SPK)
    ) prev ON prev.id_customer = po.ID_CUSTOMER 
    WHERE CURRENT_STATE NOT IN (6, 7, 8)
    AND ORDER_TYPE_ID NOT IN (2, 7)
    AND DATE_ADD >= dateadd(DAY, -90, dd.DATE_SPK)
    AND DATE_ADD <= period_end) x

在您的情况下，问题似乎出在函数中的相关子查询中 - Snowflake 中的子查询的使用非常有限，并且（除其他外）此类子查询仅允许返回单个结果，而不管值如何参与。

要完成这项工作，您需要重写查询以消除子查询中的相关性