io.kestra.plugin.gcp.bigquery.models.TableDefinition Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of plugin-gcp Show documentation
Integrate Google Cloud Platform services with Kestra data workflows.
There is a newer version: 0.21.2
package io.kestra.plugin.gcp.bigquery.models;

import io.kestra.core.runners.RunContext;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Builder;
import lombok.Getter;
import lombok.extern.jackson.Jacksonized;

@Getter
@Builder
@Jacksonized
public class TableDefinition {
    @Schema(title = "The table's type.")
    private final Type type;

    @Schema(title = "The table's schema.")
    private final io.kestra.plugin.gcp.bigquery.models.Schema schema;

    @Schema(title = "The table definition if the type is `TABLE`.")
    private final StandardTableDefinition standardTableDefinition;

    @Schema(title = "The materialized view definition if the type is `MATERIALIZED_VIEW`.")
    private final MaterializedViewDefinition materializedViewDefinition;

    @Schema(title = "The view definition if the type is `VIEW`.")
    private final ViewDefinition viewDefinition;

    @Schema(title = "The external table definition if the type is `EXTERNAL`.")
    private final ExternalTableDefinition externalTableDefinition;

    public static  TableDefinition of(T tableDefinition) {
        TableDefinitionBuilder tableDefinitionBuilder = TableDefinition.builder()
            .type(Type.valueOf(tableDefinition.getType().toString()));

        if (tableDefinition.getSchema() != null) {
            tableDefinitionBuilder.schema(io.kestra.plugin.gcp.bigquery.models.Schema.of(tableDefinition.getSchema()));
        }

        if (tableDefinition instanceof com.google.cloud.bigquery.ViewDefinition) {
            var viewDefinition = ((com.google.cloud.bigquery.ViewDefinition) tableDefinition);

            tableDefinitionBuilder.viewDefinition(ViewDefinition.of(viewDefinition));
        } else if (tableDefinition instanceof com.google.cloud.bigquery.MaterializedViewDefinition) {
            var materializedViewDefinition = ((com.google.cloud.bigquery.MaterializedViewDefinition) tableDefinition);

            tableDefinitionBuilder.materializedViewDefinition(MaterializedViewDefinition.of(materializedViewDefinition));
        } else if (tableDefinition instanceof com.google.cloud.bigquery.ExternalTableDefinition) {
            var externalTableDefinition = ((com.google.cloud.bigquery.ExternalTableDefinition) tableDefinition);

            tableDefinitionBuilder.externalTableDefinition(ExternalTableDefinition.of(externalTableDefinition));
        } else if (tableDefinition instanceof com.google.cloud.bigquery.StandardTableDefinition) {
            var standardTableDefinition = ((com.google.cloud.bigquery.StandardTableDefinition) tableDefinition);

            tableDefinitionBuilder.standardTableDefinition(StandardTableDefinition.of(standardTableDefinition));
        }

        return tableDefinitionBuilder.build();
    }

    @SuppressWarnings("unchecked")
    public  T to(RunContext runContext) throws Exception {
        switch (this.type) {
            case VIEW:
                return (T) this.viewDefinition.to(runContext);
            case TABLE:
                return (T) (this.standardTableDefinition == null ? StandardTableDefinition.builder().build() : this.standardTableDefinition)
                    .to(runContext, this.schema);
            case EXTERNAL:
                return (T) this.externalTableDefinition.to(runContext, this.schema);
            case MATERIALIZED_VIEW:
                return (T) this.materializedViewDefinition.to(runContext);
            default:
                throw new Exception("Invalid type '" + this.type + "'");
        }
    }

    @SuppressWarnings("unused")
    public enum Type {
        TABLE,
        VIEW,
        MATERIALIZED_VIEW,
        EXTERNAL,
        MODEL
    }
}