启用资源参考

数据库对象和过程

执行文件 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;
Copy

内部过程

内部过程 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';
Copy

生成器方法

可以使用 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();
    }
}
Copy
语言: 中文