org.easybatch.extensions.msexcel.MsExcelRecord.java Source code

Introduction

Here is the source code for org.easybatch.extensions.msexcel.MsExcelRecord.java

Source

/**
 * The MIT License
 *
 *   Copyright (c) 2017, Mahmoud Ben Hassine (mahmoud.benhassine@icloud.com)
 *
 *   Permission is hereby granted, free of charge, to any person obtaining a copy
 *   of this software and associated documentation files (the "Software"), to deal
 *   in the Software without restriction, including without limitation the rights
 *   to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *   copies of the Software, and to permit persons to whom the Software is
 *   furnished to do so, subject to the following conditions:
 *
 *   The above copyright notice and this permission notice shall be included in
 *   all copies or substantial portions of the Software.
 *
 *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *   FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *   AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *   LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *   OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 *   THE SOFTWARE.
 */
package org.easybatch.extensions.msexcel;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.easybatch.core.record.GenericRecord;
import org.easybatch.core.record.Header;

/**
 * Record containing a {@link Row} as payload.
 *
 * @author Mahmoud Ben Hassine (mahmoud.benhassine@icloud.com)
 */
public class MsExcelRecord extends GenericRecord<Row> {

    /**
     * Create a new {@link MsExcelRecord}.
     *
     * @param header the record header
     * @param payload the record payload
     */
    public MsExcelRecord(Header header, Row payload) {
        super(header, payload);
    }

    @Override
    public String toString() {
        return "Record: {" + "header=" + header + ", payload=" + dump(payload) + '}';
    }

    private String dump(Row row) {
        StringBuilder stringBuilder = new StringBuilder();
        short lastCellNum = row.getLastCellNum();
        for (int i = 0; i < lastCellNum; i++) {
            Cell cell = row.getCell(i);
            stringBuilder.append("\"");
            switch (cell.getCellType()) {
            case Cell.CELL_TYPE_BOOLEAN:
                stringBuilder.append(cell.getBooleanCellValue());
                break;
            case Cell.CELL_TYPE_NUMERIC:
                stringBuilder.append(cell.getNumericCellValue());
                break;
            case Cell.CELL_TYPE_STRING:
                stringBuilder.append(cell.getStringCellValue());
                break;
            case Cell.CELL_TYPE_BLANK:
                stringBuilder.append("");
                break;
            }
            stringBuilder.append("\"");
            if (i < lastCellNum - 1) {
                stringBuilder.append(",");
            }
        }

        return stringBuilder.toString();
    }
}