com.exasol.adapter.document.documentfetcher.files.csv.CsvDocumentFetcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of virtual-schema-common-document-files Show documentation

Adapter for document data access from files.

There is a newer version: 8.1.3

Show newest version

package com.exasol.adapter.document.documentfetcher.files.csv;

import static com.exasol.adapter.document.documentfetcher.files.csv.CsvConfigurationHelper.getCsvConfiguration;

import java.util.List;

import com.exasol.adapter.document.documentfetcher.files.segmentation.FileSegment;
import com.exasol.adapter.document.documentfetcher.files.segmentation.FileSegmentDescription;
import com.exasol.adapter.document.documentnode.DocumentNode;
import com.exasol.adapter.document.files.FileTypeSpecificDocumentFetcher;
import com.exasol.adapter.document.iterators.CloseableIterator;
import com.exasol.adapter.document.mapping.ColumnMapping;
import com.exasol.errorreporting.ExaError;

/**
 * {@link FileTypeSpecificDocumentFetcher} for CSV files.
 */
public class CsvDocumentFetcher implements FileTypeSpecificDocumentFetcher {
    private static final long serialVersionUID = 2783593249946168796L;
    /** Additional configuration serialized as JSON */
    private String additionalConfiguration;
    /** Expected CSV columns */
    private final List csvColumns;

    /**
     * Create a new instance.
     * 
     * @param csvColumns the expected types of the CSV columns
     */
    public CsvDocumentFetcher(final List csvColumns) {
        this.csvColumns = csvColumns;
    }

    /**
     * Setter for additional configuration options
     *
     * @param additionalConfiguration Setter for additional configuration
     */
    public void setAdditionalConfiguration(final String additionalConfiguration) {
        this.additionalConfiguration = additionalConfiguration;
    }

    @Override
    public CloseableIterator readDocuments(final FileSegment segment) {
        if (!segment.getSegmentDescription().equals(FileSegmentDescription.ENTIRE_FILE)) {
            throw new IllegalStateException(ExaError.messageBuilder("F-VSDF-26")
                    .message("The CsvDocumentFetcher does not support loading split files.").ticketMitigation()
                    .toString());
        }
        return CsvIterator.create(segment.getFile(), this.csvColumns,
                getCsvConfiguration(this.additionalConfiguration));
    }

    @Override
    public boolean supportsFileSplitting() {
        return false;
    }
}