禁用资源参考¶

数据库对象和过程¶

执行文件 ingestion/resource_management.sql 时将创建以下数据库对象。

PUBLIC.DISABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)¶

入口点过程可用于 ADMIN 角色。此过程会调用 Java 函数 DisableResourceHandler.disableResource。

PUBLIC.PRE_DISABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)¶

用于添加连接器特定逻辑的过程，该过程在禁用资源前调用。默认情况下，它会返回 'response_code': 'OK'。它由 DefaultPreDisableResourceCallback 调用。在 SQL 和 Java 中都可以被覆盖。

PUBLIC.POST_DISABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)¶

用于添加连接器特定逻辑的过程，该过程在禁用资源后调用。默认情况下，它会返回 'response_code': 'OK'。它由 DefaultPostDisableResourceCallback 调用。在 SQL 和 Java 中都可以被覆盖。

自定义处理程序¶

可以使用以下方法对处理程序及其内部组件进行自定义。

过程替换方法¶

可以使用 SQL 替换以下组件。

处理程序¶

若要提供 DisableResourceHandler 的自定义实施，请替换 PUBLIC.DISABLE_RESOURCE 过程。

例如：

CREATE OR REPLACE PROCEDURE PUBLIC.DISABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)
  RETURNS VARIANT
  LANGUAGE JAVA
  RUNTIME_VERSION = '11'
  PACKAGES = ('com.snowflake:snowpark:1.11.0')
  IMPORTS = ('/connectors-native-sdk.jar')
  HANDLER = 'com.custom.handler.CustomDisableResourceHandler.disableResource';

GRANT USAGE ON PROCEDURE PUBLIC.DISABLE_RESOURCE(VARCHAR) TO APPLICATION ROLE ADMIN;

内部过程¶

内部过程 PRE_DISABLE_RESOURCE 和 POST_DISABLE_RESOURCE 也可通过 SQL 进行自定义。这些程序还可以调用其他 Java 处理程序：

CREATE OR REPLACE PROCEDURE PUBLIC.PRE_DISABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)
  RETURNS VARIANT
  LANGUAGE SQL
  EXECUTE AS OWNER
  AS
  BEGIN
      -- SOME CUSTOM LOGIC BEGIN
      SELECT sysdate();
      -- SOME CUSTOM LOGIC END

      RETURN OBJECT_CONSTRUCT('response_code', 'OK');
  END;

CREATE OR REPLACE PROCEDURE PUBLIC.PRE_DISABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)
  RETURNS VARIANT
  LANGUAGE JAVA
  RUNTIME_VERSION = '11'
  PACKAGES = ('com.snowflake:snowpark:1.11.0')
  IMPORTS = ('/connectors-native-sdk.jar')
  HANDLER = 'com.custom.handler.CustomHandler.disableResourceValidate';

生成器方法¶

可以使用 DisableResourceHandlerBuilder 对 DisableResourceHandler 进行自定义。此生成器允许用户提供以下界面的自定义实施：

PreDisableResourceCallback
PostDisableResourceCallback
ConnectorErrorHelper

如果没有提供函数，则可使用 SDK 提供的默认实施。

class CustomPreDisableResourceCallback implements PreDisableResourceCallback {
  @Override
  public ConnectorResponse execute(String resourceIngestionDefinitionId) {
    // CUSTOM LOGIC
    return ConnectorResponse.success();
  }
}

class CustomHandler {

  // Path to this method needs to be specified in the PUBLIC.DISABLE_RESOURCE procedure using SQL
  public static Variant disableResource(Session session, String resourceIngestionDefinitionId) {
    //Using builder
    var handler = DisableResourceHandlerBuilder.builder(session)
      .withPreDisableResourceCallback(new CustomPreDisableResourceCallback())
      .build();
    return handler.disableResource(resourceIngestionDefinitionId).toVariant();
  }
}