启用资源参考¶

数据库对象和过程¶

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

PUBLIC.ENABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)¶

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

PUBLIC.ENABLE_RESOURCE_VALIDATE(resource_ingestion_definition_id VARCHAR)¶

用于启用过程的连接器特定验证的过程。默认情况下，它会返回 'response_code': 'OK'。它由 DefaultEnableResourceValidator 调用。在 SQL 和 Java 中都可以被覆盖。

PUBLIC.PRE_ENABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)¶

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

PUBLIC.POST_ENABLE_RESOURCE(resource_ingestion_definition_id VARCHAR)¶

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

自定义处理程序¶

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

过程替换方法¶

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

处理程序¶

若要提供 EnableResourceHandler 的完整自定义实施，则必须替换 PUBLIC.ENABLE_RESOURCE 过程。例如：

CREATE OR REPLACE PROCEDURE PUBLIC.ENABLE_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.CustomEnableResourceHandler.enableResource';

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

内部过程¶

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

CREATE OR REPLACE PROCEDURE PUBLIC.ENABLE_RESOURCE_VALIDATE(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.ENABLE_RESOURCE_VALIDATE(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.enableResourceValidate';

生成器方法¶

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

EnableResourceValidator
PreEnableResourceCallback
PostEnableResourceCallback
ConnectorErrorHelper

如果未提供函数，则将使用 SDK 提供的默认实施。

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

class CustomHandler {

  // Path to this method needs to be specified in the PUBLIC.ENABLE_RESOURCE procedure using SQL
  public static Variant enableResource(Session session, String resourceIngestionDefinitionId) {
    //Using builder
    var handler = EnableResourceHandlerBuilder.builder(session)
      .withPreEnableResourceCallback(new CustomPreEnableResourceCallback())
      .build();
    return handler.enableResource(resourceIngestionDefinitionId).toVariant();
  }
}