예산에 대한 사용자 지정 작업¶

저장 프로시저 요구 사항¶

사용자 지정 작업에 의해 호출되는 저장 프로시저는 다음 요구 사항을 충족해야 합니다.

• 프로시저는 호출자의 권한이 아닌 소유자의 권한으로 실행해야 합니다. 자세한 내용은 호출자 권한 및 소유자 권한 저장 프로시저 이해하기 섹션을 참조하십시오.

• 프로시저를 완료하는 데는 30분 이상 소요될 수 없습니다.

• 프로시저에는 OUTPUT 인자가 포함될 수 없습니다.

• Snowflake는 실패한 작업을 한 번 재시도하므로 중복되거나 의도하지 않은 결과를 초래하지 않고 여러 번 호출되는 것을 처리하도록 프로시저를 설계합니다.

• 프로시저의 필수 인자는 다음 데이터 타입 중 하나여야 합니다.

  • 숫자 데이터 타입

  • 문자열 및 이진 데이터 타입

  • 논리 데이터 타입

  • 날짜 및 시간 데이터 타입

이러한 요구 사항을 충족하는 저장 프로시저를 생성한 후에는 프로시저 및 해당 상위 데이터베이스/스키마에 대한 SNOWFLAKE 애플리케이션 USAGE 권한을 부여해야 합니다. 예를 들어, 저장 프로시저의 정규화된 이름이 ``code_db.sch1.alert_team``인 경우 다음 명령을 실행합니다.

GRANT USAGE ON DATABASE code_db TO APPLICATION SNOWFLAKE;
GRANT USAGE ON SCHEMA code_db.sch1 TO APPLICATION SNOWFLAKE;
GRANT USAGE ON PROCEDURE code_db.sch1.alert_team(STRING, NUMBER) TO APPLICATION SNOWFLAKE;

예산에 사용자 지정 작업 추가¶

계정 예산 또는 사용자 지정 예산에 여러 사용자 지정 작업을 추가할 수 있지만, 동일한 예산에 10개 이상의 사용자 지정 작업을 추가할 수는 없습니다. 사용자 지정 작업은 다음 구성 요소로 구성됩니다.

• 저장 프로시저: 호출할 프로시저에 대한 참조입니다.

• 인자: 저장 프로시저에 전달할 인자의 배열입니다.

• 임계값: 사용자 지정 작업을 트리거하는 예산 한도의 백분율(예: 75%)입니다.

• 트리거 유형: 사용자 지정 작업이 예상 소비량 또는 실제 소비량을 기준으로 트리거되는지 여부입니다.

예산에 사용자 지정 작업을 추가하려면 예산 인스턴스에서 ADD_CUSTOM_ACTION 메서드를 호출합니다. 예를 들어, 다음 코드는 지출이 예산 한도의 75%를 초과할 것으로 예상되는 경우 send_email_notification 저장 프로시저를 호출하는 사용자 지정 작업을 추가합니다.

CALL budget_db.sch1.my_budget!ADD_CUSTOM_ACTION(
  SYSTEM$REFERENCE('PROCEDURE', 'code_db.sch1.alert_team(string, string, string)'),
  ARRAY_CONSTRUCT('admin@example.com', 'Budget Alert', 'Spending at 75% of budget limit'),
  'PROJECTED',
  75);

사용자 지정 작업에 의해 호출되는 저장 프로시저 생성을 포함하는 전체 예제는 확장된 예 섹션을 참조하세요.

예산에서 사용자 지정 작업 제거¶

예산에서 사용자 지정 작업을 제거하려면 예산 인스턴스에서 REMOVE_CUSTOM_ACTIONS 메서드를 호출합니다. 이 메서드를 사용하여 다음을 수행할 수 있습니다.

• 예산에서 모든 사용자 지정 작업을 제거합니다. 예:

  CALL budget_db.sch1.my_budget!REMOVE_CUSTOM_ACTIONS();
  

  Copy

• 임계값이 지정된 모든 사용자 지정 작업을 제거합니다. 예를 들어, 사용량이 75%에 도달할 때 트리거되는 모든 사용자 지정 작업을 제거하려면 다음 명령을 실행합니다.

  CALL budget_db.sch1.my_budget!REMOVE_CUSTOM_ACTIONS(75);
  

  Copy

• 예산에서 지정된 사용자 지정 작업을 제거합니다. 예를 들어, 사용량이 75%에 도달할 때 my_sp 저장 프로시저를 호출하는 사용자 지정 작업을 제거하려면 다음 명령을 실행합니다.

  CALL budget_db.sch1.my_budget!REMOVE_CUSTOM_ACTIONS(75, 'code_db.sch1.my_sp');
  

  Copy

  팁

  특정 작업을 제거하는 경우 GET_CUSTOM_ACTIONS 메서드에서 반환되는 정규화된 프로시저 이름을 사용합니다.

확장된 예¶

다음 예제에서는 사용자 지정 작업으로 호출되는 저장 프로시저를 작성하고, 프로시저에 필요한 권한을 부여한 다음, 예산에 사용자 지정 작업을 추가하는 방법을 보여줍니다.

1. :ref:`모든 요구 사항을 준수 <label-budget_custom_actions_stored_procedure>`하는 저장 프로시저를 생성합니다.

   CREATE OR REPLACE PROCEDURE code_db.sch1.alert_team(
       integration_name string,
       email_list string,
       email_subject string,
       email_content string)
   RETURNS STRING
   LANGUAGE JAVASCRIPT
   EXECUTE AS OWNER
   AS
   $$
       var sql_command = "CALL SYSTEM$SEND_EMAIL('" + INTEGRATION_NAME + "', " +
                                               "'" +  EMAIL_LIST + "', " +
                                               "'" + EMAIL_SUBJECT + "'," +
                                               "'" + EMAIL_CONTENT + "'" + ");";
       var statement1 = snowflake.createStatement({sqlText: sql_command});
       statement1.execute();
       return "alert sent";
   $$;
   

   Copy

2. SNOWFLAKE 애플리케이션에 저장 프로시저에 대한 권한을 부여합니다.

   GRANT USAGE ON DATABASE code_db TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON SCHEMA code_db.sch1 TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON PROCEDURE code_db.sch1.alert_team(STRING, STRING, STRING, STRING)
     TO APPLICATION SNOWFLAKE;
   

   Copy

3. 사용량이 예산 지출 한도의 90%에 도달할 때 트리거되도록 예산에 사용자 지정 작업을 추가합니다.

   CALL budget_db.sch1.my_budget!ADD_CUSTOM_ACTION(
     SYSTEM$REFERENCE('PROCEDURE', 'code_db.sch1.alert_team(string, string, string, string)'),
     ARRAY_CONSTRUCT('my_int', 'admin@example.com', 'Budget Alert', 'Spending at 90% of budget limit'),
     'ACTUAL',
     90);
   

   Copy

사용자 지정 작업 문제 해결하기¶

사용자 지정 작업이 예상대로 작동하지 않으면 다음 방법을 사용하여 문제를 진단합니다.

SELECT th.*, ci.name AS budget_name
  FROM SNOWFLAKE.ACCOUNT_USAGE.TASK_HISTORY th
  JOIN SNOWFLAKE.ACCOUNT_USAGE.CLASS_INSTANCES ci
    ON th.instance_id = ci.id
  WHERE ci.class_name = 'BUDGET'
    AND th.name ILIKE 'BUDGET_CUSTOM_ACTION_TRIGGER_AT_%'
    AND ci.name = '<budget_name>'
  ORDER BY th.completed_time DESC
  LIMIT 10;

SELECT th.*, ci.name as budget_name
  FROM SNOWFLAKE.ACCOUNT_USAGE.TASK_HISTORY th
  JOIN SNOWFLAKE.ACCOUNT_USAGE.CLASS_INSTANCES ci
    ON th.instance_id = ci.id
  WHERE ci.class_name = 'BUDGET'
    AND th.name ILIKE 'BUDGET_CUSTOM_ACTION_TRIGGER_AT_%'
    AND ci.name = '<budget_name>'
    AND th.COMPLETED_TIME >= DATEADD('day', -7, CURRENT_TIMESTAMP())
  ORDER BY th.completed_time DESC;

CALL <budget_name>!GET_CUSTOM_ACTIONS();

SELECT th.*, ci.name AS budget_name
  FROM SNOWFLAKE.ACCOUNT_USAGE.TASK_HISTORY th
  JOIN SNOWFLAKE.ACCOUNT_USAGE.CLASS_INSTANCES ci
    ON th.instance_id = ci.id
  WHERE ci.class_name = 'BUDGET'
    AND th.name ILIKE 'BUDGET_CUSTOM_ACTION_TRIGGER_AT_%'
    AND th.query_text ILIKE '%<action_id>%'
    AND ci.name = '<budget_name>'
    AND th.COMPLETED_TIME >= DATEADD('day', -7, CURRENT_TIMESTAMP())
  ORDER BY th.completed_time DESC;

CALL budget_db.sch1.my_budget!CONFIRM_CUSTOM_ACTIONS_ACCESS();

CALL budget_db.sch1.my_budget!GET_CONFIG();

CALL budget_db.sch1.my_budget!GET_SPENDING_LIMIT();

CALL budget_db.sch1.my_budget!GET_SPENDING_HISTORY();