org.repodriller.persistence.csv.CSVFile.java Source code

Introduction

Here is the source code for org.repodriller.persistence.csv.CSVFile.java

Source

/**
 * Copyright 2014 Maurcio Aniche
    
 * 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 org.repodriller.persistence.csv;

import java.io.File;
import java.io.FileOutputStream;
import java.io.PrintStream;
import java.util.Arrays;

import org.apache.commons.lang3.StringEscapeUtils;

import org.repodriller.persistence.PersistenceMechanism;

public class CSVFile implements PersistenceMechanism {

    private PrintStream ps;

    private String[] header = null;

    public CSVFile(String fileName) {
        this(fileName, false);
    }

    public CSVFile(String fileName, boolean append) {
        try {
            ps = new PrintStream(new FileOutputStream(fileName, append));
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public CSVFile(String path, String name) {
        this(path, name, false);
    }

    public CSVFile(String path, String name, boolean append) {
        this(verifyPath(path) + name, append);
    }

    public CSVFile(String path, String name, String[] header) {
        this(verifyPath(path) + name, header);
    }

    public CSVFile(String fileName, String[] header) {
        this(fileName, header, false);
    }

    public CSVFile(String fileName, String[] header, boolean append) {
        this.header = header;
        try {
            ps = new PrintStream(new FileOutputStream(fileName, append));
            if (header != null) {
                String headers = Arrays.toString(header);
                headers = headers.substring(1, headers.length() - 1);
                ps.println(headers);
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public synchronized void write(Object... line) throws CSVFileFormatException {
        if (header != null && header.length != line.length)
            throw new CSVFileFormatException("CSV Header Columns Number Differs From Writer Columns Number.");

        boolean first = true;
        for (Object o : line) {
            if (!first)
                ps.print(",");

            if (o == null)
                ps.print("null");
            else {
                String field = o.toString();
                field = StringEscapeUtils.escapeCsv(field);
                ps.print(field);
            }
            first = false;
        }

        ps.println();
        ps.flush();

    }

    public void close() {
        ps.close();
    }

    private static String verifyPath(String path) {
        char lastchar = path.charAt(path.length() - 1);
        if (lastchar != File.separatorChar)
            path = path + File.separator;
        return path;
    }
}