org.openmetadata.csv.CsvUtil Maven / Gradle / Ivy
/*
* Copyright 2021 Collate
* 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.openmetadata.csv;
import static org.openmetadata.common.utils.CommonUtil.listOf;
import static org.openmetadata.common.utils.CommonUtil.listOrEmpty;
import static org.openmetadata.common.utils.CommonUtil.nullOrEmpty;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVFormat.Builder;
import org.apache.commons.csv.CSVPrinter;
import org.apache.commons.csv.CSVRecord;
import org.openmetadata.schema.type.EntityReference;
import org.openmetadata.schema.type.TagLabel;
import org.openmetadata.schema.type.csv.CsvFile;
import org.openmetadata.schema.type.csv.CsvHeader;
public final class CsvUtil {
public static final String SEPARATOR = ",";
public static final String FIELD_SEPARATOR = ";";
public static final String LINE_SEPARATOR = "\r\n";
private CsvUtil() {
// Utility class hides the constructor
}
public static String formatCsv(CsvFile csvFile) throws IOException {
// CSV file is generated by the backend and the data exported is expected to be correct. Hence,
// no validation
StringWriter writer = new StringWriter();
List headers = getHeaders(csvFile.getHeaders());
CSVFormat csvFormat =
Builder.create(CSVFormat.DEFAULT).setHeader(headers.toArray(new String[0])).build();
try (CSVPrinter printer = new CSVPrinter(writer, csvFormat)) {
for (List csvRecord : listOrEmpty(csvFile.getRecords())) {
printer.printRecord(csvRecord);
}
}
return writer.toString();
}
/** Get headers from CsvHeaders */
public static List getHeaders(List csvHeaders) {
List headers = new ArrayList<>();
for (CsvHeader header : csvHeaders) {
String headerString = header.getName();
if (Boolean.TRUE.equals(header.getRequired()))
headerString = String.format("%s*", header.getName());
headers.add(headerString);
}
return headers;
}
public static String recordToString(CSVRecord csvRecord) {
return recordToString(csvRecord.toList());
}
public static String recordToString(List fields) {
return nullOrEmpty(fields)
? ""
: fields.stream().map(CsvUtil::quoteCsvField).collect(Collectors.joining(SEPARATOR));
}
public static String recordToString(String[] fields) {
return recordToString(Arrays.asList(fields));
}
public static List fieldToStrings(String field) {
// Split a field that contains multiple strings separated by FIELD_SEPARATOR
return field == null ? null : listOf(field.split(FIELD_SEPARATOR));
}
public static String quote(String field) {
return String.format("\"%s\"", field);
}
/** Quote a CSV field made of multiple strings that has SEPARATOR or FIELD_SEPARATOR with " " */
public static String quoteField(List field) {
return nullOrEmpty(field)
? ""
: field.stream().map(CsvUtil::quoteCsvField).collect(Collectors.joining(FIELD_SEPARATOR));
}
public static void addField(List csvRecord, Boolean field) {
csvRecord.add(field == null ? "" : field.toString());
}
public static List addField(List csvRecord, String field) {
csvRecord.add(field);
return csvRecord;
}
public static List addFieldList(List csvRecord, List field) {
csvRecord.add(quoteField(field));
return csvRecord;
}
public static List addEntityReferences(
List csvRecord, List refs) {
csvRecord.add(
nullOrEmpty(refs)
? null
: refs.stream()
.map(EntityReference::getFullyQualifiedName)
.collect(Collectors.joining(FIELD_SEPARATOR)));
return csvRecord;
}
public static List addEntityReference(List csvRecord, EntityReference ref) {
csvRecord.add(nullOrEmpty(ref) ? null : ref.getFullyQualifiedName());
return csvRecord;
}
public static List addTagLabels(List csvRecord, List tags) {
csvRecord.add(
nullOrEmpty(tags)
? null
: tags.stream()
.filter(
tagLabel ->
tagLabel.getSource().equals(TagLabel.TagSource.CLASSIFICATION)
&& !tagLabel.getTagFQN().split("\\.")[0].equals("Tier")
&& !tagLabel.getLabelType().equals(TagLabel.LabelType.DERIVED))
.map(TagLabel::getTagFQN)
.collect(Collectors.joining(FIELD_SEPARATOR)));
return csvRecord;
}
public static List addGlossaryTerms(List csvRecord, List tags) {
csvRecord.add(
nullOrEmpty(tags)
? null
: tags.stream()
.filter(
tagLabel ->
tagLabel.getSource().equals(TagLabel.TagSource.GLOSSARY)
&& !tagLabel.getTagFQN().split("\\.")[0].equals("Tier"))
.map(TagLabel::getTagFQN)
.collect(Collectors.joining(FIELD_SEPARATOR)));
return csvRecord;
}
public static List addTagTiers(List csvRecord, List tags) {
csvRecord.add(
nullOrEmpty(tags)
? null
: tags.stream()
.filter(
tagLabel ->
tagLabel.getSource().equals(TagLabel.TagSource.CLASSIFICATION)
&& tagLabel.getTagFQN().split("\\.")[0].equals("Tier"))
.map(TagLabel::getTagFQN)
.collect(Collectors.joining(FIELD_SEPARATOR)));
return csvRecord;
}
public static void addOwner(List csvRecord, EntityReference owner) {
csvRecord.add(nullOrEmpty(owner) ? null : owner.getType() + FIELD_SEPARATOR + owner.getName());
}
public static void addUserOwner(List csvRecord, EntityReference owner) {
csvRecord.add(nullOrEmpty(owner) ? null : owner.getName());
}
private static String quoteCsvField(String str) {
if (str.contains(SEPARATOR) || str.contains(FIELD_SEPARATOR)) {
return quote(str);
}
return str;
}
}