Atualização da referência de warehouse¶

Objetos e procedimentos de banco de dados¶

Os seguintes objetos de banco de dados são criados por meio do configuration/update_warehouse.sql.

PUBLIC.UPDATE_WAREHOUSE(warehouse_name STRING)¶

Procedimento de ponto de entrada disponível para a função ADMIN. Este procedimento invoca o manipulador Java UpdateWarehouseHandler.updateWarehouse.

PUBLIC.UPDATE_WAREHOUSE_INTERNAL(warehouse_name STRING)¶

Procedimento usado para fornecer lógica específica do conector adicional. Por padrão, ela retorna 'response_code': 'OK'. É invocado por padrão UpdateWarehouseCallback. Pode ser sobrescrito em ambos SQL e Java.

Manipulador personalizado¶

O manipulador e seus componentes internos podem ser personalizados usando as duas abordagens a seguir.

Abordagem de substituição de procedimento¶

Os seguintes componentes podem ser substituídos usando SQL.

Manipulador¶

Para fornecer uma implementação personalizada de UpdateWarehouseHandler, o procedimento PUBLIC.UPDATE_WAREHOUSE deve ser substituído. Por exemplo:

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;

Procedimento interno¶

O procedimento INTERNAL também pode ser personalizado através SQL. Ele pode até invocar outro manipulador 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;

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';

Abordagem do construtor¶

UpdateWarehouseHandler pode ser personalizado usando UpdateWarehouseHandlerBuilder. Este construtor permite que o desenvolvedor forneça implementações personalizadas das seguintes interfaces:

UpdateWarehouseInputValidator
UpdateWarehouseCallback
ConnectorErrorHelper

Caso um deles não seja fornecido - a implementação padrão fornecida pelo SDK será usada.

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();
  }
}