io.kestra.plugin.jdbc.vertica.Batch Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of plugin-jdbc-vertica Show documentation
Manage data in Vertica databases with Kestra's JDBC plugin.
The newest version!
package io.kestra.plugin.jdbc.vertica;

import io.kestra.core.models.annotations.Example;
import io.kestra.core.models.annotations.Plugin;
import io.kestra.core.models.tasks.RunnableTask;
import io.kestra.plugin.jdbc.AbstractCellConverter;
import io.kestra.plugin.jdbc.AbstractJdbcBatch;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.ToString;
import lombok.experimental.SuperBuilder;

import java.sql.DriverManager;
import java.sql.SQLException;
import java.time.ZoneId;

@SuperBuilder
@ToString
@EqualsAndHashCode
@Getter
@NoArgsConstructor
@Schema(
    title = "Execute a batch query on a Vertica server."
)
@Plugin(
    examples = {
        @Example(
            title = "Fetch rows from a table and bulk insert to another one.",
            full = true,
            code = """
                   id: vertica_batch_query
                   namespace: company.team
                   
                   tasks:
                     - id: query
                       type: io.kestra.plugin.jdbc.vertica.Query
                       url: jdbc:vertica://dev:56982/db
                       username: vertica_user
                       password: vertica_password
                       sql: |
                         SELECT *
                         FROM xref
                         LIMIT 1500;
                       fetchType: FETCH
                       fetchType: STORE
                     
                     - id: update
                       type: io.kestra.plugin.jdbc.vertica.Batch
                       from: "{{ outputs.query.uri }}"
                       url: jdbc:vertica://prod:56982/db
                       username: vertica_user
                       password: vertica_password
                       sql: insert into xref values( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? )
                   """
        ),
        @Example(
            title = "Fetch rows from a table and bulk insert to another one, without using sql query.",
            full = true,
            code = {
                "id: vertica_batch_query",
                "namespace: company.team",
                "",
                "tasks:",
                "  - id: query",
                "    type: io.kestra.plugin.jdbc.vertica.Query",
                "    url: jdbc:vertica://dev:56982/db",
                "    username: vertica_user",
                "    password: vertica_passwd",
                "    sql: |",
                "      SELECT *",
                "      FROM xref",
                "      LIMIT 1500;",
                "    fetchType: FETCH",
                "    fetchType: STORE",
                "",
                "  - id: update",
                "    type: io.kestra.plugin.jdbc.vertica.Batch",
                "    from: \"{{ outputs.query.uri }}\"",
                "    url: jdbc:vertica://prod:56982/db",
                "    username: vertica_user",
                "    password: vertica_passwd",
                "    table: xref",
            }
        )
    }
)
public class Batch extends AbstractJdbcBatch implements RunnableTask {
    @Override
    protected AbstractCellConverter getCellConverter(ZoneId zoneId) {
        return new VerticaCellConverter(zoneId);
    }

    @Override
    public void registerDriver() throws SQLException {
        DriverManager.registerDriver(new com.vertica.jdbc.Driver());
    }
}