de.symeda.sormas.ui.utils.GridExportStreamResource.java Source code

Introduction

Here is the source code for de.symeda.sormas.ui.utils.GridExportStreamResource.java

Source

/*******************************************************************************
 * SORMAS - Surveillance Outbreak Response Management & Analysis System
 * Copyright  2016-2018 Helmholtz-Zentrum fr Infektionsforschung GmbH (HZI)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *******************************************************************************/
package de.symeda.sormas.ui.utils;

import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

import com.opencsv.CSVWriter;
import com.vaadin.data.provider.Query;
import com.vaadin.server.Page;
import com.vaadin.server.StreamResource;
import com.vaadin.ui.Grid;
import com.vaadin.ui.Grid.Column;
import com.vaadin.ui.Notification;
import com.vaadin.ui.Notification.Type;

import de.symeda.sormas.api.FacadeProvider;
import de.symeda.sormas.api.i18n.I18nProperties;
import de.symeda.sormas.api.i18n.Strings;
import de.symeda.sormas.api.utils.CSVUtils;
import de.symeda.sormas.api.utils.DateHelper;
import de.symeda.sormas.api.utils.YesNoUnknown;

@SuppressWarnings("serial")
public class GridExportStreamResource extends StreamResource {

    public GridExportStreamResource(Grid<?> grid, String tempFilePrefix, String filename,
            String... ignoredPropertyIds) {
        super(new StreamSource() {
            @SuppressWarnings({ "unchecked", "rawtypes" })
            @Override
            public InputStream getStream() {
                List<String> ignoredPropertyIdsList = Arrays.asList(ignoredPropertyIds);
                List<Column> columns = new ArrayList<>(grid.getColumns());
                columns.removeIf(c -> c.isHidden());
                columns.removeIf(c -> ignoredPropertyIdsList.contains(c.getId()));

                try (ByteArrayOutputStream byteStream = new ByteArrayOutputStream()) {
                    try (CSVWriter writer = CSVUtils.createCSVWriter(
                            new OutputStreamWriter(byteStream, StandardCharsets.UTF_8.name()),
                            FacadeProvider.getConfigFacade().getCsvSeparator())) {

                        List<String> headerRow = new ArrayList<>();
                        columns.forEach(c -> {
                            headerRow.add(c.getCaption());
                        });
                        writer.writeNext(headerRow.toArray(new String[headerRow.size()]));

                        String[] rowValues = new String[columns.size()];

                        int totalRowCount = grid.getDataProvider().size(new Query());
                        for (int i = 0; i < totalRowCount; i += 100) {
                            grid.getDataProvider().fetch(new Query(i, 100, grid.getSortOrder(), null, null))
                                    .forEach(row -> {
                                        for (int c = 0; c < columns.size(); c++) {
                                            Column column = columns.get(c);
                                            Object value = column.getValueProvider().apply(row);
                                            String valueString;
                                            if (value != null) {
                                                if (value instanceof Date) {
                                                    valueString = DateHelper.formatLocalDateTime((Date) value);
                                                } else if (value instanceof Boolean) {
                                                    if ((Boolean) value == true) {
                                                        valueString = I18nProperties
                                                                .getEnumCaption(YesNoUnknown.YES);
                                                    } else
                                                        valueString = I18nProperties
                                                                .getEnumCaption(YesNoUnknown.NO);
                                                } else {
                                                    valueString = value.toString();
                                                }
                                            } else {
                                                valueString = "";
                                            }
                                            rowValues[c] = valueString;
                                        }
                                        writer.writeNext(rowValues);
                                    });
                            writer.flush();
                        }
                    }
                    return new BufferedInputStream(new ByteArrayInputStream(byteStream.toByteArray()));
                } catch (IOException e) {
                    // TODO This currently requires the user to click the "Export" button again or reload the page as the UI
                    // is not automatically updated; this should be changed once Vaadin push is enabled (see #516)
                    new Notification(I18nProperties.getString(Strings.headingExportFailed),
                            I18nProperties.getString(Strings.messageExportFailed), Type.ERROR_MESSAGE, false)
                                    .show(Page.getCurrent());
                    return null;
                }
            }
        }, filename);
        setMIMEType("text/csv");
        setCacheTime(0);
    }

}