Snowflake JavaScript存储过程未继承Java会话查询标签问题求解

以下是几种可行的解决方法：

方法1：将Query Tag作为参数传入存储过程 #

直接把Java端设置的Query Tag作为参数传递给存储过程，在存储过程内部显式设置会话的Query Tag，确保后续查询继承该标签。

Java调用代码示例 #

// 获取当前已设置的Query Tag
String currentQueryTag = connection.getMetaData().getQueryTag();
// 调用存储过程并传入Query Tag参数
CallableStatement stmt = connection.prepareCall("CALL YOUR_STORED_PROCEDURE(?)");
stmt.setString(1, currentQueryTag);
stmt.execute();

JavaScript存储过程示例 #

CREATE OR REPLACE PROCEDURE YOUR_STORED_PROCEDURE(p_query_tag STRING)
RETURNS VARCHAR
LANGUAGE JAVASCRIPT
AS
$$
    // 显式设置会话的Query Tag
    snowflake.execute({sqlText: `ALTER SESSION SET QUERY_TAG = '${p_query_tag}'`});
    
    // 执行存储过程的业务逻辑查询
    let result = snowflake.execute({sqlText: "SELECT * FROM YOUR_TABLE"});
    
    return "Success";
$$;

方法2：利用会话变量传递Query Tag #

在调用存储过程前，将当前Query Tag存入会话变量，存储过程内部读取该变量并设置到会话中。

Java端前置操作 #

// 将当前Query Tag存入会话变量
Statement stmt = connection.createStatement();
stmt.execute("SET MY_CALLER_QUERY_TAG = CURRENT_QUERY_TAG()");
stmt.close();

// 调用存储过程
CallableStatement procStmt = connection.prepareCall("CALL YOUR_STORED_PROCEDURE()");
procStmt.execute();

JavaScript存储过程示例 #

CREATE OR REPLACE PROCEDURE YOUR_STORED_PROCEDURE()
RETURNS VARCHAR
LANGUAGE JAVASCRIPT
AS
$$
    // 读取会话变量中的Query Tag
    let tagResult = snowflake.execute({sqlText: "SELECT $MY_CALLER_QUERY_TAG"});
    let queryTag = tagResult.getColumnValue(1);
    
    // 设置会话Query Tag
    snowflake.execute({sqlText: `ALTER SESSION SET QUERY_TAG = '${queryTag}'`});
    
    // 执行业务逻辑
    snowflake.execute({sqlText: "INSERT INTO YOUR_TABLE VALUES (1, 'test')"});
    
    return "Success";
$$;

方法3：使用调用者会话上下文执行存储过程 #

创建存储过程时指定EXECUTE AS CALLER，让存储过程在调用者的会话上下文中运行，自动继承调用者的Query Tag（注意此方法会让存储过程以调用者的权限执行，需评估权限风险）。

创建存储过程示例 #

CREATE OR REPLACE PROCEDURE YOUR_STORED_PROCEDURE()
RETURNS VARCHAR
LANGUAGE JAVASCRIPT
EXECUTE AS CALLER -- 关键配置：使用调用者的会话
AS
$$
    // 直接执行业务逻辑，自动继承调用者的Query Tag
    let result = snowflake.execute({sqlText: "SELECT COUNT(*) FROM YOUR_TABLE"});
    
    return result.getColumnValue(1).toString();
$$;

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