更新仓库参考

数据库对象和过程

以下数据库对象是通过 configuration/update_warehouse.sql 创建的。

PUBLIC.UPDATE_WAREHOUSE(warehouse_name STRING)

入口点过程可用于 ADMIN 角色。此过程会调用 Java UpdateWarehouseHandler.updateWarehouse 处理程序。

PUBLIC.UPDATE_WAREHOUSE_INTERNAL(warehouse_name STRING)

用于提供附加连接器特定逻辑的过程。默认情况下，它会返回 'response_code': 'OK'。它由默认 UpdateWarehouseCallback 调用。在 SQL 和 Java 中都可以被覆盖。

自定义处理程序

可以使用以下两种方法对处理程序及其内部结构进行自定义。

过程替换方法

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

处理程序

若要提供 UpdateWarehouseHandler 的自定义实施，必须替换 PUBLIC.UPDATE_WAREHOUSE 过程。例如：

CREATE OR REPLACE PROCEDURE PUBLIC.UPDATE_WAREHOUSE(warehouse_name STRING)
RETURNS VARIANT
LANGUAGE JAVA
RUNTIME_VERSION = '11'
PACKAGES = ('com.snowflake:snowpark:1.11.0')
IMPORTS = ('/connectors-native-sdk.jar')
HANDLER = 'com.custom.handler.CustomUpdateWarehouseHandler.updateWarehouse';

GRANT USAGE ON PROCEDURE PUBLIC.UPDATE_WAREHOUSE(STRING) TO APPLICATION ROLE ADMIN;
Copy

内部过程

也可以通过 SQL 对 INTERNAL 过程进行自定义。它还可以调用另一个 Java 处理程序：

CREATE OR REPLACE PROCEDURE PUBLIC.UPDATE_WAREHOUSE_INTERNAL(warehouse_name STRING)
RETURNS VARIANT
LANGUAGE SQL
EXECUTE AS OWNER
AS
BEGIN
    -- SOME CUSTOM LOGIC

    RETURN OBJECT_CONSTRUCT('response_code', 'OK');
END;
Copy
CREATE OR REPLACE PROCEDURE PUBLIC.UPDATE_WAREHOUSE_INTERNAL(warehouse_name STRING)
    RETURNS VARIANT
    LANGUAGE JAVA
    RUNTIME_VERSION = '11'
    PACKAGES = ('com.snowflake:snowpark:1.11.0')
    IMPORTS = ('/connectors-native-sdk.jar')
    HANDLER = 'com.custom.handler.CustomUpdateWarehouseCallback.execute';
Copy

生成器方法

可以使用 UpdateWarehouseHandlerBuilder 对 UpdateWarehouseHandler 进行自定义。此生成器允许开发者提供以下界面的自定义实施：

  • UpdateWarehouseInputValidator

  • UpdateWarehouseCallback

  • ConnectorErrorHelper

如果没有提供其中一项，则将使用 SDK 提供的默认实施。

class CustomUpdateWarehouseInputValidator implements UpdateWarehouseInputValidator {

    @Override
    public ConnectorResponse validate(Identifier warehouse) {
        // CUSTOM VALIDATION LOGIC
        return ConnectorResponse.success();
    }
}

class CustomHandler {

    // Path to this method needs to be specified in the PUBLIC.UPDATE_WAREHOUSE procedure using SQL
    public static Variant updateWarehouse(Session session, String warehouseName) {
        // Using the builder
        var handler = UpdateWarehouseHandler.builder(session)
            .withInputValidator(new CustomUpdateWarehouseInputValidator())
            .build();
        return handler.updateWarehouse(warehouseName).toVariant();
    }
}
Copy
语言: 中文