在 CTE 中使用用户定义的函数

Question

我有一个简单的函数，用于计算 Oracle 数据库中数据帧中的行数（ 这里是本教程中的回收代码）。

CREATE OR REPLACE FUNCTION TABCOUNT (tbl_name IN VARCHAR2)
   RETURN PLS_INTEGER
IS
   retval   PLS_INTEGER;
BEGIN
   EXECUTE IMMEDIATE 'SELECT COUNT(*) FROM ' || tbl_name INTO retval;

   RETURN retval;
END;

如果我想将此函数应用于存储表，它执行得很好：

CREATE TABLE EXAMPLE_TABLE ("ROW_ID" INT);
INSERT INTO EXAMPLE_TABLE VALUES (1);
INSERT INTO EXAMPLE_TABLE VALUES (2);

SELECT TABCOUNT('EXAMPLE_TABLE') AS "N_ROW" FROM DUAL;

但是，我想在 CTE 中使用这种类型的函数，如下所示。

WITH TABLE_1 AS (SELECT * FROM EXAMPLE_TABLE)
SELECT 'T1' AS "TABLE_NAME", TABCOUNT (TABLE_1) AS "N_ROW"
  FROM DUAL;

出于某种原因，我无法运行它。 有人可以告诉我怎么做吗？ 如果这是一个基本问题，我很抱歉，因为我对 PL/SQL 函数很陌生。

我想在 CTE 中执行此操作的简化原因是，使用上面的示例，我想将其应用于 CTE 中的各种不同表并合并结果，即：

编辑：在下面的示例中，“EXAMPLE_TABLE”和“EXAMPLE_TABLE_2”是占位符。 假设 TABLE_1 和 TABLE_2 是进一步处理查询所产生的中间表。 包含 'EXAMPLE_TABLE' 和 'EXAMPLE_TABLE_2' 只是为了使示例中的 CTE 运行。

CREATE TABLE EXAMPLE_TABLE_2 AS SELECT * FROM EXAMPLE_TABLE;

WITH TABLE_1 AS (SELECT * FROM EXAMPLE_TABLE),
     TABLE_2 AS (SELECT * FROM EXAMPLE_TABLE_2)
SELECT 'T1' AS "TABLE_NAME", TABCOUNT (TABLE_1) AS "N_ROW" FROM DUAL
UNION ALL
SELECT 'T2' AS "TABLE_NAME", TABCOUNT (TABLE_2) AS "N_ROW" FROM DUAL;

Answer 1

如果我理解你是正确的，那么你需要像下面这样的东西，

with table_list 
as 
( select 'EXAMPLE_TABLE' table_name from dual union all
  select 'EXAMPLE_TABLE_2' from dual
)
select t.table_name,tabcount(table_name)  no_of_rows
  from table_list t

但是，如果您使用的是 12c 或更高版本，您也可以在 WITH 子句中使用 PL/SQL 声明部分，

with 
function tabcount (tbl_name in varchar2)
return pls_integer 
is
  retval pls_integer;
begin
   execute immediate 'select count(*) from ' || tbl_name into retval;
   return retval;
end;
table_list 
as 
( select 'EXAMPLE_TABLE' table_name from dual union all
  select 'EXAMPLE_TABLE_2' from dual
)
select t.table_name,tabcount(table_name)  no_of_rows
  from table_list t

有关此处的WITH子句的更多详细信息

Answer 2

在您执行TABCOUNT (TABLE_1)时的查询中，唯一的TABLE_1是 CTE； 您需要引用该 CTE 中的列而不是表本身，这不是您想要的（因为row_id的值不是您打算作为tbl_name传递的值）。 CTE 将返回值为 1 和 2 的两行，而不是它们来自的表的名称。 但这是学术性的，因为查询不使用 CTE - 您正在查询dual 。

您可以将表名联合在一起：

SELECT 'EXAMPLE_TABLE' AS "TABLE_NAME", TABCOUNT ('EXAMPLE_TABLE') AS "N_ROW" FROM DUAL
UNION ALL
SELECT 'EXAMPLE_TABLE_2' AS "TABLE_NAME", TABCOUNT ('EXAMPLE_TABLE_2') AS "N_ROW" FROM DUAL;

但大概你不想复制那么多代码，特别是如果你有很多表要查看； 因此，如果您想使用 CTE，那么请生成名称：

WITH CTE (TBL_NAME) AS (
  SELECT 'EXAMPLE_TABLE' FROM DUAL
  UNION ALL
  SELECT 'EXAMPLE_TABLE_2' FROM DUAL
)
SELECT TBL_NAME AS "TABLE_NAME", TABCOUNT (TBL_NAME) AS "N_ROW" FROM CTE;

或者为了避免 CTE 中的重复，请使用集合：

SELECT COLUMN_VALUE AS "TABLE_NAME", TABCOUNT (COLUMN_VALUE) AS "N_ROW"
FROM TABLE(sys.odcivarchar2list('EXAMPLE_TABLE', 'EXAMPLE_TABLE_2'));

您可能会发现这个基于 XML 的技巧很有用，因为它不需要该函数； 适应使用相同的集合方法来提供名称并避免不推荐使用的函数：

select column_value as table_name,
  to_number(xmlquery('/ROWSET/ROW/C/text()'
    passing xmltype(dbms_xmlgen.getxml(
      'select count(*) as c from "' || column_value || '"'))
    returning content)) as n_row
from table(sys.odcivarchar2list('EXAMPLE_TABLE', 'EXAMPLE_TABLE_2'));

数据库<>小提琴

---

在下面的示例中，“EXAMPLE_TABLE”和“EXAMPLE_TABLE_2”是占位符。 假设 TABLE_1 和 TABLE_2 是进一步处理查询所产生的中间表。

我认识到我可以直接将函数应用于表，但在实际用例中，TABLE_1 和 TABLE_2 是进一步处理 CTE 的最终结果的表（而不仅仅是EXAMPLE_TABLE 和EXAMPLE_TABLE_2 的镜像副本）。

您不需要函数或动态 SQL 来执行此操作； 事实上你不能，因为 CTE 将超出动态 SQL 上下文的范围。 但是你可以做一个简单的计数：

WITH TABLE_1 AS (SELECT * FROM EXAMPLE_TABLE),
     TABLE_2 AS (SELECT * FROM EXAMPLE_TABLE_2)
SELECT 'T1' AS "TABLE_NAME", COUNT(*) AS "N_ROW" FROM TABLE_1
UNION ALL
SELECT 'T2' AS "TABLE_NAME", COUNT(*) AS "N_ROW" FROM TABLE_2;

数据库<>小提琴

Answer 3

问题是您引用TABLE_1和TABLE_2就好像它是一个列，但它是一个表。 您需要像下面的示例查询一样将表的名称传递给函数。

询问

WITH
    table_list (table_name)
    AS
        (SELECT 'ALL_TABLES' FROM DUAL
         UNION ALL
         SELECT 'ALL_OBJECTS' FROM DUAL)
SELECT table_name, TABCOUNT (table_name) AS N_ROW
  FROM table_list;

改进的功能

如果您确实打算实现这样的功能，我建议您进行一些保护以防止 SQL 注入，因为目前的方式非常容易受到攻击。 请参阅下面我的改进版本。

CREATE OR REPLACE FUNCTION TABCOUNT (p_table_name   all_tables.table_name%TYPE,
                                     p_owner        all_tables.owner%TYPE DEFAULT NULL)
    RETURN PLS_INTEGER
IS
    retval   PLS_INTEGER;
BEGIN
    EXECUTE IMMEDIATE   'SELECT COUNT(*) FROM '
                     || CASE
                            WHEN p_owner IS NOT NULL THEN DBMS_ASSERT.schema_name (p_owner) || '.'
                        END
                     || DBMS_ASSERT.sql_object_name (p_table_name)
        INTO retval;

    RETURN retval;
END;
/