com.centurylink.mdw.model.JsonExport Maven / Gradle / Ivy
/*
* Copyright (C) 2017 CenturyLink, 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.centurylink.mdw.model;
import com.centurylink.mdw.common.service.Query;
import com.centurylink.mdw.util.DateHelper;
import com.centurylink.mdw.util.JsonUtil;
import org.apache.commons.codec.binary.Base64;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.ParseException;
import java.util.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
* TODO: other formats (eg: CSV)
*/
public class JsonExport {
private static final int ZIP_BUFFER_KB = 16;
private Jsonable jsonable;
public JsonExport(Jsonable json) {
this.jsonable = json;
}
// TODO: more flexible approach to support ordering, types, etc (com.centurylink.mdw.model.Value?)
private Map labels;
public JsonExport(Jsonable json, Map labels) {
this.jsonable = json;
this.labels = labels;
}
private List names = new ArrayList();
public String exportXlsxBase64() throws JSONException, IOException {
return exportXlsxBase64(null);
}
/**
* Default behavior adds zip entries for each property on the first (non-mdw) top-level object.
*/
public String exportZipBase64() throws JSONException, IOException {
Map objectMap = JsonUtil.getJsonObjects(jsonable.getJson());
JSONObject mdw = null;
JSONObject contents = null;
for (String name : objectMap.keySet()) {
if ("mdw".equals(name))
mdw = objectMap.get(name);
else if (contents == null)
contents = objectMap.get(name);
}
if (contents == null)
throw new IOException("Cannot find expected contents property");
else
objectMap = JsonUtil.getJsonObjects(contents);
if (mdw != null)
objectMap.put(".mdw", mdw);
byte[] buffer = new byte[ZIP_BUFFER_KB * 1024];
ZipOutputStream zos = null;
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
try {
zos = new ZipOutputStream(outputStream);
for (String name : objectMap.keySet()) {
JSONObject json = objectMap.get(name);
ZipEntry ze = new ZipEntry(name);
zos.putNextEntry(ze);
InputStream inputStream = new ByteArrayInputStream(json.toString(2).getBytes());
int len;
while ((len = inputStream.read(buffer)) > 0) {
zos.write(buffer, 0, len);
}
}
}
finally {
if (zos != null) {
zos.closeEntry();
zos.close();
}
}
byte[] bytes = outputStream.toByteArray();
return new String(Base64.encodeBase64(bytes));
}
public String exportXlsxBase64(String name) throws JSONException, IOException {
Workbook xlsx = exportXlsx();
byte[] bytes = writeExcel(xlsx);
return new String(Base64.encodeBase64(bytes));
}
public Workbook exportXlsx() throws JSONException {
return exportXlsx(null);
}
public Workbook exportXlsx(String name) throws JSONException {
Workbook workbook = new XSSFWorkbook();
Sheet sheet = workbook.createSheet(name == null ? jsonable.getClass().getSimpleName() : name);
if (jsonable instanceof JsonArray) {
JSONArray jsonArray = ((JsonArray)jsonable).getArray();
for (int i = 0; i < jsonArray.length(); i++) {
JSONObject jsonObj = jsonArray.getJSONObject(i);
addNames(jsonObj);
setRowValues(sheet, i + 1, jsonObj);
}
setColumnLabels(sheet);
}
else if (jsonable instanceof InstanceList) {
InstanceList instanceList = (InstanceList) jsonable;
List items = instanceList.getItems();
for (int i = 0; i < items.size(); i++) {
JSONObject jsonObj = items.get(i).getJson();
addNames(jsonObj);
setRowValues(sheet, i + 1, jsonObj);
}
setColumnLabels(sheet);
}
else if (jsonable instanceof JsonListMap) {
JsonListMap listMap = (JsonListMap) jsonable;
int row = 1;
List keys = new ArrayList();
keys.addAll(listMap.getJsonables().keySet());
Collections.sort(keys);
for (String key : keys) {
List jsonableList = listMap.getJsonables().get(key);
for (Jsonable jsonable : jsonableList) {
addNames(jsonable.getJson());
}
if (!names.isEmpty() && !"".equals(names.get(0))) {
names.add(0, ""); // key column
}
for (Jsonable jsonable : jsonableList) {
Row valuesRow = setRowValues(sheet, row, jsonable.getJson(), 1);
Cell cell = valuesRow.createCell(0);
cell.setCellValue(key);
row++;
}
}
setColumnLabels(sheet);
}
else {
throw new UnsupportedOperationException("Unsupported JSON type: " + jsonable);
}
return workbook;
}
private void addNames(JSONObject json) {
for (String name: JSONObject.getNames(json)) {
if (!names.contains(name)) {
if ("id".equals(name) || "count".equals(name))
names.add(0, name);
else
names.add(name);
}
}
}
private void setColumnLabels(Sheet sheet) {
Row headerRow = sheet.createRow(0);
for (int i = 0; i < names.size(); i++) {
Cell cell = headerRow.createCell(i);
String label = names.get(i);
if (labels != null && labels.containsKey(label))
label = labels.get(label);
cell.setCellValue(label);
if (!"message".equals(label))
sheet.autoSizeColumn(i);
}
}
private CellStyle dateCellStyle;
private CellStyle getDateCellStyle(Sheet sheet) {
if (dateCellStyle == null) {
dateCellStyle = sheet.getWorkbook().createCellStyle();
CreationHelper createHelper = sheet.getWorkbook().getCreationHelper();
dateCellStyle.setDataFormat(createHelper.createDataFormat().getFormat("mm/dd/yyyy hh:mm:ss")); // TODO flexible
}
return dateCellStyle;
}
private Row setRowValues(Sheet sheet, int row, JSONObject json) throws JSONException {
return setRowValues(sheet, row, json, 0);
}
private Row setRowValues(Sheet sheet, int row, JSONObject json, int startCol) throws JSONException {
Row valueRow = sheet.createRow(row);
for (int i = startCol; i < names.size(); i++) {
Cell cell = valueRow.createCell(i);
String name = names.get(i);
if (json.has(name)) {
Object jsonValue = json.get(name);
if (jsonValue instanceof Long || jsonValue instanceof Integer) {
cell.setCellValue(new Double(jsonValue.toString()));
}
else if (jsonValue instanceof Boolean) {
cell.setCellValue((Boolean)jsonValue);
}
else if (jsonValue instanceof Date) {
cell.setCellValue((Date)jsonValue);
cell.setCellStyle(getDateCellStyle(sheet));
}
else {
String stringVal = jsonValue.toString();
if (stringVal != null && (name.endsWith("Date") || "date".equals(name))) {
// try to parse as Query date
try {
cell.setCellValue(Query.getDate(stringVal));
cell.setCellStyle(getDateCellStyle(sheet));
}
catch (ParseException ex) {
// try StringHelper date
Date d = DateHelper.stringToDate(stringVal);
if (d == null) {
cell.setCellValue(stringVal);
}
else {
cell.setCellValue(d);
cell.setCellStyle(getDateCellStyle(sheet));
}
}
}
else {
cell.setCellValue(stringVal);
}
}
}
}
return valueRow;
}
private byte[] writeExcel(Workbook workBook) throws IOException {
ByteArrayOutputStream bytes = new ByteArrayOutputStream();
workBook.write(bytes);
return bytes.toByteArray();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy