更新仓库参考

数据库对象和过程

以下数据库对象是通过 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
语言: 中文