Referenz zum Aktualisieren des Warehouses

Datenbankobjekte und Prozeduren

Die folgenden Datenbankobjekte werden über configuration/update_warehouse.sql erstellt.

PUBLIC.UPDATE_WAREHOUSE(warehouse_name STRING)

Einstiegspunkt-Prozedur, verfügbar für ADMIN-Rolle. Diese Prozedur ruft den Java-Handler UpdateWarehouseHandler.updateWarehouse auf.

PUBLIC.UPDATE_WAREHOUSE_INTERNAL(warehouse_name STRING)

Prozedur, mit der zusätzliche Konnektor-spezifische Logik bereitgestellt wird. Gibt standardmäßig 'response_code': 'OK' zurück. Wird von Standard-UpdateWarehouseCallback aufgerufen. Kann sowohl in SQL als auch Java überschrieben werden.

Kundenspezifischer Handler

Handler und ihre Interna können mit den folgenden beiden Ansätzen angepasst werden.

Ansatz: Ersetzen der Prozedur

Die folgenden Komponenten können durch SQL ersetzt werden.

Handler

Um eine kundenspezifische Implementierung von UpdateWarehouseHandler bereitzustellen, muss die Prozedur PUBLIC.UPDATE_WAREHOUSE ersetzt werden. Beispiel:

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

Interne Prozedur

Die Prozedur INTERNAL können auch über SQL angepasst werden. Kann sogar einen anderen Java-Handler aufrufen:

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

Ansatz: Builder

UpdateWarehouseHandler kann mithilfe von UpdateWarehouseHandlerBuilder angepasst werden. Dieser Builder ermöglicht es dem Entwickler, kundenspezifische Implementierungen der folgenden Schnittstellen bereitzustellen:

  • UpdateWarehouseInputValidator

  • UpdateWarehouseCallback

  • ConnectorErrorHelper

Falls eine davon nicht vorhanden ist, wird die Standardimplementierung des SDK verwendet.

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