웨어하우스 참조 업데이트

데이터베이스 오브젝트 및 프로시저

다음 데이터베이스 오브젝트는 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

내부 프로시저

INTERNAL 프로시저는 SQL을 통해 사용자 지정할 수도 있습니다. 다른 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

빌더 접근법

UpdateWarehouseHandlerUpdateWarehouseHandlerBuilder 를 사용하여 사용자 지정할 수 있습니다. 이 빌더를 사용하면 개발자는 다음 인터페이스의 사용자 지정 구현을 제공할 수 있습니다.

  • 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