创建SQL函数：如何检查调用该函数的当前查询是否使用指定表？

刚好之前研究过类似的需求，要在SQL函数里检测当前调用它的查询有没有JOIN指定表，其实得看你用的是什么数据库——不同系统提供的工具不一样，我给你拆解几种主流情况：

PostgreSQL 实现方案 #

PostgreSQL提供了直接获取当前查询的函数current_query()，你可以基于这个做两种层级的检测：

1. 简单字符串匹配（快速但有局限性） #

适合需求不那么严格的场景，直接检查查询文本里是否包含JOIN和目标表名的组合：

CREATE OR REPLACE FUNCTION check_portfolio_join()
RETURNS VARCHAR AS $$
DECLARE
    current_sql TEXT;
BEGIN
    -- 获取当前正在执行的SQL语句
    current_sql := current_query();
    
    -- 用ILIKE做大小写不敏感匹配，覆盖JOIN/INNER JOIN/LEFT JOIN等情况
    IF current_sql ILIKE '%JOIN%portfolio%' THEN
        -- 存在portfolio表的JOIN，执行你的业务逻辑
        RETURN '已关联portfolio表，返回对应结果';
    ELSE
        -- 未关联，执行另一分支逻辑
        RETURN '未关联portfolio表，返回默认结果';
    END IF;
END;
$$ LANGUAGE plpgsql;

2. AST语法树解析（精准但复杂） #

如果担心字符串匹配误判（比如查询里的WHERE条件刚好有'portfolio'字符串），可以用PostgreSQL的pg_parse_query()解析查询的抽象语法树，精准提取JOIN的表名：

CREATE OR REPLACE FUNCTION check_portfolio_join_ast()
RETURNS VARCHAR AS $$
DECLARE
    query_ast JSON;
    join_table_names TEXT[];
BEGIN
    -- 将当前查询解析为JSON格式的AST
    query_ast := pg_parse_query(current_query())::JSON;
    
    -- 遍历AST的JOIN节点，提取所有关联的表名
    SELECT array_agg(DISTINCT jsonpath_query_value(elem, '$.RangeVar.relname'))
    INTO join_table_names
    FROM jsonb_array_elements(query_ast::JSONB) AS root,
         jsonb_array_elements(root->'SelectStmt'->'jointree'->'FromExpr'->'fromlist') AS elem
    WHERE elem ? 'RangeVar';
    
    -- 检查目标表是否在关联列表中
    IF 'portfolio' = ANY(join_table_names) THEN
        RETURN '精准检测到portfolio表关联';
    ELSE
        RETURN '未检测到portfolio表关联';
    END IF;
EXCEPTION
    WHEN OTHERS THEN
        -- 处理非SELECT语句（比如INSERT/UPDATE）的解析失败情况
        RETURN '无法解析当前查询';
END;
$$ LANGUAGE plpgsql;

MySQL 实现方案 #

MySQL可以通过information_schema.processlist获取当前线程的执行SQL，逻辑和PostgreSQL的字符串匹配类似：

DELIMITER //
CREATE FUNCTION check_portfolio_join()
RETURNS VARCHAR(255)
DETERMINISTIC
BEGIN
    DECLARE current_sql TEXT;
    
    -- 读取当前连接的执行SQL
    SELECT sql_text INTO current_sql
    FROM information_schema.processlist
    WHERE id = CONNECTION_ID();
    
    -- 匹配JOIN和表名，注意大小写（MySQL默认大小写敏感取决于系统配置）
    IF current_sql LIKE '%JOIN%portfolio%' OR current_sql LIKE '%join%portfolio%' THEN
        RETURN '已关联portfolio表';
    ELSE
        RETURN '未关联portfolio表';
    END IF;
END //
DELIMITER ;

通用注意事项 #

不管用哪种方法，都要注意几个坑：

• 误判风险：字符串匹配会把查询中其他位置出现的表名字符串当成关联（比如WHERE note = 'portfolio update'），AST解析能避免这个问题，但实现更复杂
• 别名问题：如果表用了别名（比如JOIN portfolio p ON ...），字符串匹配依然能命中，但AST解析会直接提取原始表名，更精准
• 动态SQL限制：如果查询是动态生成的（比如用EXECUTE执行的动态语句），current_query()或processlist里只会显示外层的EXECUTE语句，无法检测内部的JOIN
• 权限要求：需要确保函数的执行用户有足够权限——PostgreSQL需要pg_monitor权限查看完整查询，MySQL需要PROCESS权限访问processlist

内容的提问来源于stack exchange，提问作者omega