Snowflake /当存储过程作为OWNER执行时如何获取原始调用者

Question

我创建了一个存储过程，其行为以所有者身份执行（我需要所有者的权限）。

我正在寻找调用它的用户的原始角色，有什么办法吗？ 不幸的是，正如预期的那样，select current_role() 语句返回了所有者的角色......

如果存在解决方法，如果您能向我描述一下，我将不胜感激。

按照 Mike 的建议：这里有我为测试 Harsh 的解决方案所做的代码：

又失败了^^ 这是我的代码。也许你会看到我的错误。

create or replace procedure do_noting_proc()
returns string
language javascript
execute as OWNER
AS
$$
 var retour="Resultat";
 try{
   rs0=snowflake.createStatement({sqlText:"SELECT $QUERY_TAG;"}).execute();
   cR=rs0.getColumnValue(1);
   return "QRValue= "+callerRole;        
 }catch(err){
    return "do_noting_proc KO: "+ err.message+";";
 }
$$;

create or replace procedure test_proc()
returns string
language javascript
execute as CALLER
AS
$$
var rString="Result -->";
try{
  rs0=snowflake.createStatement({sqlText:"SELECT CURRENT_ROLE();"}).execute();
  rs0.next();
  cr="testQT"+rs0.getColumnValue(1);
  snowflake.createStatement({sqlText:"alter session set QUERY_TAG="+cr+";"}).execute();
  rs=snowflake.createStatement({sqlText:"call  PUBLIC.do_noting_proc();"}).execute();
  rs.next();
  rString+=rs.getColumnValue(1);
}catch(err){
    return "test_proc: "+ err.message+";";
}
return "test_proc : "+rString;
$$;

那么call share_view_or_table(); ==>test_proc : ** 结果 -->do_noting_proc KO：在所有者权限存储过程中不允许使用会话变量“$QUERY_TAG”；** （由系统管理员创建并以系统管理员身份执行的过程）

另外，如果我运行“代码” 选择 $QUERY_TAG；

我收到错误：SQL 编译错误：第 1 行错误，位置 7 会话变量“$QUERY_TAG”不存在

而

SHOW PARAMETERs LIKE 'QUERY_TAG' IN SESSION ; 

工作正常并打印 TESTQTSYSADMIN

@Harsh 非常感谢您抽出宝贵时间

Answer 1

当使用EXECUTE AS OWNERrules do not permit reading the caller's session details时，包括他们的身份或角色：

所有者权限存储过程在会话中遵循以下规则：

[…]

• 无法访问大多数特定于呼叫者的信息。例如：

• 无法查看、设置或取消设置调用者的会话变量。

• 只能读取特定的会话参数，不能设置或取消设置调用者的任何会话参数。

• 无法查询基于当前用户返回结果的 INFORMATION_SCHEMA 表函数，例如 AUTOMATIC_CLUSTERING_HISTORY。

虽然不方便，但您可以使用存储过程中的逻辑来强制调用者的角色名称应作为参数传递：

CREATE OR REPLACE PROCEDURE PROC(ROLE_NAME STRING)
[…]
var VALID_ROLES_ARRAY = ["FOO", "BAR"];
var IS_VALID_CALLER_ROLENAME = VALID_ROLES_ARRAY.includes(ROLE_NAME);
[…];

CALL PROC(CURRENT_ROLE());