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

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

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();
  }
}
Copy
Linguagem: Português