Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2021 EPAM Systems, Inc
*
* See the NOTICE file distributed with this work for additional information
* regarding copyright ownership. 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.epam.deltix.util.io;
import java.io.*;
/**
* Helper for writing correctly formatted CSV files.
*/
public class CSVWriter extends FilterWriter {
private static final char DEFAULT_SEPARATOR = ',';
private boolean closeDelegate = true;
private boolean flushEveryLine = false;
private final char separator;
public CSVWriter (String f) throws IOException {
this (new File (f));
}
public CSVWriter (String f, char separator) throws IOException {
this (new File (f), separator);
}
public CSVWriter (String f, boolean append) throws IOException {
this (new File (f), append);
}
public CSVWriter (String f, boolean append, char separator ) throws IOException {
this (new File (f), append, separator);
}
public CSVWriter (File f) throws IOException {
this (f, false);
}
public CSVWriter (File f, char separator) throws IOException {
this (f, false, separator);
}
public CSVWriter (File f, boolean append) throws IOException {
this (new BufferedWriter (new FileWriter (f, append)));
}
public CSVWriter (File f, boolean append, char separator) throws IOException {
this (new BufferedWriter (new FileWriter (f, append)), separator);
}
public CSVWriter (Writer out) {
this (out, DEFAULT_SEPARATOR);
}
public CSVWriter (Writer out, char separator) {
super (out);
this.separator = separator;
}
public CSVWriter (OutputStream os) {
this (new OutputStreamWriter (os));
}
public CSVWriter (OutputStream os, char separator, String charsetName) throws UnsupportedEncodingException {
this (new BufferedWriter (new OutputStreamWriter (os,
charsetName)),
separator);
}
public boolean getCloseDelegate () {
return closeDelegate;
}
public void setCloseDelegate (boolean closeDelegate) {
this.closeDelegate = closeDelegate;
}
public boolean getFlushEveryLine () {
return flushEveryLine;
}
public void setFlushEveryLine (boolean flushEveryLine) {
this.flushEveryLine = flushEveryLine;
}
/**
* Writes out the specified CharSequence as a separate cell.
*
* @param unescapedText The text of a single cell to print.
* @throws java.io.IOException If writer fails to write
*/
public void writeCell (CharSequence unescapedText) throws IOException {
if (unescapedText != null)
synchronized (lock) {
printCell (unescapedText, out);
}
}
/**
*/
public void writeCells (Object ... args) throws IOException {
writeCells (true, args);
}
/**
*/
public void writeCells (boolean first, Object ... args) throws IOException {
synchronized (lock) {
for (Object arg : args) {
if (first)
first = false;
else
write (separator);
if (arg != null)
printCell (arg.toString (), out);
}
}
}
/**
* Writes out the specified CharSequence as a separate cell.
*
* @param cell The text of a single cell to print.
* @throws java.io.IOException If writer fails to write
*/
public void writeCell (Object cell) throws IOException {
if (cell != null)
synchronized (lock) {
printCell (cell.toString (), out);
}
}
public void writeLine () throws IOException {
synchronized (lock) {
write ('\n');
if (flushEveryLine)
flush ();
}
}
public void writeSeparator () throws IOException {
synchronized (lock) {
write (separator);
}
}
/**
* Method with variable arguments, which writes each object out as
* a separate cell, then outputs a line break.
* @param args Any objects to be written out (using toString()).
* @throws java.io.IOException If writer fails to write
*/
public void writeLine (Object ... args) throws IOException {
synchronized (lock) {
writeCells (args);
write ('\n');
if (flushEveryLine)
flush ();
}
}
@Override
public void close () throws IOException {
if (closeDelegate)
super.close ();
else
super.flush ();
}
/**
* Prints text to CSV cell, escaping it if necessary.
*
* @param unescapedText The text of a single cell to print.
* @param wr The CSV format writer
* @throws java.io.IOException If writer fails to write
*/
public static void printCell (CharSequence unescapedText, Appendable wr) throws IOException {
int len = unescapedText.length ();
if (len == 0)
return;
boolean needEscape = false;
search: for (int ii = 0; ii < len; ii++) {
switch (unescapedText.charAt (ii)) {
case '"':
case ',':
needEscape = true;
break search;
}
}
if (needEscape)
wr.append ('"');
for (int ii = 0; ii < len; ii++) {
char ch = unescapedText.charAt (ii);
if (ch == '"')
wr.append ('"');
wr.append (ch);
}
if (needEscape)
wr.append ('"');
}
}
