com.lithium.flow.util.CsvFormats.java Source code

Introduction

Here is the source code for com.lithium.flow.util.CsvFormats.java

Source

/*
 * Copyright 2015 Lithium Technologies, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.lithium.flow.util;

import static com.google.common.base.Preconditions.checkNotNull;

import com.lithium.flow.config.Config;

import java.util.List;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.QuoteMode;

/**
 * @author Matt Ayres
 */
public class CsvFormats {
    @Nonnull
    public static CSVFormat fromConfig(@Nonnull Config config) {
        checkNotNull(config);
        switch (config.getString("csv.format", "default")) {
        case "default":
            return CSVFormat.DEFAULT;
        case "excel":
            return CSVFormat.EXCEL;
        case "mysql":
            return CSVFormat.MYSQL;
        case "rfc4180":
            return CSVFormat.RFC4180;
        case "tdf":
            return CSVFormat.TDF;
        case "custom":
            return CSVFormat.newFormat(getChar(config, "csv.delimiter", ','))
                    .withAllowMissingColumnNames(getBoolean(config, "csv.allowMissingColumnNames"))
                    .withCommentMarker(getChar(config, "csv.commentMarker"))
                    .withEscape(getChar(config, "csv.escape")).withHeader(getHeader(config, "csv.header"))
                    .withIgnoreEmptyLines(getBoolean(config, "csv.ignoreEmptyLines"))
                    .withIgnoreSurroundingSpaces(getBoolean(config, "csv.ignoreSurroundingSpaces"))
                    .withNullString(getString(config, "csv.nullString")).withQuote(getChar(config, "csv.quote"))
                    .withQuoteMode(getQuoteMode(config, "csv.quoteMode"))
                    .withRecordSeparator(getString(config, "csv.recordSeparator"))
                    .withSkipHeaderRecord(getBoolean(config, "csv.skipHeaderRecord"));
        default:
            return CSVFormat.DEFAULT;
        }
    }

    private static char getChar(@Nonnull Config config, @Nonnull String key, char def) {
        String value = config.getString(key);
        return value.isEmpty() ? def : value.charAt(0);
    }

    private static Character getChar(@Nonnull Config config, @Nonnull String key) {
        String value = config.getString(key);
        return value.isEmpty() ? null : value.charAt(0);
    }

    private static boolean getBoolean(@Nonnull Config config, @Nonnull String key) {
        return config.containsKey(key) && config.getBoolean(key);
    }

    @Nullable
    public static String[] getHeader(@Nonnull Config config, @Nonnull String key) {
        if (!config.containsKey(key)) {
            return null;
        } else {
            List<String> headers = config.getList(key);
            return headers.toArray(new String[headers.size()]);
        }
    }

    @Nullable
    private static String getString(@Nonnull Config config, @Nonnull String key) {
        return config.containsKey(key) ? config.getString(key) : null;
    }

    @Nullable
    private static QuoteMode getQuoteMode(@Nonnull Config config, @Nonnull String key) {
        switch (config.getString(key, "default")) {
        case "default":
            return null;
        case "all":
            return QuoteMode.ALL;
        case "minimal":
            return QuoteMode.MINIMAL;
        case "non_numeric":
            return QuoteMode.NON_NUMERIC;
        case "none":
            return QuoteMode.NONE;
        default:
            return null;
        }
    }
}