主页 / dba / 问题 / 338569
Accepted
Björn Morén
Asked: 2024-04-12 13:37:18 +0800 CST

LEFT JOIN LATERAL 带有函数结果，如何防止不必要的函数调用？

  • 772

我有一个昂贵的函数，它仅在满足某些条件时提供相关数据。该数据通过 LEFT JOIN LATERAL 插入到结果中，如下所示。现在，该函数会为每一行调用。

有没有办法重写它，以便仅对 t1.type 等于 5 的行调用它？尽管我设置了条件，但下面仍然为每一行调用该函数。

SELECT t1.name, t1.type, t2.col1, t2.col2
FROM table1 t1
LEFT JOIN LATERAL function_name(t1.col1, t1.col2) AS t2 ON t1.type = 5;

测试用例：

CREATE TABLE table1 (
    id SERIAL PRIMARY KEY,
    name VARCHAR(100),
    type INTEGER,
    col1 INTEGER,
    col2 INTEGER
);

-- Sample data for table1
INSERT INTO table1 (name, type, col1, col2)
VALUES
    ('Row 1', 5, 10, 20),
    ('Row 2', 7, 15, 25),
    ('Row 3', 10, 30, 40); 

CREATE OR REPLACE FUNCTION function_name(col1_arg INTEGER, col2_arg INTEGER)
RETURNS TABLE (col1 INTEGER, col2 INTEGER)
AS $$
BEGIN
    RAISE LOG 'Function called';
    RETURN QUERY SELECT col1_arg * 2 AS col1, col2_arg * 3 AS col2;
END;
$$ LANGUAGE plpgsql;


-- Test Query
SELECT t1.name, t1.type, t2.col1, t2.col2
FROM table1 t1
LEFT JOIN LATERAL function_name(t1.col1, t1.col2) AS t2 ON t1.type = 5;
postgresql

2 个回答

  1. Best Answer
    SELECT t0.name, t0.type, t2.col1, t2.col2
FROM table1 t0
LEFT JOIN table1 t1 ON t0.id = t1.id AND t1.type = 5
LEFT JOIN LATERAL function_name(t1.col1, t1.col2) AS t2 ON t1.id IS NOT NULL;

    小提琴

    当我在 PostgreSQL 服务器上运行你的代码时，它每次都会调用该函数。– 比约恩·莫伦

    测试类似的东西：

    WITH 
cte AS (
    SELECT * FROM table1 WHERE type = 5
    )
SELECT t1.name, t1.type, t2.col1, t2.col2
FROM cte t1
CROSS JOIN LATERAL function_name(t1.col1, t1.col2) AS t2
UNION ALL 
SELECT name, type, NULL, NULL
FROM table1 
WHERE type <> 5 -- OR type IS NULL
;
    • 1

  2. 尝试#1：将条件放入 WHERE 子句中有效，但它会更改查询结果，因为它将删除 type!=5 的所有行：

    SELECT t1.name, t1.type, t2.col1, t2.col2
FROM table1 t1
LEFT JOIN LATERAL function_name(t1.col1, t1.col2) AS t2 ON (true) 
WHERE t1.type=5;

    ...但是您可以将行放回：

    SELECT t1.name, t1.type, t2.col1, t2.col2
FROM table1 t1
LEFT JOIN LATERAL function_name(t1.col1, t1.col2) AS t2 ON (true) 
WHERE t1.type=5
UNION ALL
SELECT t1.name, t1.type, NULL, NULL
FROM table1 t1
WHERE t1.type!=5

    如果“table1”不是表而是复杂的查询，您可以将其放入物化 CTE 中以避免执行两次。

    尝试＃2：将“RETURNS NULL ON NULL INPUT”添加到函数中，并在您实际上不想调用它时使用空值来调用它，这不会执行该函数，这要快得多：

    SELECT t1.name, t1.type, t2.col1, t2.col2
FROM table1 t1
LEFT JOIN LATERAL (SELECT t1.col1,t1.col2 WHERE t1.type = 5) tt ON (true)
LEFT JOIN LATERAL function_name(tt.col1, tt.col2) t2 ON (true);

    （当您不想调用函数时，也可以使用 CASE 将函数的所有参数设置为 NULL）。

    尝试#3

    SELECT t1.name, t1.type, (f).*
FROM table1 t1
LEFT JOIN LATERAL (
  SELECT function_name(tt.col1, tt.col2) f FROM (SELECT t1.col1,t1.col2 WHERE t1.type=5) tt
) t2 ON (true);

    这似乎工作正常，函数仅在应该调用时才被调用。

    尝试#4：

    SELECT t1.name, t1.type, t2.col1, t2.col2
FROM table1 t1
LEFT JOIN LATERAL (
  SELECT t3.* FROM 
  (SELECT t1.col1,t1.col2 WHERE t1.type = 5) tt 
  LEFT JOIN LATERAL function_name(tt.col1, tt.col2) t3 ON (true)) t2 
 ON (true);

    这也很好用。

    • 1

相关问题