org.devocative.wickomp.grid.toolbar.OExportExcelButton Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wickomp Show documentation
Web components for Apache Wicket
There is a newer version: 2.11
package org.devocative.wickomp.grid.toolbar;

import org.devocative.adroit.ExcelExporter;
import org.devocative.wickomp.grid.IGridDataSource;
import org.devocative.wickomp.grid.WSortField;
import org.devocative.wickomp.grid.column.OColumn;
import org.devocative.wickomp.grid.column.OPropertyColumn;
import org.devocative.wickomp.html.HTMLBase;
import org.devocative.wickomp.resource.OutputStreamResource;

import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public class OExportExcelButton extends OLinkButton {
	private static final long serialVersionUID = -8571985043735727439L;

	private IGridDataSource dataSource;

	private List sortFieldList = new ArrayList<>();
	private String fileName = "export.xlsx";
	private Integer maxRowsCount = 1000;

	// ------------------------------

	public OExportExcelButton(HTMLBase html, IGridDataSource dataSource) {
		super(html);

		this.dataSource = dataSource;
	}

	// ------------------------------

	public OExportExcelButton setSortFieldList(List sortFieldList) {
		this.sortFieldList = sortFieldList;
		return this;
	}

	public OExportExcelButton setFileName(String fileName) {
		this.fileName = fileName;
		return this;
	}

	public OExportExcelButton setMaxRowsCount(Integer maxRowsCount) {
		this.maxRowsCount = maxRowsCount;
		return this;
	}

	// ---------------

	@Override
	public void onClick() {
		sendResource(new OutputStreamResource("application/excel", fileName) {
			private static final long serialVersionUID = 8680636627426356006L;

			@Override
			protected void handleStream(OutputStream stream) throws IOException {
				List columnsTitle = getColumnList()
					.getVisibleColumns()
					.stream()
					.filter(column -> column instanceof OPropertyColumn)
					.map(OColumn::getTitle)
					.collect(Collectors.toList());

				ExcelExporter exporter = new ExcelExporter(fileName);
				exporter.setColumnsHeader(columnsTitle);

				List rawData = dataSource.list(1, maxRowsCount, sortFieldList);
				for (int rowNo = 0; rowNo < rawData.size(); rowNo++) {
					T bean = rawData.get(rowNo);
					List rowResult = new ArrayList<>();
					List> cols = getColumnList().getVisibleColumns();
					for (int colNo = 0; colNo < cols.size(); colNo++) {
						OColumn column = cols.get(colNo);
						if (column instanceof OPropertyColumn) {
							String cellValue = column.onCellRender(bean, String.valueOf(rowNo)) ?
								column.cellValue(bean, String.valueOf(rowNo), colNo, null) :
								"";
							rowResult.add(cellValue);
						}
					}
					exporter.addRowData(rowResult);
				}

				exporter.generate(stream);
			}
		});
	}
}