Referência da retomada do conector

Objetos e procedimentos de banco de dados

Os seguintes objetos de banco de dados são criados por meio do arquivo lifecycle/resume.sql.

PUBLIC.RESUME_CONNECTOR()

O procedimento de ponto de entrada disponível para a função ADMIN. Este procedimento invoca a função Java ResumeConnectorHandler.resumeConnector

PUBLIC.RESUME_CONNECTOR_VALIDATE()

O procedimento usado para validação específica do conector do processo de pausa. Por padrão, ela retorna 'response_code': 'OK'. É invocado por DefaultResumeConnectorStateValidator. Pode ser sobrescrito em ambos SQL e Java.

PUBLIC.RESUME_CONNECTOR_INTERNAL()

O procedimento usado para tarefas de pausa adicionais específicas do conector. Por padrão, ela retorna 'response_code': 'OK'. É invocado pelo InternalResumeConnectorCallback. 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 implementação personalizada completa do ResumeConnectorHandler, o procedimento PUBLIC.RESUME_CONNECTOR deve ser substituído. Por exemplo:

CREATE OR REPLACE PROCEDURE PUBLIC.RESUME_CONNECTOR()
RETURNS VARIANT
LANGUAGE JAVA
RUNTIME_VERSION = '11'
PACKAGES = ('com.snowflake:snowpark:1.11.0')
IMPORTS = ('/connectors-native-sdk.jar')
HANDLER = 'com.custom.handler.CustomResumeConnectorHandler.resumeConnector';

GRANT USAGE ON PROCEDURE PUBLIC.RESUME_CONNECTOR() TO APPLICATION ROLE ADMIN;

Procedimentos internos

Os procedimentos internos VALIDATE e INTERNAL também podem ser personalizados através do SQL. Eles podem até invocar outro manipulador Java:

CREATE OR REPLACE PROCEDURE PUBLIC.RESUME_CONNECTOR_INTERNAL()
RETURNS VARIANT
LANGUAGE SQL
EXECUTE AS OWNER
AS
BEGIN
    -- SOME CUSTOM LOGIC BEGIN
    SELECT sysdate();
    -- SOME CUSTOM LOGIC END

    RETURN OBJECT_CONSTRUCT('response_code', 'OK');
END;

CREATE OR REPLACE PROCEDURE PUBLIC.RESUME_CONNECTOR_VALIDATE()
RETURNS VARIANT
LANGUAGE JAVA
RUNTIME_VERSION = '11'
PACKAGES = ('com.snowflake:snowpark:1.11.0')
IMPORTS = ('/connectors-native-sdk.jar')
HANDLER = 'com.custom.handler.CustomResumeConnectorInternalHandler.resumeConnector';

Abordagem do construtor

ResumeConnectorHandler pode ser personalizado usando ResumeConnectorHandlerBuilder. Este construtor permite que o usuário forneça implementações personalizadas das seguintes interfaces:

  • ResumeConnectorStateValidator

  • ResumeConnectorCallback

  • ConnectorErrorHelper

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

class CustomResumeConnectorStateValidator implements ResumeConnectorStateValidator {
    @Override
    public ConnectorResponse validate() {
        // CUSTOM LOGIC
        return ConnectorResponse.success();
    }
}

class CustomHandler {

    // Path to this method needs to be specified in the PUBLIC.RESUME_CONNECTOR procedure using SQL
    public static Variant resumeConnector(Session session) {
            //Using builder
        var handler = ResumeConnectorHandlerBuilder.builder(session)
            .withStateValidator(new CustomResumeConnectorStateValidator())
            .build();
        return handler.resumeConnector().toVariant();
    }
}