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